为工厂类创建的对象注入模拟

java junit spring mockito

2022-09-02 20:56:20

我有以下类：

public class MyClass {        
    private Apple apple;

    public void myMethod() {
       apple = AppleFactory.createInstance(someStringVariable);
       ....
       ....
       ....
    }
}

和测试类：

@RunWith(MockitoJUnitRunner.class)
public class MyClassTest {

        @InjectMocks 
        MyClass myClass;

        @Test
        public void myMethod(){
         ...
         ...
         ...
        }
    }

如何在MyClass中将Apple实例作为模拟注入？

答案 1

您有3种可能性来解决这个问题：

抽象工厂：不使用静态方法，而是使用具体的工厂类：

public abstract class AppleFactory {
    public Apple createInstance(final String str);
}

public class AppleFactoryImpl implements AppleFactory {
    public Apple createInstance(final String str) { // Implementation }
}

在测试类中，模拟工厂：

@RunWith(MockitoJUnitRunner.class)
public class MyClassTest {

    @Mock
    private AppleFactory appleFactoryMock;

    @Mock
    private Apple appleMock;

    @InjectMocks 
    MyClass myClass;

    @Before
    public void setup() {
        when(appleFactoryMock.createInstance(Matchers.anyString()).thenReturn(appleMock);
    }

    @Test
    public void myMethod(){
     ...
     ...
     ...
    }
}

PowerMock：使用 PowerMock 创建静态方法的模拟。看看我对相关问题的回答，看看它是如何完成的。

可测试类：将创建包装在方法中，并创建一个重写它的测试类：Appleprotected

public class MyClass {
   private Apple apple;

   public void myMethod() {
       apple = createApple();
       ....
       ....
       ....
   }

   protected Apple createApple() {
       return AppleFactory.createInstance(someStringVariable);
   }
}


@RunWith(MockitoJUnitRunner.class)
public class MyClassTest {

    @Mock
    private Apple appleMock;

    @InjectMocks 
    MyClass myClass;

    @Test
    public void myMethod(){
     ...
     ...
     ...
    }

    private class TestableMyClass extends MyClass {
       @Override
       public void createApple() {
          return appleMock;
       }
    }
}

当然，在你的测试类中，你应该测试而不是。TestableMyClassMyClass

我会告诉你我对每种方法的看法：

抽象工厂方法是最好的 - 这是一个清晰的设计，隐藏了实现细节
可测试类 - 是第二个选项，需要最少的更改
这个选项是我最不喜欢的 - 而不是去一个更好的设计，你忽略并隐藏你的问题。但这仍然是一个有效的选择。PowerMock

答案 2

除了Avi提出的解决方案外，您还可以选择第四种可能性：

注入工厂：对我来说，当你已经有代码要重新计算时，这是最好的选择。使用此解决方案，您不必更改属性代码，而只需更改工厂类和测试。

public class AppleFactory
{
    private static Apple _injectedApple;

    public static createInstance(String str)
    {
        if (_injectedApple != null)
        {
            var currentApple = _injectedApple;
            _injectedApple = null;
            return currentApple;
        }

        //standard implementation
    }

    public static setInjectedApple(Apple apple)
    {
        _injectedApple = apple;
    }
}

现在，您只需使用静态工厂：

@RunWith(MockitoJUnitRunner.class)
public class MyClassTest {

    @Mock
    private Apple appleMock;

    @InjectMocks 
    MyClass myClass;

    @Before
    public void setup() {
        AppleFactory.setInjectedApple(appleMock);
    }

    @Test
    public void myMethod(){
     ...
     ...
     ...
    }
}