Apache Beam, имитация внешних клиентов, инициализированных в методе @Setup Lifecycle DoFn. ⇐ JAVA

[Anonymous]

Я создал следующий класс 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