def test_get_object_function_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=1,
oc_positiontype=0)
mock_database.GamFunction.create(of_name="test4")
self.assertEqual(db_func.get_object_function(1), "test4")
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_object_function_WHEN_not_present_THEN_returns_none(self):
with mock_database.Database():
mock_database.GamObject.create(ob_name="test1", ob_objecttype=1)
self.assertIsNone(db_func.get_object_function(2))
def test_get_object_function_WHEN_no_connection_THEN_returns_none(self):
self.assertIsNone(db_func.get_object_function(1))