def test_append_save(self): i1 = MockInfo() kb.append('a', 'b', i1) kb.raw_write('a', 'b', 3) self.assertEqual(kb.raw_read('a', 'b'), 3)
def test_observer_multiple_observers(self): observer1 = Mock() observer2 = Mock() kb.add_observer(None, None, observer1) kb.add_observer(None, None, observer2) kb.raw_write('a', 'b', 1) observer1.assert_called_once_with('a', 'b', 1) observer2.assert_called_once_with('a', 'b', 1)
def test_observer_multiple_observers(self): observer1 = Mock() observer2 = Mock() kb.add_observer(observer1) kb.add_observer(observer2) kb.raw_write('a', 'b', 1) observer1.append.assert_called_once_with('a', 'b', 1, ignore_type=True) observer2.append.assert_called_once_with('a', 'b', 1, ignore_type=True)
def test_observer_all(self): observer = Mock() kb.add_observer(None, None, observer) kb.raw_write('a', 'b', 1) observer.assert_called_once_with('a', 'b', 1) observer.reset_mock() i = MockInfo() kb.append('a', 'c', i) observer.assert_called_with('a', 'c', i)
def test_observer_location_b(self): observer = Mock() kb.add_observer('a', 'b', observer) kb.raw_write('a', 'b', 1) observer.assert_called_once_with('a', 'b', 1) observer.reset_mock() # Shouldn't call the observer kb.raw_write('a', 'xyz', 1) self.assertFalse(observer.called) i = MockInfo() kb.append('a', 'b', i) observer.assert_called_with('a', 'b', i)
def test_types_observer(self): observer = Mock() info_inst = MockInfo() kb.add_types_observer(Info, observer) kb.append('a', 'b', info_inst) observer.assert_called_once_with('a', 'b', info_inst) observer.reset_mock() info_inst = MockInfo() kb.append('a', 'c', info_inst) observer.assert_called_with('a', 'c', info_inst) observer.reset_mock() # Should NOT call it because it is NOT an Info instance some_int = 3 kb.raw_write('a', 'd', some_int) self.assertEqual(observer.call_count, 0)
def test_raw_write_list(self): """ Test for _get_uniq_id which needs to be able to hash any object type. """ kb.raw_write('a', 'b', [1,2,3]) self.assertEqual(kb.raw_read('a','b'), [1,2,3])
def test_overwrite(self): kb.raw_write('a', 'b', 'abc') kb.raw_write('a', 'b', 'def') self.assertEqual(kb.raw_read('a', 'b'), 'def')
def test_dump(self): kb.raw_write('a', 'b', 1) self.assertEqual(kb.dump(), {'a': {'b': [1]}})
def test_all_of_klass_str(self): kb.raw_write('a', 'b', 'abc') self.assertEqual(kb.get_all_entries_of_class(str), ['abc'])
def test_all_of_klass(self): kb.raw_write('a', 'b', 1) self.assertEqual(kb.get_all_entries_of_class(int), [1])
def test_raw_write_dict(self): kb.raw_write('a', 'b', {}) self.assertEqual(kb.raw_read('a', 'b'), {})
def test_basic(self): kb.raw_write('a', 'b', 'c') data = kb.raw_read('a', 'b') self.assertEqual(data, 'c')
def test_clear(self): kb.raw_write('a', 'b', 'abc') kb.raw_write('a', 'c', 'abc') kb.clear('a', 'b') self.assertEqual(kb.raw_read('a', 'b'), []) self.assertEqual(kb.raw_read('a', 'c'), 'abc')
def test_default_first_saved(self): kb.raw_write('a', 'b', 'c') self.assertEqual(kb.get('a', 'not-exist'), []) self.assertEqual(kb.raw_read('a', 'not-exist'), [])