def test_remove_event_type(self):
event = Event(1940000001)
event_type = EventType(EventTypeIdentifier(1, 5), "Test")
self.event_service.remove_event_type(event, event_type)
self.event_type_dao.unlink_event_type_from_event_id.assert_called_once_with(
1940000001, event_type)
def test_add_event_type(self):
event = Event(1940000001)
event_type = EventType(EventTypeIdentifier(1, 5), "Test")
self.event_service.add_event_type(event, event_type)
self.event_type_dao.join_event_type_to_event_id.assert_called_once_with(
1940000001, event_type)
def test_get_event_types(self):
event = Event(1940000001)
event_type = EventType(EventTypeIdentifier(5, 1), "Test")
self.event_type_dao.get_event_types_for_event_id.return_value = [
event_type
]
result = self.event_service.get_event_types(event)
self.assertEqual(1, len(result))
self.assertEqual(event_type, result[0])
self.event_type_dao.get_event_types_for_event_id.assert_called_once_with(
1940000001)
def get_event_type_tree(self):
event_types = (
EventType(EventTypeIdentifier(0, 0), "root"),
EventType(EventTypeIdentifier(1, 0), "parent1"),
EventType(EventTypeIdentifier(1, 1), "child11"),
EventType(EventTypeIdentifier(1, 2), "child12"),
EventType(EventTypeIdentifier(2, 0), "parent2"),
EventType(EventTypeIdentifier(2, 1), "child21"),
EventType(EventTypeIdentifier(2, 2), "child22"),
)
return Tree(event_types)
def test_get_event_type_tree(self):
self.event_type_dao.find_all.return_value = [
EventType(EventTypeIdentifier(1, 0), "one_zero"),
EventType(EventTypeIdentifier(1, 1), "one_one"),
EventType(EventTypeIdentifier(2, 0), "two_zero"),
EventType(EventTypeIdentifier(2, 1), "two_one")
]
tree = self.event_service.get_event_type_tree()
self.assertFalse(tree is None)
self.assertFalse(tree.root_node is None)
self.assertEqual(2, len(tree.root_node.children))
self.assertEqual("one_zero",
tree.root_node.children[0].entity.description)
self.assertEqual("two_zero",
tree.root_node.children[1].entity.description)
self.assertEqual(1, len(tree.root_node.children[0].children))
self.assertEqual(1, len(tree.root_node.children[1].children))
self.assertEqual(
"one_one",
tree.root_node.children[0].children[0].entity.description)
self.assertEqual(
"two_one",
tree.root_node.children[1].children[0].entity.description)
def test_string_representation(self):
event_type = EventType(EventTypeIdentifier(1, 2), "My event type")
self.assertEqual("My event type", "%s" % event_type)
def test_greater_or_equal_than_2(self):
event_type1 = EventType(EventTypeIdentifier(1, 7), "Test")
event_type2 = EventType(EventTypeIdentifier(1, 7), "Test2")
self.assertTrue(event_type2 >= event_type1)
def test_less_or_equal_than_1(self):
event_type1 = EventType(EventTypeIdentifier(1, 7), "Test")
event_type2 = EventType(EventTypeIdentifier(2, 1), "Test2")
self.assertTrue(event_type1 <= event_type2)
def test_greater_than_1(self):
event_type1 = EventType(EventTypeIdentifier(1, 2), "Test")
event_type2 = EventType(EventTypeIdentifier(1, 3), "Test2")
self.assertTrue(event_type2 > event_type1)
def test_less_than_1(self):
event_type1 = EventType(EventTypeIdentifier(1, 2), "Test")
event_type2 = EventType(EventTypeIdentifier(1, 3), "Test2")
self.assertTrue(event_type1 < event_type2)
def test_compare_to_none(self):
event_type1 = EventType(EventTypeIdentifier(1, 2), "Test")
self.assertNotEqual(event_type1, None)
def test_not_equals(self):
event_type1 = EventType(EventTypeIdentifier(1, 2), "Test")
event_type2 = EventType(EventTypeIdentifier(1, 3), "Test")
self.assertNotEqual(event_type1, event_type2)
