def test_02_get_customers_by_name(self): cust = CustomerDto(customer_id=None, first_name='Charlie', last_name='Bone', sex='M', age=10, birthday='2008-04-02', email_address='*****@*****.**', mail_shot_date='11/25', iso_country_code='UK') cust_in_db = type(self).dao.create_customer(cust) self.assertEqual(cust.get_first_name(), cust_in_db.get_first_name())
def test_11_update_customer_that_does_not_exist(self): cust = CustomerDto(customer_id=90, first_name='Pangur', last_name='Ban', sex='M', age=23, birthday='1995-10-12', email_address='*****@*****.**', mail_shot_date='09/08', iso_country_code='USA') type(self).dao.update_customer(cust) actual = type(self).dao.get_customer_by_id(90) self.assertEqual(None, actual)
def test_01_get_customers(self): cust = CustomerDto(customer_id=None, first_name='Spooky', last_name='Dogg', sex='M', age=10, birthday='2008-04-02', email_address='*****@*****.**', mail_shot_date='11/25', iso_country_code='UK') cust_2 = CustomerDto(customer_id=None, first_name='Charlie', last_name='Bone', sex='M', age=10, birthday='2008-04-02', email_address='*****@*****.**', mail_shot_date='11/25', iso_country_code='UK') cust_1_in_db = type(self).dao.create_customer(cust) cust_2_in_bd = type(self).dao.create_customer(cust_2) actual = type(self).dao.get_customers() self.assertEqual([cust_1_in_db, cust_2_in_bd], actual)
def test_04_create_customer(self): cust = CustomerDto(customer_id=None, first_name='Pangur', last_name='Ban', sex='M', age=23, birthday='1995-10-12', email_address='*****@*****.**', mail_shot_date='09/08', iso_country_code='USA') new_cust_in_db = type(self).dao.create_customer(cust) self.assertEqual(cust.get_first_name(), new_cust_in_db.get_first_name()) self.assertEqual(cust.get_last_name(), new_cust_in_db.get_last_name()) self.assertEqual(cust.get_sex(), new_cust_in_db.get_sex()) self.assertEqual(cust.get_age(), new_cust_in_db.get_age()) self.assertEqual(cust.get_birthday(), new_cust_in_db.get_birthday()) self.assertEqual(cust.get_email_addr(), new_cust_in_db.get_email_addr()) self.assertEqual(cust.get_mail_shot_date(), new_cust_in_db.get_mail_shot_date()) self.assertEqual(cust.get_iso_country_code(), new_cust_in_db.get_iso_country_code())
def test_09_update_customer(self): cust = CustomerDto(customer_id=None, first_name='Pangur', last_name='Ban', sex='M', age=23, birthday='1995-10-12', email_address='*****@*****.**', mail_shot_date='09/08', iso_country_code='USA') cust_in_db = type(self).dao.create_customer(cust) expected = CustomerDto(customer_id=cust_in_db.get_customer_id(), first_name='Pangur', last_name='Ban', sex='M', age=24, birthday='1995-10-12', email_address='*****@*****.**', mail_shot_date='09/08', iso_country_code='UK') type(self).dao.update_customer(expected) actual = type(self).dao.get_customer_by_id( cust_in_db.get_customer_id()) self.assertEqual(expected, actual)
def test_08_delete_customer(self): cust = CustomerDto(customer_id=None, first_name='Pangur', last_name='Ban', sex='M', age=23, birthday='1995-10-12', email_address='*****@*****.**', mail_shot_date='09/08', iso_country_code='USA') cust_in_db = type(self).dao.create_customer(cust) type(self).dao.delete_customer(cust_in_db) actual = type(self).dao.get_customer_by_id( cust_in_db.get_customer_id()) self.assertEqual(None, actual)
def test_12_ROLLBACK(self): cust = CustomerDto(customer_id=None, first_name=None, last_name='Bibi', sex='F', age=56, birthday='1962-5-5', email_address='*****@*****.**', mail_shot_date='12/08', iso_country_code='UK') with self.assertRaises(Exception) as e: type(self).dao.create_customer(cust) actual = type(self).dao.get_customers_by_name('Julia', 'Bibi') self.assertEqual([], actual)
def test_00_test_create_dto_from_row(self): expected = CustomerDto(customer_id=None, first_name='Spooky', last_name='Dogg', sex='M', age=10, birthday='2008-04-02', email_address='*****@*****.**', mail_shot_date='11/25', iso_country_code='UK') row = [ None, 'Spooky', 'Dogg', 'M', 10, '2008-04-02', '*****@*****.**', '11/25', 'UK' ] actual = CustomerPostgresDao._create_cust_dto_from_row(row) self.assertEqual(expected, actual)
def create_cust(c): return CustomerDto(c["customer_id"], c["first_name"], c["last_name"], c["sex"], c["age"], c["birthday"], c["email_address"], c["mail_shot_date"], c["iso_country_code"])
def _create_cust_dto_from_row(row): return CustomerDto(customer_id=row[0], first_name=row[1], last_name=row[2], sex=row[3], age=row[4], birthday=row[5], email_address=row[6], mail_shot_date=row[7], iso_country_code=row[8])