1

我正在尝试测试一个DoFn<KV<String, twitter4j.Status>, String>实现，并提供测试数据作为输入。我正在探索的一个途径是使用一个Mockito.mock对象作为输入，因为有大量的抽象方法可以实现。但是，在 my 中调用模拟方法DoFn会更改对象，因此测试框架会抱怨“输出后不得以任何方式改变值”。

有没有另一种方法来完成我在这里尝试的事情？测试代码大致为：

 Status status = mock(Status.class, withSettings().serializable());
 when(status.getText()).thenReturn("bar");

 Pipeline p = TestPipeline.create();
 PCollection<String> strings = p
     .apply(Create.of(KV.of("foo", status)))
     .apply(MapElements.via(new TwitterUtils.StatusToJsonFn()));

 DataflowAssert.thatSingleton(strings).isEqualTo(...);
4

2 回答 2

2

查看 DoFnTester，它允许您通过提供输入数据并断言结果来测试 DoFn：https ://cloud.google.com/dataflow/pipelines/testing-your-pipeline

于 2016-05-25T04:02:19.447 回答
1

一种方法是创建一个不可变的包装类：

class StatusWrapper implements Serializable {
    private Status status;
    private String mockText;

    // Constructor for real usage
    StatusWrapper(Status status) { this.status = status; }

    // Constructor for mocks
    StatusWrapper() {}

    StatusWrapper withMockText(String text) {
        this.mockText = text; return this;
    }

    Status toStatus() {
        if (status != null) {
            return status;
        }
        Status status = mock(Status.class);
        when(status.getText()).thenReturn(mockText);
        return status;
    }
}

Pipeline p = TestPipeline.create();
PCollection<String> strings = p
    .apply(Create.of(new StatusWrapper().withMockText("bar")))
    .apply(MapElements.via(new TwitterUtils.StatusToJsonFn()));
DataflowAssert.thatSingleton(strings).isEqualTo(...);

StatusToJsonFn将需要 takeStatusWrapper作为参数而不是Status, 并调用toStatus()它 - 因为如果它采取Status，它将遇到同样的问题。

在真正的管道中，可以这样使用：

PCollection<Status> realStatuses = ...;
PCollection<String> strings = realStatuses
    .apply(MapElements.via(StatusWrapper::new))
    .apply(MapElements.via(new TwitterUtils.StatusToJsonFn()));
于 2016-05-21T01:24:35.303 回答