Skip to content

Commit

Permalink
Explicit Description of Simulation Dispatcher Tests
Browse files Browse the repository at this point in the history
  • Loading branch information
@AvocadoMoon
AvocadoMoon committed Sep 10, 2024
1 parent b015f24 commit da869c5
Showing 1 changed file with 29 additions and 12 deletions.
41 changes: 29 additions & 12 deletions ...l-server/src/test/java/cbit/vcell/message/server/dispatcher/SimulationDispatcherTest.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@
public class SimulationDispatcherTest {
public static ExtendedLogger lg = LoggerContext.getContext().getLogger(SimulationDispatcher.class);
private final static User testUser = DispatcherTestUtils.alice;
private MockSimulationDB mockSimulationDB = new MockSimulationDB();
private final MockSimulationDB mockSimulationDB = new MockSimulationDB();
private final MockMessagingService mockMessagingServiceInternal = new MockMessagingService();
private final MockMessagingService mockMessagingServiceSim = new MockMessagingService();
private final MockHtcProxy mockHtcProxy = new MockHtcProxy(null, "htcUser", mockSimulationDB);
Expand Down Expand Up @@ -82,25 +82,37 @@ public void onStopRequestTest() throws DataAccessException, SQLException {


//###################### Test Dispatcher Thread ###########################

@Test
public void dispatcherThreadTest() throws SQLException, DataAccessException, InterruptedException, PropertyVetoException, MathException, ExpressionBindingException {
public void dispatcherThreadFailsJobsWithNoSimulationReference() throws SQLException, DataAccessException, InterruptedException {
DispatcherTestUtils.insertOrUpdateStatus(mockSimulationDB, SimulationJobStatus.SchedulerStatus.WAITING);
SimulationDispatcher simulationDispatcher = SimulationDispatcher.simulationDispatcherCreator(mockSimulationDB, mockMessagingServiceInternal,
mockMessagingServiceSim, mockHtcProxy, true);
SimulationDispatcher.DispatchThread thread = simulationDispatcher.dispatchThread;
synchronized (thread.dispatcherNotifyObject){
thread.dispatcherNotifyObject.notify();
}

// Check that the simulation is in waiting, for the dispatcher hasn't consumed it's request yet
SimulationJobStatus jobStatus = mockSimulationDB.getLatestSimulationJobStatus(DispatcherTestUtils.simKey, 0);
Assertions.assertTrue(jobStatus.getSchedulerStatus().isWaiting(), "Still waiting.");

synchronized (thread.finishListener){
thread.finishListener.wait();
}

// Makes sure that requests that have no simulation reference within the DB are failed
jobStatus = mockSimulationDB.getLatestSimulationJobStatus(DispatcherTestUtils.simKey, 0);
Assertions.assertTrue(jobStatus.getSchedulerStatus().isFailed(), "Simulation gets aborted since theres no simulation in DB.");

}

@Test
public void dispatcherThreadDispatchesWaitingJobsWithSimulationsIn() throws SQLException, DataAccessException, InterruptedException, PropertyVetoException, MathException, ExpressionBindingException {
SimulationDispatcher simulationDispatcher = SimulationDispatcher.simulationDispatcherCreator(mockSimulationDB, mockMessagingServiceInternal,
mockMessagingServiceSim, mockHtcProxy, true);
SimulationDispatcher.DispatchThread thread = simulationDispatcher.dispatchThread;
// Create and insert simulation. Then ensure that this simulation has it's job status changed to dispatched
Simulation mockSimulation = DispatcherTestUtils.createMockSimulation(20, 20, 20);
mockSimulationDB.insertSimulation(DispatcherTestUtils.alice, mockSimulation);
DispatcherTestUtils.insertOrUpdateStatus(mockSimulation.getKey(), DispatcherTestUtils.jobIndex, DispatcherTestUtils.taskID, DispatcherTestUtils.alice,
Expand All @@ -112,14 +124,15 @@ public void dispatcherThreadTest() throws SQLException, DataAccessException, Int
thread.finishListener.wait();
}

jobStatus = mockSimulationDB.getLatestSimulationJobStatus(mockSimulation.getKey(), 0);
SimulationJobStatus jobStatus = mockSimulationDB.getLatestSimulationJobStatus(mockSimulation.getKey(), 0);
Assertions.assertTrue(jobStatus.getSchedulerStatus().isDispatched(), "Dispatches");
}



//###################### Test Simulation Monitor ##########################

// Rig the mock simulation DB to return a simulation job status that features some misbehavior of what's expected,
// which prompts for removal by the zombie killer
@Test
public void zombieKillerTest() throws SQLException, DataAccessException, InterruptedException, IOException {
SimulationDispatcher.INITIAL_ZOMBIE_DELAY = 10;
Expand All @@ -146,7 +159,7 @@ public void zombieKillerTest() throws SQLException, DataAccessException, Interru
}

@Test
public void queueFlusherTest() throws SQLException, DataAccessException, InterruptedException {
public void queueFlusherKillIdleJobs() throws SQLException, DataAccessException, InterruptedException {
SimulationDispatcher simulationDispatcher = SimulationDispatcher.simulationDispatcherCreator(mockSimulationDB, mockMessagingServiceInternal,
mockMessagingServiceSim, mockHtcProxy, false);
DispatcherTestUtils.insertOrUpdateStatus(mockSimulationDB);
Expand Down Expand Up @@ -177,16 +190,20 @@ public void queueFlusherTest() throws SQLException, DataAccessException, Interru
Assertions.assertTrue(mockHtcProxy.jobsKilledUnsafely.contains(status.getSimulationExecutionStatus().getHtcJobID()));
Assertions.assertTrue(logOutPut.toString().contains(SimulationDispatcher.SimulationMonitor.QueueFlusher.timeOutFailure));

// reset for next test
simulationDispatcher.simDispatcherEngine.resetTimeStamps();
mockHtcProxy.jobsKilledUnsafely.clear();
mockSimulationDB.resetDataBase();
}

@Test
public void queueFlusherKillsNoLongerReferencedSims() throws SQLException, DataAccessException, InterruptedException {
SimulationDispatcher simulationDispatcher = SimulationDispatcher.simulationDispatcherCreator(mockSimulationDB, mockMessagingServiceInternal,
mockMessagingServiceSim, mockHtcProxy, false);
SimulationDispatcher.SimulationMonitor simMonitor = simulationDispatcher.simMonitor;
SimulationDispatcher.SimulationMonitor.QueueFlusher queueFlusher = simMonitor.initialQueueFlusher;

mockSimulationDB.insertUnreferencedSimKey(DispatcherTestUtils.simKey);
DispatcherTestUtils.insertOrUpdateStatus(mockSimulationDB);
queueThread = new Thread(queueFlusher);
Thread queueThread = new Thread(queueFlusher);
queueThread.start();
retries = 0;
int retries = 0;
while (queueThread.getState() != Thread.State.TIMED_WAITING){
if (retries == 10){
break;
Expand All @@ -200,7 +217,7 @@ public void queueFlusherTest() throws SQLException, DataAccessException, Interru
synchronized (queueFlusher.finishListener){
queueFlusher.finishListener.wait();
}
status = mockSimulationDB.getLatestSimulationJobStatus(DispatcherTestUtils.simKey, DispatcherTestUtils.jobIndex);
SimulationJobStatus status = mockSimulationDB.getLatestSimulationJobStatus(DispatcherTestUtils.simKey, DispatcherTestUtils.jobIndex);
Assertions.assertTrue(status.getSchedulerStatus().isFailed());
Assertions.assertTrue(mockHtcProxy.jobsKilledUnsafely.contains(status.getSimulationExecutionStatus().getHtcJobID()));
Assertions.assertTrue(logOutPut.toString().contains(SimulationDispatcher.SimulationMonitor.QueueFlusher.unreferencedFailure));
Expand Down

0 comments on commit da869c5

Please sign in to comment.