Я создал следующий класс DoFn, но я не уверен, как имитировать APIClient, поскольку создание экземпляра JsonToGenericRecordMapper в Junit с имитируемым объектом APICall переопределяется методом @setup, который создает фактический экземпляр APICall. ?
public class JsonToGenericRecordMapper extends DoFn {
private APICall apiCall;
@setup
public void setup ()
{
api = new APICall();
}
@ProcessElement
public void processElement(ProcessContext processContext) {
String enrichedItemCost = processContext.element();
Schema schema = CommonUtils.schema;
GenericRecord genericRecord = new GenericData.Record(schema);
genericRecord.put(CommonUtils.SCHEMA_FIELD_KEY, enrichedItemCost);
processContext.output(genericRecord);
}
}
TestPipeline testPipeline = TestPipeline.create();
APICAll apiCall = Mockito.mock(APICall.class);
dataPCollection.apply(new JsonToGenericRecordMapper(apiCall));
Подробнее здесь: https://stackoverflow.com/questions/677 ... thod-of-do
