@Test
public void testCascadingShutdown() {
ViewProcessorTestEnvironment env = new ViewProcessorTestEnvironment();
env.init();
ViewProcessorImpl vp = env.getViewProcessor();
vp.start();
ViewClient client1 = vp.createViewClient(ViewProcessorTestEnvironment.TEST_USER);
client1.attachToViewProcess(
env.getViewDefinition().getUniqueId(), ExecutionOptions.infinite(MarketData.live()));
ViewClient client2 = vp.createViewClient(ViewProcessorTestEnvironment.TEST_USER);
client2.attachToViewProcess(
env.getViewDefinition().getUniqueId(), ExecutionOptions.infinite(MarketData.live()));
ViewProcessImpl view = env.getViewProcess(vp, client1.getUniqueId());
vp.stop();
assertFalse(vp.isRunning());
assertFalse(view.isRunning());
assertTrue(view.getState() == ViewProcessState.TERMINATED);
assertFalse(client1.isAttached());
assertFalse(client2.isAttached());
client1.shutdown();
client2.shutdown();
}
@Test
public void testChangeOfListeners() throws InterruptedException {
ViewProcessorTestEnvironment env = new ViewProcessorTestEnvironment();
SynchronousInMemoryLKVSnapshotProvider marketDataProvider =
new SynchronousInMemoryLKVSnapshotProvider();
marketDataProvider.addValue(ViewProcessorTestEnvironment.getPrimitive1(), 0);
marketDataProvider.addValue(ViewProcessorTestEnvironment.getPrimitive2(), 0);
env.setMarketDataProvider(marketDataProvider);
env.init();
ViewProcessorImpl vp = env.getViewProcessor();
vp.start();
ViewClient client = vp.createViewClient(ViewProcessorTestEnvironment.TEST_USER);
client.setFragmentResultMode(ViewResultMode.FULL_ONLY);
TestViewResultListener resultListener1 = new TestViewResultListener();
client.setResultListener(resultListener1);
// Start live computation and collect the initial result
marketDataProvider.addValue(ViewProcessorTestEnvironment.getPrimitive1(), 2);
client.attachToViewProcess(
env.getViewDefinition().getUniqueId(), ExecutionOptions.infinite(MarketData.live()));
ViewProcessImpl viewProcess = env.getViewProcess(vp, client.getUniqueId());
assertEquals(ViewProcessState.RUNNING, viewProcess.getState());
ViewComputationJob recalcJob = env.getCurrentComputationJob(viewProcess);
resultListener1.assertViewDefinitionCompiled(TIMEOUT);
resultListener1.assertCycleInitiated(TIMEOUT);
resultListener1.assertCycleFragmentCompleted(TIMEOUT);
resultListener1.assertCycleCompleted(TIMEOUT);
assertEquals(0, resultListener1.getQueueSize());
// Push through a second result
marketDataProvider.addValue(ViewProcessorTestEnvironment.getPrimitive1(), 3);
recalcJob.marketDataChanged();
resultListener1.assertCycleInitiated(TIMEOUT);
resultListener1.assertCycleFragmentCompleted(TIMEOUT);
resultListener1.assertCycleCompleted(TIMEOUT);
assertEquals(0, resultListener1.getQueueSize());
// Change listener
TestViewResultListener resultListener2 = new TestViewResultListener();
client.setResultListener(resultListener2);
// Push through a result which should arrive at the new listeners
recalcJob.marketDataChanged();
resultListener2.assertCycleInitiated(TIMEOUT);
resultListener2.assertCycleFragmentCompleted(TIMEOUT);
resultListener2.assertCycleCompleted(TIMEOUT);
assertEquals(0, resultListener1.getQueueSize());
assertEquals(0, resultListener2.getQueueSize());
client.setResultListener(null);
client.shutdown();
assertEquals(ViewProcessState.TERMINATED, viewProcess.getState());
vp.stop();
}
@Test
public void testSingleViewMultipleClients() {
ViewProcessorTestEnvironment env = new ViewProcessorTestEnvironment();
env.init();
ViewProcessorImpl vp = env.getViewProcessor();
vp.start();
ViewClient client1 = vp.createViewClient(ViewProcessorTestEnvironment.TEST_USER);
assertNotNull(client1.getUniqueId());
client1.attachToViewProcess(
env.getViewDefinition().getUniqueId(), ExecutionOptions.infinite(MarketData.live()));
ViewProcessImpl client1Process = env.getViewProcess(vp, client1.getUniqueId());
assertTrue(client1Process.getState() == ViewProcessState.RUNNING);
ViewClient client2 = vp.createViewClient(ViewProcessorTestEnvironment.TEST_USER);
assertNotNull(client2.getUniqueId());
assertFalse(client1.getUniqueId().equals(client2.getUniqueId()));
client2.attachToViewProcess(
env.getViewDefinition().getUniqueId(), ExecutionOptions.infinite(MarketData.live()));
ViewProcessImpl client2Process = env.getViewProcess(vp, client2.getUniqueId());
assertEquals(client1Process, client2Process);
assertTrue(client2Process.getState() == ViewProcessState.RUNNING);
client1.detachFromViewProcess();
assertTrue(client2Process.getState() == ViewProcessState.RUNNING);
client2.detachFromViewProcess();
assertTrue(client2Process.getState() == ViewProcessState.TERMINATED);
client1.shutdown();
client2.shutdown();
}
@Test
public void testComputationResultsFlow() throws InterruptedException {
ViewProcessorTestEnvironment env = new ViewProcessorTestEnvironment();
SynchronousInMemoryLKVSnapshotProvider marketDataProvider =
new SynchronousInMemoryLKVSnapshotProvider();
marketDataProvider.addValue(ViewProcessorTestEnvironment.getPrimitive1(), (byte) 0);
marketDataProvider.addValue(ViewProcessorTestEnvironment.getPrimitive2(), (byte) 0);
env.setMarketDataProvider(marketDataProvider);
env.init();
ViewProcessorImpl vp = env.getViewProcessor();
vp.start();
ViewClient client = vp.createViewClient(ViewProcessorTestEnvironment.TEST_USER);
client.setFragmentResultMode(ViewResultMode.FULL_ONLY);
TestViewResultListener resultListener = new TestViewResultListener();
client.setResultListener(resultListener);
// Client not attached - should not have been listening to anything that might have been going
// on
assertEquals(0, resultListener.getQueueSize());
marketDataProvider.addValue(ViewProcessorTestEnvironment.getPrimitive1(), (byte) 1);
marketDataProvider.addValue(ViewProcessorTestEnvironment.getPrimitive2(), (byte) 2);
assertEquals(0, resultListener.getQueueSize());
client.attachToViewProcess(
env.getViewDefinition().getUniqueId(), ExecutionOptions.infinite(MarketData.live()));
ViewProcessImpl viewProcess = env.getViewProcess(vp, client.getUniqueId());
assertTrue(viewProcess.getState() == ViewProcessState.RUNNING);
resultListener.assertViewDefinitionCompiled(TIMEOUT);
resultListener.assertCycleInitiated(TIMEOUT);
ViewResultModel result1Fragment =
resultListener.getCycleFragmentCompleted(TIMEOUT).getFullFragment();
assertNotNull(result1Fragment);
ViewComputationResultModel result1 = resultListener.getCycleCompleted(TIMEOUT).getFullResult();
assertNotNull(result1);
Map<ValueRequirement, Object> expected = new HashMap<ValueRequirement, Object>();
expected.put(ViewProcessorTestEnvironment.getPrimitive1(), (byte) 1);
expected.put(ViewProcessorTestEnvironment.getPrimitive2(), (byte) 2);
assertComputationResult(expected, env.getCalculationResult(result1));
assertComputationResult(expected, env.getCalculationResult(result1Fragment));
assertTrue(client.isResultAvailable());
assertEquals(result1, client.getLatestResult());
client.pause();
marketDataProvider.addValue(ViewProcessorTestEnvironment.getPrimitive1(), (byte) 3);
marketDataProvider.addValue(ViewProcessorTestEnvironment.getPrimitive2(), (byte) 4);
env.getCurrentComputationJob(viewProcess)
.marketDataChanged(); // Need to get it to perform another cycle
// Should have been merging results received in the meantime
client.resume();
resultListener.assertCycleInitiated(TIMEOUT);
resultListener.assertCycleFragmentCompleted(TIMEOUT);
ViewComputationResultModel result2 = resultListener.getCycleCompleted(TIMEOUT).getFullResult();
expected = new HashMap<ValueRequirement, Object>();
expected.put(ViewProcessorTestEnvironment.getPrimitive1(), (byte) 3);
expected.put(ViewProcessorTestEnvironment.getPrimitive2(), (byte) 4);
assertComputationResult(expected, env.getCalculationResult(result2));
}