如何为单元测试期间在流程中调用的静态方法返回不同的值？

Question

我正在尝试为以下代码段编写单元测试。

class ABC {
    int getMyValue(final Activity activity) {
        if(MyClass.getInstance(activity).getValue() == 1) return 10;
        else return 20;
    }

    void doSomething() {
    }
}

我已经尝试过类似的方法来测试doSomething函数。

mABC = new ABC();

public void test_doSomething() {
   doReturn(20).when(mABC).getMyValue();
   //validate
}

如何getMyValue类似方式测试getMyValue ？ 我想断言当值为 1 时，它返回给我 10，而在所有其他情况下，它返回给我 20。

我在我的 android 应用程序中这样做。 是否有任何现有的框架可以帮助我做到这一点？

编辑：

MyClass看起来像这样

public class MyClass {

   private static Context mContext;
   public static getInstance(Context context) {
     mContext = context;
     return new MyClass();
   }

   private MyClass() {}

   public void getDreamValue() {
     Settings.Secure.getInt(mContext.getContentResolver(), "dream_val", -1);
   }
}

Answer 1

您可以考虑如下修改您的MyClass 。

public class MyClass {

   private static Context mContext;

   // Create a private variable that holds the instance. 
   private Myclass instance;

   public static getInstance(Context context) {
     mContext = context;

     if (instance == null) 
         instance = new MyClass(); // Assign the instance here 

     return instance;
   }

   private MyClass() {}

   public void getDreamValue() {
     Settings.Secure.getInt(mContext.getContentResolver(), "dream_val", -1);
   }
}

现在，当您使用 Robolectric 时，您可以在测试类中将instance值设置为模拟，如下所示。

@RunWith(RobolectricTestRunner.class) 
public class ABCTest {

    @Mock
    MyClass mockInstance; 

    @Mock
    Context mockContext; 

    @Before
    public void setup() {
        MockitoAnnotations.initMocks(this);

        // Set the mock instance for MyClass
        ReflectionHelpers.setStaticField(MyClass.class, "instance", mockInstance);
    }

    @Test
    public void testWhen1() {
       doReturn(1).when(mockInstance).getDreamValue();
       Assert.assertEquals(10, new ABC().getMyValue());
    }

    @Test
    public void testWhenNot1() {
       doReturn(2).when(mockInstance).getDreamValue();
       Assert.assertEquals(20, new ABC().getMyValue());
    }

    @After
    public void tearDown() {
        // Set the instance to null again to enable further tests to run 
        ReflectionHelpers.setStaticField(MyClass.class, "instance", null);
    }
}

我希望这有帮助。

注意：看起来您正在尝试提供MyClass的单例实例。 因此，您真的不应该在getInstance函数中创建MyClass的新实例。 我避免每次都创建一个新实例，在我的代码中使用null检查。