如何对私有方法触发的Observer进行单元测试？

我有一个名为SampleViewModel的ViewModel类，它包含一个DataSourceObserver观察者，该观察者的onDataUpdated事件由ViewModel的数据源类ViewModelDataSource的私有方法触发。我想要为ViewModel的观察者处理onDataUpdated的逻辑添加单元测试，使用Mockito进行对象模拟/存根，但不知道如何触发observer.onDataUpdated并验证其处理逻辑中的方法调用。

代码示例 #

@HiltViewModel
public class SampleViewModel extends ViewModel {

  private final @NonNull ViewModelDataSource viewModelDataSource;

  private final @NonNull ViewModelDataSource.DataSourceObserver observer = new ViewModelDataSource.DataSourceObserver() {
    @Override
    public void onDataUpdated(){
      // 这里是需要测试的处理逻辑
    }
  };

  @Inject
  public SampleViewModel(final @NonNull ViewModelDataSource dataSource){
    viewModelDataSource = dataSource;
    viewModelDataSource.setObserver(observer);
  }
}

public class ViewModelDataSource{
  public interface DataSourceObserver {
    void onDataUpdated();
  }

  private final ContactObservers.Observer contactObserver = new ContactObservers.Observer(){
    @Override
    public void onContactUpdated(){
      if (myObserver != null){
        myObserver.onDataUpdated();
      }
    }
  };

  private @Nullable DataSourceObserver myObserver;
  
  @Inject
  public ViewModelDataSource(){
    // 其他初始化逻辑
  }

  public void setObserver(DataSourceObserver observer){
    myObserver = observer;
  }
}

针对这个场景，有三种可行的测试方案，根据你的需求和源码修改权限选择：

方案一：直接获取Observer并手动触发（最简单） #

这种方法跳过数据源的内部逻辑，直接测试Observer的处理逻辑，适合单独验证onDataUpdated里的业务代码。

import org.junit.Test;
import org.junit.Assert;
import java.lang.reflect.Field;

public class SampleViewModelTest {

    @Test
    public void testOnDataUpdatedHandling() {
        // 1. 模拟ViewModelDataSource（构造需要，用mock即可）
        ViewModelDataSource mockDataSource = org.mockito.Mockito.mock(ViewModelDataSource.class);
        
        // 2. 创建ViewModel实例
        SampleViewModel viewModel = new SampleViewModel(mockDataSource);
        
        // 3. 通过反射获取私有observer成员
        ViewModelDataSource.DataSourceObserver observer = null;
        try {
            Field observerField = SampleViewModel.class.getDeclaredField("observer");
            observerField.setAccessible(true); // 允许访问私有字段
            observer = (ViewModelDataSource.DataSourceObserver) observerField.get(viewModel);
        } catch (NoSuchFieldException | IllegalAccessException e) {
            e.printStackTrace();
            Assert.fail("无法获取ViewModel中的observer字段");
        }
        
        // 4. 手动调用onDataUpdated，触发处理逻辑
        observer.onDataUpdated();
        
        // 5. 验证处理逻辑是否正确执行
        // 示例：如果处理逻辑更新了某个LiveData，验证其值
        // Assert.assertEquals("预期值", viewModel.getSomeLiveData().getValue());
        // 示例：如果处理逻辑调用了某个依赖方法，提前mock该依赖并验证调用次数
        // org.mockito.Mockito.verify(mockDependency).someMethod();
    }
}

方案二：反射触发数据源的私有事件（测试完整调用链） #

如果你需要测试从ContactObservers.Observer.onContactUpdated到DataSourceObserver.onDataUpdated的完整流程，可以通过反射调用数据源的私有观察者方法。

import org.junit.Test;
import org.junit.Assert;
import java.lang.reflect.Field;

public class SampleViewModelTest {

    @Test
    public void testContactUpdatedTriggersDataUpdated() throws Exception {
        // 1. 创建真实的ViewModelDataSource实例（如果有依赖，需要提前模拟或注入）
        ViewModelDataSource dataSource = new ViewModelDataSource();
        
        // 2. 创建ViewModel并绑定数据源
        SampleViewModel viewModel = new SampleViewModel(dataSource);
        
        // 3. 反射获取数据源中的私有contactObserver
        Field contactObserverField = ViewModelDataSource.class.getDeclaredField("contactObserver");
        contactObserverField.setAccessible(true);
        ContactObservers.Observer contactObserver = (ContactObservers.Observer) contactObserverField.get(dataSource);
        
        // 4. 调用onContactUpdated，触发整个流程
        contactObserver.onContactUpdated();
        
        // 5. 验证ViewModel的处理逻辑是否执行
        // 例如：Assert.assertEquals("预期结果", viewModel.getSomeLiveData().getValue());
    }
}

方案三：添加测试专用触发方法（最优雅，需修改源码） #

如果有权限修改ViewModelDataSource，可以添加一个仅用于测试的方法，暴露触发逻辑，避免反射带来的耦合。

修改后的ViewModelDataSource： #

import androidx.annotation.VisibleForTesting;

public class ViewModelDataSource{
  // ... 原有代码 ...
  
  // 仅用于测试，暴露触发onContactUpdated的入口
  @VisibleForTesting
  public void triggerOnContactUpdated() {
    contactObserver.onContactUpdated();
  }
}

测试代码： #

import org.junit.Test;
import org.junit.Assert;

public class SampleViewModelTest {

    @Test
    public void testOnDataUpdatedViaTrigger() {
        ViewModelDataSource dataSource = new ViewModelDataSource();
        SampleViewModel viewModel = new SampleViewModel(dataSource);
        
        // 直接调用测试专用方法触发流程
        dataSource.triggerOnContactUpdated();
        
        // 验证处理逻辑
        // Assert.assertEquals("预期值", viewModel.getSomeLiveData().getValue());
    }
}

• 方案一：快速简单，不依赖数据源内部结构，适合单独测试Observer的业务逻辑，但无法验证完整调用链。
• 方案二：测试完整流程，但依赖类的私有成员结构，后续源码修改可能导致测试失效。
• 方案三：代码最清晰、可维护性最高，但需要修改原有源码，适合有权限调整代码的场景。

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