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_get_all_returns_all_elements(self): repo = EntityRepo() in_element1 = Mock() in_element1.id_ = 'fake_id1' repo.add(in_element1) in_element2 = Mock() in_element2.id_ = 'fake_id2' repo.add(in_element2) self.assertEqual(set(repo.get_all()), set([in_element1, in_element2]))
def test_get_an_element_by_name(self): repo = EntityRepo() in_element1 = Mock() in_element1.id_ = 'fake_id1' in_element1.name = 'fake_name1' repo.add(in_element1) in_element2 = Mock() in_element2.id_ = 'fake_id2' in_element2.name = 'fake_name2' repo.add(in_element2) self.assertEqual(repo.get_by_name('fake_name1'), in_element1)
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)