def test_get_type_id_WHEN_multi_same_THEN_returns_first(self): with mock_database.Database(): obj = mock_database.GamObjecttype.create(ot_name="test1", ot_objectclass=0) mock_database.GamObjecttype.create(ot_name="test1", ot_objectclass=0) self.assertEqual(db_func.get_type_id("test1"), obj.ot_id)
def test_get_all_object_names_WHEN_multi_present_THEN_returns_correct( self): with mock_database.Database(): mock_database.GamObject.create(ob_name="test1", ob_objecttype=1) mock_database.GamObject.create(ob_name="test2", ob_objecttype=1) self.assertListEqual(db_func.get_all_object_names(), ["test1", "test2"])
def test_get_type_id_WHEN_multi_present_THEN_returns_correct(self): with mock_database.Database(): mock_database.GamObjecttype.create(ot_name="test1", ot_objectclass=0) obj = mock_database.GamObjecttype.create(ot_name="test2", ot_objectclass=0) self.assertEqual(db_func.get_type_id("test2"), obj.ot_id)
def test_get_object_display_group_WHEN_present_THEN_returns_it(self): with mock_database.Database(): mock_database.GamObject.create(ob_name="test1", ob_objecttype=1, ob_displaygroup=1) mock_database.GamDisplaygroup.create(dg_name="test2") self.assertEqual(db_func.get_object_display_group(1), "test2")
def test_get_object_type_WHEN_multi_present_THEN_returns_correct(self): with mock_database.Database(): mock_database.GamObject.create(ob_name="test1", ob_objecttype=1) mock_database.GamObject.create(ob_name="test2", ob_objecttype=2) mock_database.GamObjecttype.create(ot_name="test3", ot_objectclass=0) self.assertEqual(db_func.get_object_type(1), "test3")
def test_get_measurement_types_WHEN_class_not_present_THEN_returns_none( self): with mock_database.Database(): mock_database.GamObjectclass.create(oc_name="test1", oc_function=0, oc_positiontype=0) self.assertIsNone(db_func.get_measurement_types(2))
def test_add_object_WHEN_connection_AND_comment_THEN_has_comment(self): with mock_database.Database(): ob_id = db_func.add_object("test1", 0, 0, "test_comment") self.assertEqual(ob_id, 1) self.assertIsNotNone(db_func.get_object(ob_id)) self.assertIsNotNone(db_func.get_object_id("test1")) self.assertEqual(db_func.get_object(1).ob_comment, "test_comment")
def test_GIVEN_mock_database_AND_value_present_WHEN_get_module_object_THEN_return_value(self): with mock_database.Database(): GamObject.create(ob_name="test", ob_objecttype=1) module = GamObject.create(ob_name="test", ob_objecttype=0) GamObjectrelation.create(or_object=1, or_object_id_assigned=2, or_date_assignment="") self.assertEqual(_get_module_object(1, 0), module)
def test_get_all_display_names_WHEN_multi_present_THEN_returns_correct( self): with mock_database.Database(): mock_database.GamDisplaygroup.create(dg_name="test1") mock_database.GamDisplaygroup.create(dg_name="test2") self.assertListEqual(db_func.get_all_display_names(), ["test1", "test2"])
def test_GIVEN_mock_database_AND_value_present_WHEN_get_module_object_AND_has_removal_date_THEN_return_none(self): with mock_database.Database(): GamObject.create(ob_name="test", ob_objecttype=1) GamObject.create(ob_name="test", ob_objecttype=0) GamObjectrelation.create(or_object=1, or_object_id_assigned=2, or_date_assignment="", or_date_removal="NOTNULL") self.assertIsNone(_get_module_object(1, 0))
def test_get_measurement_types_WHEN_no_types_THEN_returns_list_of_none( self): with mock_database.Database(): mock_database.GamObjectclass.create(oc_name="test1", oc_function=0, oc_positiontype=0) self.assertListEqual(db_func.get_measurement_types(1), [None, None, None, None, None])
def test_get_class_id_WHEN_multi_present_THEN_returns_correct(self): with mock_database.Database(): mock_database.GamObjecttype.create(ot_name="test3", ot_objectclass=1) test_class = mock_database.GamObjectclass.create(oc_name="test4", oc_function=0, oc_positiontype=0) self.assertEqual(db_func.get_class_id(1), test_class.oc_id)
def test_get_object_display_group_WHEN_present_but_unnamed_THEN_returns_none( self): with mock_database.Database(): mock_database.GamObject.create(ob_name="test1", ob_objecttype=1, ob_displaygroup=1) mock_database.GamDisplaygroup.create() self.assertIsNone(db_func.get_object_display_group(1))
def test_get_object_class_WHEN_present_THEN_returns_it(self): with mock_database.Database(): mock_database.GamObject.create(ob_name="test1", ob_objecttype=1) mock_database.GamObjecttype.create(ot_name="test2", ot_objectclass=1) mock_database.GamObjectclass.create(oc_name="test3", oc_function=0, oc_positiontype=0) self.assertEqual(db_func.get_object_class(1), "test3")
def test_get_max_object_id_WHEN_multi_present_THEN_returns_highest(self): with mock_database.Database(): mock_database.GamObject.create(ob_name="test", ob_objecttype=1) mock_database.GamObject.create(ob_name="test", ob_objecttype=1) mock_database.GamObject.create(ob_name="test", ob_objecttype=1) mock_database.GamObject.create(ob_name="test", ob_objecttype=1) obj = mock_database.GamObject.create(ob_name="test", ob_objecttype=1) self.assertEqual(db_func.get_max_object_id(), obj.ob_id)
def test_add_object_WHEN_connection_AND_valid_display_group_THEN_object_added_AND_has_display_group( self): with mock_database.Database(): mock_database.GamDisplaygroup.create(dg_name="test2") ob_id = db_func.add_object("test1", 0, 1) self.assertEqual(ob_id, 1) self.assertIsNotNone(db_func.get_object(ob_id)) self.assertIsNotNone(db_func.get_object_id("test1")) self.assertEqual(db_func.get_object_display_group(ob_id), "test2")
def test_add_object_WHEN_connection_AND_add_multi_THEN_returns_id(self): with mock_database.Database(): ob_id = db_func.add_object("test1", 0) self.assertEqual(ob_id, 1) self.assertIsNotNone(db_func.get_object(ob_id)) self.assertIsNotNone(db_func.get_object_id("test1")) ob_id = db_func.add_object("test2", 0) self.assertEqual(ob_id, 2) self.assertIsNotNone(db_func.get_object(ob_id)) self.assertIsNotNone(db_func.get_object_id("test2"))
def test_add_relation_WHEN_connection_AND_valid_objects_THEN_adds_relation( self): with mock_database.Database(): mock_database.GamObjecttype.create(ot_name="test1", ot_objectclass=0) mock_database.GamObjecttype.create(ot_name="test2", ot_objectclass=0) db_func.add_relation(1, 2) self.assertIsNotNone( mock_database.GamObjectrelation.get_or_none(1))
def test_add_object_WHEN_connection_AND_valid_type_THEN_object_added_AND_has_type( self): with mock_database.Database(): mock_database.GamObjecttype.create(ot_name="test2", ot_objectclass=0) ob_id = db_func.add_object("test1", 1) self.assertEqual(ob_id, 1) self.assertIsNotNone(db_func.get_object(ob_id)) self.assertIsNotNone(db_func.get_object_id("test1")) self.assertEqual(db_func.get_object_type(ob_id), "test2")
def test_get_object_function_WHEN_multi_present_THEN_returns_correct(self): with mock_database.Database(): mock_database.GamObject.create(ob_name="test1", ob_objecttype=1) mock_database.GamObject.create(ob_name="test2", ob_objecttype=2) mock_database.GamObjecttype.create(ot_name="test3", ot_objectclass=1) mock_database.GamObjectclass.create(oc_name="test4", oc_function=1, oc_positiontype=0) mock_database.GamFunction.create(of_name="test5") self.assertEqual(db_func.get_object_function(1), "test5")
def test_get_object_function_WHEN_present_but_unnamed_THEN_returns_none( self): with mock_database.Database(): mock_database.GamObject.create(ob_name="test1", ob_objecttype=1) mock_database.GamObjecttype.create(ot_name="test2", ot_objectclass=1) mock_database.GamObjectclass.create(oc_name="test3", oc_function=1, oc_positiontype=0) mock_database.GamFunction.create() self.assertIsNone(db_func.get_object_function(1))
def test_get_measurement_types_WHEN_present_THEN_returns_it(self): with mock_database.Database(): mock_database.GamObjectclass.create(oc_name="test1", oc_function=0, oc_positiontype=0, oc_measuretype1="test2", oc_measuretype2="test3", oc_measuretype3="test4", oc_measuretype4="test5", oc_measuretype5="test6") self.assertListEqual(db_func.get_measurement_types(1), ['test2', 'test3', 'test4', 'test5', 'test6'])
def test_create_module_if_required_WHEN_connection_AND_gcm_required_THEN_gcm( self): with mock_database.Database(): mock_database.GamObjecttype.create(ot_name="test3", ot_id=GCM, ot_objectclass=0) db_func.create_module_if_required(1, "test1", "test2", GAS_COUNTER) self.assertIsNotNone( mock_database.GamObjectrelation.get_or_none(1)) obj = mock_database.GamObject.get_or_none(1) self.assertEqual("GCM \"test1\" (ID: 1)", obj.ob_name) self.assertEqual("Gas Counter Module for test2 \"test1\" (ID: 1)", obj.ob_comment) self.assertEqual(GCM, obj.ob_objecttype.ot_id)
def test_create_module_if_required_WHEN_connection_AND_sld_required_for_cryostat_THEN_sld( self): with mock_database.Database(): mock_database.GamObjecttype.create(ot_name="test3", ot_id=SLD, ot_objectclass=0) db_func.create_module_if_required(1, "test1", "test2", CRYOSTAT) self.assertIsNotNone( mock_database.GamObjectrelation.get_or_none(1)) obj = mock_database.GamObject.get_or_none(1) self.assertEqual("SLD \"test1\" (ID: 1)", obj.ob_name) self.assertEqual( "Software Level Device for test2 \"test1\" (ID: 1)", obj.ob_comment) self.assertEqual(SLD, obj.ob_objecttype.ot_id)
def test_GIVEN_mock_database_AND_value_not_present_WHEN_get_module_object_THEN_return_none(self): with mock_database.Database(): GamObject.create(ob_name="test", ob_objecttype=1) self.assertIsNone(_get_module_object(1, 0))
def test_get_object_GIVEN_object_THEN_returns_it(self): with mock_database.Database(): obj = mock_database.GamObject.create(ob_name="test", ob_objecttype=1) self.assertEqual(db_func.get_object(1), obj)
def test_get_object_GIVEN_no_object_THEN_returns_none(self): with mock_database.Database(): self.assertIsNone(db_func.get_object(1))
def test_get_max_object_id_WHEN_not_present_THEN_returns_zero(self): with mock_database.Database(): self.assertTrue(db_func.db_connected()) self.assertEqual(db_func.get_max_object_id(), 0)
def test_get_object_id_WHEN_multiple_present_THEN_returns_lowest_id(self): with mock_database.Database(): obj = mock_database.GamObject.create(ob_name="test", ob_objecttype=1) mock_database.GamObject.create(ob_name="test", ob_objecttype=1) self.assertEqual(db_func.get_object_id("test"), obj.ob_id)
def test_create_module_if_required_WHEN_connection_AND_module_not_required_THEN_no_module( self): with mock_database.Database(): db_func.create_module_if_required(1, "test1", "test2", 1) self.assertIsNone(mock_database.GamObjectrelation.get_or_none(1)) self.assertIsNone(db_func.get_object_id("SLD \"test1\" (ID: 1)"))