def test_filter_dict_with_field(self):
data = {
'key1': Dummy(id=1, name='NAME_A', check='NAME_X'),
'key2': Dummy(id=2, name='NAME_A', check='NAME_A'),
'key3': Dummy(id=3, name='NAME_B', check='NAME_X'),
'key4': Dummy(id=4, name='NAME_B', check='NAME_X'),
'key5': Dummy(id=5, name='NAME_B', check='NAME_B'),
'key6': Dummy(id=6, name='NAME_B', check='NAME_B')
}
lm = EmbeddedManager(data, type=Dummy)
self.assertEqual(3, len(lm.filter(name=F('check'))))
self.assertEqual(2, lm.filter(name=F('check'))['key2'].id)
self.assertEqual(5, lm.filter(name=F('check'))['key5'].id)
self.assertEqual(6, lm.filter(name=F('check'))['key6'].id)
def test_filter_with_field(self):
data = [
Dummy(id=1, name='NAME_A', check='NAME_X'),
Dummy(id=2, name='NAME_A', check='NAME_A'),
Dummy(id=3, name='NAME_B', check='NAME_X'),
Dummy(id=4, name='NAME_B', check='NAME_X'),
Dummy(id=5, name='NAME_B', check='NAME_B'),
Dummy(id=6, name='NAME_B', check='NAME_B')
]
lm = EmbeddedManager(data, type=Dummy)
self.assertEqual(3, len(lm.filter(name=F('check'))))
self.assertEqual(2, lm.filter(name=F('check'))[0].id)
self.assertEqual(5, lm.filter(name=F('check'))[1].id)
self.assertEqual(6, lm.filter(name=F('check'))[2].id)
def test_get_from_dict_with_field(self):
data = {
'key1': Dummy(id=1, name='NAME_A', check='NAME_X'),
'key2': Dummy(id=2, name='NAME_A', check='NAME_A'),
'key3': Dummy(id=3, name='NAME_B', check='NAME_X'),
'key4': Dummy(id=4, name='NAME_B', check='NAME_X'),
'key5': Dummy(id=5, name='NAME_B', check='NAME_B'),
'key6': Dummy(id=6, name='NAME_B', check='NAME_B')
}
lm = EmbeddedManager(data, type=Dummy)
self.assertEqual(2, lm.get(name=F('check')).id)
def test_get_with_field(self):
data = [
Dummy(id=1, name='NAME_A', check='NAME_X'),
Dummy(id=2, name='NAME_A', check='NAME_A'),
Dummy(id=3, name='NAME_B', check='NAME_X'),
Dummy(id=4, name='NAME_B', check='NAME_X'),
Dummy(id=5, name='NAME_B', check='NAME_B'),
Dummy(id=6, name='NAME_B', check='NAME_B')
]
lm = EmbeddedManager(data, type=Dummy)
self.assertEqual(2, lm.get(name=F('check')).id)