def test_add_one_element(self):
repo = EntityRepo()
in_element = Mock()
in_element.id_ = 'fake_id'
repo.add(in_element)
out_element = repo.get_by_id('fake_id')
self.assertEqual(in_element, out_element)
def test_add_two_elements(self):
repo = EntityRepo()
in_element1 = Mock()
in_element1.id_ = 'fake_id1'
in_element2 = Mock()
in_element2.id_ = 'fake_id2'
repo.add(in_element1)
repo.add(in_element2)
out_element1 = repo.get_by_id('fake_id1')
self.assertEqual(in_element1, out_element1)
self.assertNotEqual(in_element2, out_element1)
