def setUp(self):
adapter = mock.Mock()
taskModel = mock.Mock()
self.km = KlassMap(adapter = adapter, namespace = "foo-ins1", taskModel = taskModel)
self.tm1 = TaskMapBase(klass = 0)
self.tm1.__class__._Item = DummyTaskItem
self.km.addTaskMap(self.tm1)
self.tm2 = TaskMapBase(klass = 1)
self.tm2.__class__._Item = DummyTaskItem
self.km.addTaskMap(self.tm2)
self.am = mock.Mock()
self.km.setAdapterMap(self.am)
def setUp(self):
self.tm = TaskMapBase(klass = 0)
self.tm.__class__._Item = DummyTaskItem
class TaskMapTest(TestCase):
def setUp(self):
self.tm = TaskMapBase(klass = 0)
self.tm.__class__._Item = DummyTaskItem
def test_ordereddict(self):
self.tm[1] = 1
self.tm[3] = 2
self.tm[2] = 3
self.tm[4] = 4
self.assertEqual(list(self.tm.__iter__()),
[1, 3, 2, 4])
self.assertEqual(self.tm[1], 1)
self.assertEqual(self.tm[3], 2)
self.assertEqual(self.tm[2], 3)
self.assertEqual(self.tm[4], 4)
self.assertRaises(KeyError, self.tm.__getitem__, "error")
# __contains__
for key in [1, 2, 3, 4]:
self.assertTrue(key in self.tm)
for key in [0, -1, 5]:
self.assertTrue(key not in self.tm)
def test_updateData_insert_modify(self):
klassMap = mock.Mock()
taskModel = mock.Mock()
klassMap.beforeInsert = lambda *args: True
self.tm.setKlassMap(klassMap)
self.tm.setTaskModel(taskModel)
self.tm.namespace = "foo-1"
self.tm.updateData({
2: 1,
})
self.tm.updateData({
1: 11,
2: 13,
})
self.assertEqual(list(self.tm.__iter__()),
[2, 1])
self.assertTrue(1 in self.tm)
self.assertTrue(2 in self.tm)
self.assertEqual(self.tm[1].value, 11)
self.assertEqual(self.tm[2].value, 13)
def test_keys(self):
klassMap = mock.Mock()
taskModel = mock.Mock()
klassMap.beforeInsert = lambda *args: True
self.tm.setKlassMap(klassMap)
self.tm.setTaskModel(taskModel)
self.tm.namespace = "foo-1"
self.tm.updateData({
2: 1,
})
self.tm.updateData({
1: 11,
2: 13,
})
self.assertListEqual(list(self.tm.keys()), [2, 1])
def setUp(self):
self.tm = TaskMapBase(klass=0)
self.tm.__class__._Item = DummyTaskItem
class TaskMapTest(TestCase):
def setUp(self):
self.tm = TaskMapBase(klass=0)
self.tm.__class__._Item = DummyTaskItem
def test_ordereddict(self):
self.tm[1] = 1
self.tm[3] = 2
self.tm[2] = 3
self.tm[4] = 4
self.assertEqual(list(self.tm.__iter__()), [1, 3, 2, 4])
self.assertEqual(self.tm[1], 1)
self.assertEqual(self.tm[3], 2)
self.assertEqual(self.tm[2], 3)
self.assertEqual(self.tm[4], 4)
self.assertRaises(KeyError, self.tm.__getitem__, "error")
# __contains__
for key in [1, 2, 3, 4]:
self.assertTrue(key in self.tm)
for key in [0, -1, 5]:
self.assertTrue(key not in self.tm)
def test_updateData_insert_modify(self):
klassMap = mock.Mock()
taskModel = mock.Mock()
klassMap.beforeInsert = lambda *args: True
self.tm.setKlassMap(klassMap)
self.tm.setTaskModel(taskModel)
self.tm.namespace = "foo-1"
self.tm.updateData({
2: 1,
})
self.tm.updateData({
1: 11,
2: 13,
})
self.assertEqual(list(self.tm.__iter__()), [2, 1])
self.assertTrue(1 in self.tm)
self.assertTrue(2 in self.tm)
self.assertEqual(self.tm[1].value, 11)
self.assertEqual(self.tm[2].value, 13)
def test_keys(self):
klassMap = mock.Mock()
taskModel = mock.Mock()
klassMap.beforeInsert = lambda *args: True
self.tm.setKlassMap(klassMap)
self.tm.setTaskModel(taskModel)
self.tm.namespace = "foo-1"
self.tm.updateData({
2: 1,
})
self.tm.updateData({
1: 11,
2: 13,
})
self.assertListEqual(list(self.tm.keys()), [2, 1])
class KlassMapTest(TestCase):
def setUp(self):
adapter = mock.Mock()
taskModel = mock.Mock()
self.km = KlassMap(adapter = adapter, namespace = "foo-ins1", taskModel = taskModel)
self.tm1 = TaskMapBase(klass = 0)
self.tm1.__class__._Item = DummyTaskItem
self.km.addTaskMap(self.tm1)
self.tm2 = TaskMapBase(klass = 1)
self.tm2.__class__._Item = DummyTaskItem
self.km.addTaskMap(self.tm2)
self.am = mock.Mock()
self.km.setAdapterMap(self.am)
def test_create_init(self):
self.assertEqual(self.km.namespace, "foo-ins1")
self.assertEqual(len(self.km), 0)
self.assertRaises(NotImplementedError, self.km.__setitem__, "what", "ever")
# namespace is set
self.assertEqual(self.tm1.namespace, "foo-ins1")
def test_create_same_klass(self):
# try add it again, but with the same class
tm1_1 = TaskMapBase(klass = 0)
self.assertRaises(RuntimeError, self.km.addTaskMap, tm1_1)
def test_task_add_update_delete(self):
self.tm1.updateData({
"1": "task1",
"2": "task2",
})
self.assertEqual(len(self.km), 2)
self.assertEqual(self.km["1"].value, "task1")
self.assertEqual(self.km["1"].klass, 0)
self.assertEqual(self.km["2"].value, "task2")
self.assertEqual(self.km["2"].klass, 0)
self.assertEqual(self.km["2"].isDeletionPending, False)
# change item
self.tm1.updateData({
"1": "task1!",
"2": "task2",
})
self.assertEqual(len(self.km), 2)
self.assertEqual(self.km["1"].value, "task1!")
self.assertEqual(self.km["2"].value, "task2")
self.assertEqual(self.km["2"].isDeletionPending, False)
# remove one item
self.tm1.updateData({
"1": "task1!",
})
self.assertEqual(len(self.km), 2)
self.assertEqual(self.km["2"].isDeletionPending, True)
# try to remove it again, shouldn't remove
self.tm1.updateData({
"1": "task1!",
})
self.assertEqual(len(self.km), 2)
self.assertEqual(self.km["2"].isDeletionPending, True)
# updateData with taskMap2
self.tm2.updateData({})
self.assertEqual(len(self.km), 1)
self.assertRaises(KeyError, self.km.__getitem__, "2")
def test_task_move(self):
# order: 1, 2, then move 1 to 2
self.tm1.updateData({
"1": "task1",
})
self.tm2.updateData({
"1": "task1??",
})
self.tm1.updateData({})
self.assertEqual(self.km["1"].isDeletionPending, True)
# until tm1 deletes it, it shouldn't be in tm2
self.assertEqual(len(self.tm1), 1)
self.assertEqual(len(self.tm2), 0)
# after tm2 updates again, it should be moved
self.tm2.updateData({
"1": "task1??",
})
self.assertEqual(len(self.tm1), 0)
self.assertEqual(len(self.tm2), 1)
self.assertEqual(self.tm2["1"].value, "task1??")
self.assertEqual(self.tm2["1"].isDeletionPending, False)
self.assertFalse(self.am.afterMove.called)
def test_task_move_model_move_down(self):
# set tm1 = [0,1,2,3] tm2=[4,5]
self.tm1.updateData({
"0": "task0",
})
self.tm1.updateData({
"0": "task0",
"1": "task1",
})
self.tm1.updateData({
"0": "task0",
"1": "task1",
"2": "task2",
})
self.tm1.updateData({
"0": "task0",
"1": "task1",
"2": "task2",
"3": "task3",
})
self.tm2.updateData({
"4": "task4",
"5": "task5",
})
# move 2 from tm1 to tm2
self.assertFalse(self.km["2"].isDeletionPending)
self.tm1.updateData({
"0": "task0",
"1": "task1",
"3": "task3",
})
self.assertTrue(self.km["2"].isDeletionPending)
self.tm2.updateData({
"2": "task2",
"4": "task4",
"5": "task5",
})
self.assertFalse(self.km["2"].isDeletionPending)
self.am.beforeMove.assert_called_with("foo-ins1", 2, 6)
def test_task_move_model_move_up(self):
# tm1 is empty; tm2's 3rd item is task2
self.tm1.updateData({})
self.tm2.updateData({
"0": "task0",
"1": "task1",
})
self.tm2.updateData({
"0": "task0",
"1": "task1",
"2": "task2",
})
self.tm2.updateData({
"0": "task0",
"1": "task1",
"2": "task2",
"3": "task3",
"4": "task4",
"5": "task5",
})
# move task2 to tm1 from tm2
self.assertFalse(self.km["2"].isDeletionPending)
self.tm2.updateData({
"0": "task0",
"1": "task1",
"3": "task3",
"4": "task4",
"5": "task5",
})
self.assertTrue(self.km["2"].isDeletionPending)
self.tm1.updateData({
"2": "task2moved",
})
self.assertFalse(self.km["2"].isDeletionPending)
self.am.beforeMove.assert_called_with("foo-ins1", 2, 0)
def test_iter_index(self):
self.tm1.updateData({
"1": 10,
})
self.assertEqual(self.km.index("1"), 0)
self.tm2.updateData({
"4": 41,
})
self.assertEqual(self.km.index("4"), 1)
self.tm2.updateData({
"4": 41,
"2": 25,
})
self.assertEqual(self.km.index("2"), 2)
self.tm1.updateData({
"1": 10,
"3": 37,
})
self.assertEqual(self.km.index("3"), 1)
# index()
with self.assertRaises(ValueError):
self.km.index("0")
# __contains__
for key in ["1", "2", "3", "4"]:
self.assertTrue(key in self.km, key)
for key in ["5", "6", "7"]:
self.assertFalse(key in self.km, key)
# __iter__
self.assertListEqual(list(self.km.__iter__()), ['1', '3', '4', '2'])
# items()
self.assertListEqual(
list(map(lambda pair: (pair[0], pair[1].value), self.km.items())),
[("1", 10), ("3", 37), ("4", 41), ("2", 25)],
)
# values()
self.assertListEqual(
list(map(lambda i: i.value, self.km.values())),
[10, 37, 41, 25],
)
def test_get_klassMap(self):
self.assertEqual(self.km.klass(0), self.tm1)
self.assertNotEqual(self.km.klass(1), self.tm1)
def test_findItemKlass(self):
self.tm1.updateData({
"1": 1,
"2": 2,
})
self.tm2.updateData({
"3": 3,
})
self.assertEqual(self.km.findItemKlass("1"), 0)
self.assertEqual(self.km.findItemKlass("2"), 0)
self.assertEqual(self.km.findItemKlass("3"), 1)
with self.assertRaises(KeyError):
self.km.findItemKlass("4")
