如何对 WorkManager Worker 进行单元测试

Question

如何对 WorkManager Worker 进行单元测试？

这里是示例 Worker：

public class SampleWorker extends Worker {

    private static final String TAG = "SampleWorker";
    private static final String WORKER_TAG = "SAMPLE";

    static final String KEY_DATA_1 = "KEY_DATA_1";
    static final String KEY_DATA_2 = "KEY_DATA_2";


    public SampleWorker(@NonNull Context context, @NonNull WorkerParameters workerParams) {
        super(context, workerParams);
    }

    public static WorkRequest createWorkRequest(String data1, String data2) {
        Constraints constraints = new Constraints.Builder()
                .setRequiredNetworkType(NetworkType.CONNECTED)
                .build();

        Data inputData = createData(data1, data2);

        return new OneTimeWorkRequest.Builder(SampleWorker.class)
                .setConstraints(constraints)
                .setInputData(inputData)
                .addTag(WORKER_TAG)
                .build();
    }

    static Data createData(String data1, String data2) {
        return new Data.Builder()
                .putString(KEY_DATA_1, data1)
                .putString(KEY_DATA_2, data2)
                .build();
    }

    @NonNull
    @Override
    public Result doWork() {
        try {
            Data inputData = getInputData();
            businessLogicHere(inputData.getString(KEY_DATA_1), inputData.getString(KEY_DATA_2));

            return Result.SUCCESS;
        } catch (Exception e) {
            Log.e(TAG, "doWork failed", e);
            return Result.FAILURE;
        }
    }

    private void businessLogicHere(String data1, String data2) {
        // no-op
    }
}

Answer 1

在 Work Manager alpha10 版本中，添加了一个带有 WorkerParameters 的构造函数，因此更容易传递输入 Data

注意：我在 Worker 类中创建了工厂方法，因此更容易读取在 Worker 中传递和使用的输入数据。

这里是示例单元测试：

  @RunWith(PowerMockRunner.class)
    @PrepareForTest({WorkerParameters.class})
    public class SampleWorkerTest {

    @Mock
    Context context;

    WorkerParameters workerParams;

    Data inputData;

    SampleWorker worker;

    @Before
    public void setUp() throws Exception {
        MockitoAnnotations.initMocks(this);
        workerParams = PowerMockito.mock(WorkerParameters.class);

        inputData = SampleWorker.createData("data one", "data two");
        when(workerParams.getInputData()).thenReturn(inputData);

        worker = new SampleWorker(context, workerParams);
    }

    @Test
    public void testCreateWorkRequestFactoryMethod() {
        WorkRequest workRequest = SampleWorker.createWorkRequest("data one", "data two");
        WorkSpec workSpec = workRequest.getWorkSpec();

        // verify constraints
        assertThat(workSpec.constraints.getRequiredNetworkType(), is(equalTo(NetworkType.CONNECTED)));

        // verify input data
        assertThat(workSpec.input.getString(SampleWorker.KEY_DATA_1), is(equalTo("data one")));
        assertThat(workSpec.input.getString(SampleWorker.KEY_DATA_2), is(equalTo("data two")));
    }

    @Test
    public void testDoWorkSuccess() {
        Worker.Result result = worker.doWork();

        // verify business logic here

        assertThat(result, is(equalTo(Worker.Result.SUCCESS)));
    }
}