def test_create_with_no_type(self): data = [ Dummy(id=1, name='NAME_A'), Dummy(id=2, name='NAME_A'), Dummy(id=3, name='NAME_B'), Dummy(id=4, name='NAME_B') ] lm = EmbeddedManager(data) with self.assertRaisesRegex(TypeError, 'No type defined for EmbeddedManager'): lm.create(id=5, name='CREATED')
def test_create_with_no_key_in_dict_with_type(self): data = { 'key1': Dummy(id=1, name='NAME_A'), 'key2': Dummy(id=2, name='NAME_A'), 'key3': Dummy(id=3, name='NAME_B'), 'key4': Dummy(id=4, name='NAME_B') } lm = EmbeddedManager(data, type=Dummy) with self.assertRaises(ValueError): created = lm.create(id=5, name='CREATED')
def test_create_with_key_in_dict_with_type(self): data = { 'key1': Dummy(id=1, name='NAME_A'), 'key2': Dummy(id=2, name='NAME_A'), 'key3': Dummy(id=3, name='NAME_B'), 'key4': Dummy(id=4, name='NAME_B') } lm = EmbeddedManager(data, type=Dummy) created = lm.create('key5', id=5, name='CREATED') self.assertIsNotNone(created) self.assertTrue(isinstance(created, Dummy)) self.assertEqual(5, created.id) self.assertEqual('CREATED', created.name) self.assertEqual(5, len(lm)) got = lm.get('key5') self.assertEqual(5, got.id) self.assertEqual('CREATED', got.name)
def test_create_with_type(self): data = [ Dummy(id=1, name='NAME_A'), Dummy(id=2, name='NAME_A'), Dummy(id=3, name='NAME_B'), Dummy(id=4, name='NAME_B') ] lm = EmbeddedManager(data, type=Dummy) created = lm.create(id=5, name='CREATED') self.assertIsNotNone(created) self.assertTrue(isinstance(created, Dummy)) self.assertEqual(5, created.id) self.assertEqual('CREATED', created.name) self.assertEqual(5, len(lm)) got = lm.get(id=5) self.assertEqual(5, got.id) self.assertEqual('CREATED', got.name)