@Test public void testBuilderRunWithContextBeingPassed() throws Exception { DataFlowInstance dataFlowInstance = new DataFlowInstance(); dataFlowInstance.setId("testflow"); dataFlowInstance.setDataFlow(dataFlow); { DataBuilderContext context = new DataBuilderContext(); context.saveContextData(KEY, VALUE); DataDelta dataDelta = new DataDelta( Lists.newArrayList( new TestDataA("Hello"), new TestDataB("World"), new TestDataD("this"), new TestDataG("Hmmm"))); DataExecutionResponse response = executor.run(dataFlowInstance, dataDelta); Assert.assertEquals(3, response.getResponses().size()); Assert.assertTrue(response.getResponses().containsKey("C")); Assert.assertTrue(response.getResponses().containsKey("E")); Assert.assertTrue(response.getResponses().containsKey("F")); } }
@Override public void beforeExecute( DataBuilderContext builderContext, DataFlowInstance dataFlowInstance, DataBuilderMeta builderToBeApplied, DataDelta dataDelta, Map<String, Data> prevResponses) throws Exception { Assert.assertNotNull(builderContext); Assert.assertEquals(builderContext.getContextData(KEY, String.class), VALUE); }
@Override public void afterException( DataBuilderContext builderContext, DataFlowInstance dataFlowInstance, DataBuilderMeta builderToBeApplied, DataDelta dataDelta, Map<String, Data> prevResponses, Throwable frameworkException) throws Exception { Assert.assertNotNull(builderContext); Assert.assertEquals(builderContext.getContextData(KEY, String.class), VALUE); }