def setUp(self):
"""Setup method to instance an object of Customer class"""
code = "cust_002"
first_name = "Jimena"
last_name = "Sanabria"
date_of_birth = "21/08/1980"
address_1 = "Calle Nueva Granada #1837"
phone_1 = "70759942"
phone_2 = "4295449"
email = "*****@*****.**"
membership = "Gold"
status = "ACTIVE"
self.required_customer = Customer("cust_001", "Andrea", "Vargas", "01/01/2000")
self.customer = Customer(code, first_name, last_name, date_of_birth, [address_1], [phone_1, phone_2], email, membership, status)
class CustomerTest(unittest.TestCase):
def setUp(self):
"""Setup method to instance an object of Customer class"""
code = "cust_002"
first_name = "Jimena"
last_name = "Sanabria"
date_of_birth = "21/08/1980"
address_1 = "Calle Nueva Granada #1837"
phone_1 = "70759942"
phone_2 = "4295449"
email = "*****@*****.**"
membership = "Gold"
status = "ACTIVE"
self.required_customer = Customer("cust_001", "Andrea", "Vargas", "01/01/2000")
self.customer = Customer(code, first_name, last_name, date_of_birth, [address_1], [phone_1, phone_2], email, membership, status)
def test_create_customer_with_required_data(self):
"""Test if an instance of Customer class is created with required"""
self.assertIsInstance(self.required_customer, Customer)
def test_update_name_of_required_customer(self):
"""Test to update the first_name attribute"""
update_name = "Pamela"
self.required_customer.set_first_name(update_name)
self.assertEqual(update_name, self.required_customer.get_first_name())
def test_update_last_name_of_required_customer(self):
"""Test to update the last_name attribute"""
update_last_name = "Veizaga"
self.required_customer.set_last_name(update_last_name)
self.assertEqual(update_last_name, self.required_customer.get_last_name())
def test_update_date_of_birth_of_required_customer(self):
"""Test to update the date_of_birth attribute"""
update_date_of_birth = "01/10/2001"
self.required_customer.set_date_of_birth(update_date_of_birth)
self.assertEqual(update_date_of_birth, self.required_customer.get_date_of_birth())
def test_add_addresses_of_required_customer(self):
"""Test to add addresses attribute"""
addresses = ["Calle Melchor Perez #3232", "Av. Blanco Galindo"]
self.required_customer.set_addresses(addresses)
self.assertEqual(addresses, self.required_customer.get_addresses())
def test_add_phones_of_required_customer(self):
"""Test to add phones attribute"""
phones = ["3254789", "7894789"]
self.required_customer.set_phones(phones)
self.assertEqual(phones, self.required_customer.get_phones())
def test_add_email_of_required_customer(self):
"""Test to add the email attribute"""
email = "*****@*****.**"
self.required_customer.set_email(email)
self.assertEqual(email, self.required_customer.get_email())
def test_updated_memebership_type_of_required_customer(self):
"""Test to update the membership attribute"""
update_membership = "Gold"
self.required_customer.set_membership(update_membership)
self.assertEqual(update_membership, self.required_customer.get_membership())
def test_update_status_of_required_customer(self):
"""Test to update the status attribute"""
update_status = "INACTIVE"
self.required_customer.set_status(update_status)
self.assertEqual(update_status, self.required_customer.get_status())
def test_create_customer(self):
"""Test if an instance of Customer class is created"""
self.assertIsInstance(self.customer, Customer)
def test_update_name_of_customer(self):
"""Test to update the first_name attribute"""
update_name = "Giovanna"
self.customer.set_first_name(update_name)
self.assertEqual(update_name, self.customer.get_first_name())
def test_update_last_name_of_customer(self):
"""Test to update the last_name attribute"""
update_last_name = "Tames"
self.customer.set_last_name(update_last_name)
self.assertEqual(update_last_name, self.customer.get_last_name())
def test_update_date_of_birth_of_customer(self):
"""Test to update the date_of_birth attribute"""
update_date_of_birth = "22/08/2000"
self.customer.set_date_of_birth(update_date_of_birth)
self.assertEqual(update_date_of_birth, self.customer.get_date_of_birth())
def test_update_addresses_of_customer(self):
"""Test to update the addresses attribute"""
update_addresses = ["Calle Madrid # 333", "Calle Villa de Oropeza # 111"]
self.customer.set_addresses(update_addresses)
self.assertEqual(update_addresses, self.customer.get_addresses())
def test_update__phone_numbers_of_customer(self):
"""Test to update the phones attribute"""
update_phones = ["78957894", "60745628"]
self.customer.set_phones(update_phones)
self.assertEqual(update_phones, self.customer.get_phones())
def test_update_email_account_of_customer(self):
"""Test to update the email attribute"""
update_email = "*****@*****.**"
self.customer.set_email(update_email)
self.assertEqual(update_email, self.customer.get_email())
def test_update_membership_type_of_customer(self):
"""Test to update the membership attribute"""
update_membership = "Platinum"
self.customer.set_membership(update_membership)
self.assertEqual(update_membership, self.customer.get_membership())
def test_update_status_of_customer(self):
"""Test to update the status attribute"""
update_status = "INACTIVE"
self.customer.set_status(update_status)
self.assertEqual(update_status, self.customer.get_status())
def create_customer_object(self, code, data):
"""Get customer object with data from csv file.
Keyword arguments:
code -- the str with code to create customer object
data -- the string dictionary with data to create customer object
Return in a customer object
"""
customer = Customer(code)
customer.set_first_name(data.get("First Name"))
customer.set_last_name(data.get("Last Name"))
customer.set_date_of_birth(data.get("Date of birth"))
customer.set_email(data.get("Email"))
customer.set_membership(data.get("Membership"))
customer.set_status(data.get("Status"))
customer.set_addresses(get_string_list_data(data, "Addresses"))
customer.set_phones(get_integer_list_data(data, "Phone numbers"))
return customer
