Java中能否用if-else返回不同类对象？自动化测试场景咨询

我来分两个问题给你详细解答：

当然可以！不过有个关键前提：这些不同的类必须继承自同一个父类，或者实现同一个接口。因为Java是强类型语言，方法的返回类型必须是明确的，所以你需要用这个共同的父类/接口作为方法的返回类型，这样在if-else分支里返回它的子类实现就完全合法。

举个简单的示例代码：

// 定义一个通用接口
interface Animal {
    void makeSound();
}

class Dog implements Animal {
    @Override
    public void makeSound() {
        System.out.println("汪汪汪");
    }
}

class Cat implements Animal {
    @Override
    public void makeSound() {
        System.out.println("喵喵喵");
    }
}

public class Demo {
    public static Animal getAnimal(boolean isDog) {
        if (isDog) {
            return new Dog(); // 返回Dog实例，属于Animal类型
        } else {
            return new Cat(); // 返回Cat实例，也属于Animal类型
        }
    }

    public static void main(String[] args) {
        Animal animal = getAnimal(true);
        animal.makeSound(); // 输出"汪汪汪"
    }
}

通过统一的返回类型，就能在不同分支返回不同类的对象啦。

针对你的场景，我们可以按照以下步骤来实现：

第一步：给四个认证类定义统一的父类型 #

首先让A、B、C、D这四个类实现同一个接口（或者继承同一个抽象类），比如叫AuthStep2，里面可以定义第二步需要用到的通用方法，比如输入信息、验证页面等：

// 定义第二步的统一接口
interface AuthStep2 {
    void fillAuthInfo(); // 示例：填充认证信息
    void verifyPage();   // 示例：验证当前页面是否正确
}

// 认证类A实现接口
class AuthStep2A implements AuthStep2 {
    @Override
    public void fillAuthInfo() {
        // 实现A类第二步的填充逻辑
        System.out.println("填充认证类A的信息");
    }

    @Override
    public void verifyPage() {
        // 验证A类第二步页面
        System.out.println("验证认证类A的页面");
    }
}

// 同理实现B、C、D类
class AuthStep2B implements AuthStep2 {
    @Override
    public void fillAuthInfo() {
        System.out.println("填充认证类B的信息");
    }

    @Override
    public void verifyPage() {
        System.out.println("验证认证类B的页面");
    }
}

// AuthStep2C、AuthStep2D的实现类似，这里省略

第二步：改造CreateClassStep1类 #

在CreateClassStep1中，我们需要先记录用户选择的认证类型（用枚举规范类型避免硬编码），然后在nextButton()方法中根据这个类型返回对应的AuthStep2实例：

先定义认证类型枚举：

enum AuthType {
    TYPE_A, TYPE_B, TYPE_C, TYPE_D
}

然后改造CreateClassStep1：

class CreateClassStep1 {
    private AuthType selectedAuthType; // 记录选中的认证类型

    // 通过构造方法传入选中的认证类型
    public CreateClassStep1(AuthType selectedAuthType) {
        this.selectedAuthType = selectedAuthType;
    }

    // 也可以用setter方法动态设置认证类型
    public void setSelectedAuthType(AuthType selectedAuthType) {
        this.selectedAuthType = selectedAuthType;
    }

    public AuthStep2 nextButton() {
        // 执行点击下一步按钮的操作
        // nexbutton.click(); 保留你原有的点击逻辑
        System.out.println("点击下一步按钮");

        // 根据选中类型返回对应第二步对象
        switch (selectedAuthType) {
            case TYPE_A:
                return new AuthStep2A();
            case TYPE_B:
                return new AuthStep2B();
            case TYPE_C:
                return new AuthStep2C();
            case TYPE_D:
                return new AuthStep2D();
            default:
                throw new IllegalArgumentException("不支持的认证类型：" + selectedAuthType);
        }

        // 如果你习惯用if-else，也可以替换成下面的逻辑：
        /*
        if (selectedAuthType == AuthType.TYPE_A) {
            return new AuthStep2A();
        } else if (selectedAuthType == AuthType.TYPE_B) {
            return new AuthStep2B();
        } else if (selectedAuthType == AuthType.TYPE_C) {
            return new AuthStep2C();
        } else if (selectedAuthType == AuthType.TYPE_D) {
            return new AuthStep2D();
        } else {
            throw new IllegalArgumentException("不支持的认证类型：" + selectedAuthType);
        }
        */
    }
}

第三步：测试调用示例 #

在你的测试代码中，可以这样使用：

public class TestDemo {
    public static void main(String[] args) {
        // 假设用户选择了认证类型A
        CreateClassStep1 step1 = new CreateClassStep1(AuthType.TYPE_A);
        AuthStep2 step2 = step1.nextButton(); // 返回AuthStep2A实例
        step2.fillAuthInfo(); // 执行A类第二步的填充逻辑
        step2.verifyPage();   // 验证A类第二步页面
    }
}

额外优化建议 #

如果后续可能增加更多认证类型，switch/if-else会越来越臃肿，这时候可以用工厂模式封装对象创建逻辑，让代码更易于维护：

class AuthStep2Factory {
    public static AuthStep2 createAuthStep2(AuthType authType) {
        switch (authType) {
            case TYPE_A:
                return new AuthStep2A();
            case TYPE_B:
                return new AuthStep2B();
            case TYPE_C:
                return new AuthStep2C();
            case TYPE_D:
                return new AuthStep2D();
            default:
                throw new IllegalArgumentException("不支持的认证类型：" + authType);
        }
    }
}

然后CreateClassStep1的nextButton()方法可以简化为：

public AuthStep2 nextButton() {
    nexbutton.click();
    return AuthStep2Factory.createAuthStep2(selectedAuthType);
}

这样后续新增认证类型时，只需要修改工厂类，不需要改动CreateClassStep1的代码，符合开闭原则。

内容的提问来源于stack exchange，提问作者Premjeet Kumar