def setUp(self): locator_mock = None self.cfg_manager = ConfigurationManager.ConfigurationManager() self.cfg_manager.append_module(configuration_module) self.coordinator = WrappedSqlCoordinator(locator_mock, self.cfg_manager, ConfirmerClass=ConfirmerMock) self.coordinator._clean() self.coordinator.add_experiment_instance_id( "lab1:inst@machine", ExperimentInstanceId('inst1', 'exp1', 'cat1'), Resource("res_type", "res_inst1")) self.coordinator.add_experiment_instance_id( "lab2:inst@machine", ExperimentInstanceId('inst2', 'exp2', 'cat1'), Resource("res_type", "res_inst2")) self.reservations_manager = self.coordinator.reservations_manager
class ReservationsManagerTestCase(unittest.TestCase): def setUp(self): locator_mock = None self.cfg_manager = ConfigurationManager.ConfigurationManager() self.cfg_manager.append_module(configuration_module) self.coordinator = WrappedSqlCoordinator(locator_mock, self.cfg_manager, ConfirmerClass=ConfirmerMock) self.coordinator._clean() self.coordinator.add_experiment_instance_id( "lab1:inst@machine", ExperimentInstanceId('inst1', 'exp1', 'cat1'), Resource("res_type", "res_inst1")) self.coordinator.add_experiment_instance_id( "lab2:inst@machine", ExperimentInstanceId('inst2', 'exp2', 'cat1'), Resource("res_type", "res_inst2")) self.reservations_manager = self.coordinator.reservations_manager def tearDown(self): self.coordinator.stop() def test_list_sessions_not_existing(self): exp_id = ExperimentId("exp.that.doesnt.exist", "cat1") self.assertRaises(CoordExc.ExperimentNotFoundError, self.reservations_manager.list_sessions, exp_id) def test_list_sessions(self): exp_id1 = ExperimentId("exp1", "cat1") exp_id2 = ExperimentId("exp2", "cat1") sessions = self.reservations_manager.list_sessions(exp_id1) self.assertEquals(0, len(sessions)) reservation1 = self.reservations_manager.create( exp_id1, "{}", REQUEST_INFO) reservation2 = self.reservations_manager.create( exp_id1, "{}", REQUEST_INFO) self.reservations_manager.create(exp_id2, "{}", REQUEST_INFO) sessions = self.reservations_manager.list_sessions(exp_id1) self.assertEquals(2, len(sessions)) self.assertTrue(reservation1 in sessions) self.assertTrue(reservation2 in sessions)
def setUp(self): locator_mock = None self.cfg_manager = ConfigurationManager.ConfigurationManager() self.cfg_manager.append_module(configuration_module) self.coordinator = WrappedSqlCoordinator(locator_mock, self.cfg_manager, ConfirmerClass = ConfirmerMock) self.coordinator._clean() self.coordinator.add_experiment_instance_id("lab1:inst@machine", ExperimentInstanceId('inst1', 'exp1','cat1'), Resource("res_type", "res_inst1")) self.coordinator.add_experiment_instance_id("lab2:inst@machine", ExperimentInstanceId('inst2', 'exp2','cat1'), Resource("res_type", "res_inst2")) self.reservations_manager = self.coordinator.reservations_manager
class ReservationsManagerTestCase(unittest.TestCase): def setUp(self): locator_mock = None self.cfg_manager = ConfigurationManager.ConfigurationManager() self.cfg_manager.append_module(configuration_module) self.coordinator = WrappedSqlCoordinator(locator_mock, self.cfg_manager, ConfirmerClass = ConfirmerMock) self.coordinator._clean() self.coordinator.add_experiment_instance_id("lab1:inst@machine", ExperimentInstanceId('inst1', 'exp1','cat1'), Resource("res_type", "res_inst1")) self.coordinator.add_experiment_instance_id("lab2:inst@machine", ExperimentInstanceId('inst2', 'exp2','cat1'), Resource("res_type", "res_inst2")) self.reservations_manager = self.coordinator.reservations_manager def tearDown(self): self.coordinator.stop() def test_list_sessions_not_existing(self): exp_id = ExperimentId("exp.that.doesnt.exist","cat1") self.assertRaises( CoordExc.ExperimentNotFoundError, self.reservations_manager.list_sessions, exp_id ) def test_list_sessions(self): exp_id1 = ExperimentId("exp1","cat1") exp_id2 = ExperimentId("exp2","cat1") sessions = self.reservations_manager.list_sessions(exp_id1) self.assertEquals(0, len(sessions)) reservation1 = self.reservations_manager.create(exp_id1, "{}", REQUEST_INFO) reservation2 = self.reservations_manager.create(exp_id1, "{}", REQUEST_INFO) self.reservations_manager.create(exp_id2, "{}", REQUEST_INFO) sessions = self.reservations_manager.list_sessions(exp_id1) self.assertEquals(2, len(sessions)) self.assertTrue(reservation1 in sessions) self.assertTrue(reservation2 in sessions)