def test_delete_a_customer(self):
""" Delete a customer """
customer = Customer(first_name="Marry",
last_name="Wang",
user_id="marrywang",
password="******",
active=True)
customer.save()
address = Address(street="100 W 100 St.",
apartment="100",
city="New York",
state="New York",
zip_code="100")
address.customer_id = customer.customer_id
address.save()
customer.address_id = address.id
customer.save()
self.assertEqual(len(Customer.all()), 1)
self.assertEqual(len(Address.all()), 1)
# delete the customer and make sure it isn't in the database
customer.delete()
self.assertEqual(len(Customer.all()), 0)
self.assertEqual(len(Address.all()), 0)
def test_key_error_on_delete(self, bad_key_mock):
""" Test KeyError on delete"""
bad_key_mock.side_effect = KeyError()
customer = Customer("Arturo", "Frank", "USA", "*****@*****.**", "IAmUser",
"password", "1231231234", True, 1)
customer.save()
time.sleep(WAIT_SECONDS)
customer.delete()
time.sleep(WAIT_SECONDS)
def test_delete_a_customer(self):
""" Delete a Customer"""
customer = Customer("Arturo", "Frank", "USA", "*****@*****.**", "IAmUser",
"password", "1231231234", True, 1)
customer.save()
time.sleep(WAIT_SECONDS)
self.assertEqual(len(Customer.all()), 1)
# delete the customer and make sure it isn't in the database
customer.delete()
self.assertEqual(len(Customer.all()), 0)
def test_delete_a_customer(self):
""" Delete a Customer """
customer = Customer(
first_name="Harry",
last_name="Potter",
email="*****@*****.**",
address="9 3/4 Gryffindor Lane",
active=True,
)
customer.create()
self.assertEqual(len(Customer.all()), 1)
customer.delete()
self.assertEqual(len(Customer.all()), 0)
def test_key_error_on_delete(self, bad_mock):
""" Test KeyError on delete """
bad_mock.side_effect = KeyError()
customer = Customer(firstname="John",
lastname="Doe",
email="*****@*****.**",
subscribed=False,
address1="123 Main St",
address2="1B",
city="New York",
country="USA",
province="NY",
zip="12310")
customer.create()
customer.delete()
def test_delete_a_customer(self):
""" Delete a Customer """
customer = Customer(firstname="John",
lastname="Doe",
email="*****@*****.**",
subscribed=False,
address1="123 Main St",
address2="1B",
city="New York",
country="USA",
province="NY",
zip="12310")
customer.save()
self.assertEqual(len(customer.all()), 1)
# delete the customer and make sure it isn't in the database
customer.delete()
self.assertEqual(len(customer.all()), 0)
