如何测试ViewModel在扩展AndroidViewModel时

在扩展AndroidViewModel的ViewModel中，我们可以使用InstrumentationRegistry.getInstrumentation().context获取ApplicationContext。然后，我们可以使用此ApplicationContext来初始化AndroidViewModel。

以下是一个示例代码：

public class MyViewModel extends AndroidViewModel {

   private someRepository repo;

   public MyViewModel(Application application) {
       super(application);
       repo = new SomeRepository(application);
   }

   public LiveData<List<SomeEntity>> getEntities() {
       return repo.getEntities();
   }
}

要测试此ViewModel，我们可以为其创建一个测试类，具体如下：

@RunWith(AndroidJUnit4.class)
public class MyViewModelTest {

   private MyViewModel viewModel;

   @Before
   public void setup() {
       Context context = InstrumentationRegistry.getInstrumentation().getTargetContext().getApplicationContext();
       viewModel = new MyViewModel(context);
   }

   @After
   public void teardown() {
       viewModel = null;
   }

   @Test
   public void testGetEntities() {
       LiveData<List<SomeEntity>> entities = viewModel.getEntities();
       assertNotNull(entities);
       entities.observeForever(new Observer<List<SomeEntity>>() {
           @Override
           public void onChanged(List<SomeEntity> someEntities) {
               assertNotNull(someEntities);
           }
       });
   }
}

在setup（）方法中，我们使用getApplicationContext（）获得ApplicationContext，然后将其传递给MyViewModel构造函数。在teardown（）方法中，我们将ViewModel设置为null以释放资源。在testGetEntities（）方法中，我们测试getEntities（）方法是否返回LiveData对象，并在Observer中断言对象是否非null。