class TestMapper(unittest.TestCase):
def setUp(self):
p = Person()
self.m = ObjectMapper(p)
def test_insert(self):
self.assertEquals(self.m.insert(), ("INSERT INTO person(age,name) VALUES ('?','?')", '22', 'john'))
def test_get_by_id(self):
self.assertEquals(self.m.get_by_id(id=1), ('SELECT age, name FROM person WHERE id_person = ?', 1))
self.assertEquals(self.m.get_by_id(id=1, fields=['age']),
('SELECT age FROM person WHERE id_person = ?', 1))
def test_select_all(self):
self.assertEquals(self.m.get_all(), "SELECT age, name FROM person")
self.assertEquals(self.m.get_all(fields=['name']), "SELECT name FROM person")
def test_delete(self):
self.assertEquals(self.m.delete(id=1), ('DELETE FROM person WHERE id_person = ?', 1))
def test_update(self):
self.assertEquals(self.m.update(id=1), ("UPDATE person SET age = '?', name = '?' WHERE id_person = ?", '22', 'john', 1))
def test_get_table(self):
self.assertEquals(self.m.get_table(), 'person')
def test_get_id(self):
self.assertEquals(self.m.get_id(), 'id_person')
