def main(): print('Customer 1') email = input('Enter email address: ') customer1 = Customer(email) customer1.input_info() customer1.verify_info() output_file = open('customers.txt', 'w') output = customer1.__str__() output_file.write(output) print('Customer 2') email = input('Enter email address: ') customer2 = Customer(email) customer2.input_info() customer2.verify_info() output_file = open('customers.txt', 'a') output = customer2.__str__() output_file.write(output) print("Data of two customers written to the file 'customers.txt'.")
class TestCriticalFunctions(unittest.TestCase): def test_customer(self): self.customer = Customer("Matti Mallikas","Abc 1 00000 Helsinki","0440000000","*****@*****.**" ) self.teststring = "Matti Mallikas, Abc 1 00000 Helsinki, 0440000000, [email protected]" self.assertEqual(self.customer.__str__(), self.teststring) #Yksikkötesti kesken def test_FailReservation(self): self.list1 = [] self.list2 = [] self.calendar = Calendar() self.room = Room(2, 2, 60.99) self.list1.append(self.room) self.hotel = Hotel("Test Hotel",self.list1, self.list2, self.calendar) self.cust = self.hotel.add_customer("Matti Mallikas","Abc 1 00000 Helsinki","0440000000","*****@*****.**" ) self.reservation1 = self.hotel.make_reservation(self.cust, "Room", "15.5.2019", "2")
class MyTestCase(unittest.TestCase): def setUp(self): self.Cust1 = Customer("Smith", "John", "555-123-5555", 1200) def tearDown(self): del self.Cust1 def test_cust_id(self): self.assertEqual(self.Cust1.customer_id, 1200) def test_last_name(self): self.assertEqual(self.Cust1.last_name, "Smith") def test_first_name(self): self.assertEqual(self.Cust1.first_name, "John") def test_phone_number(self): self.assertEqual(self.Cust1.phone_number, "555-123-5555") def test_bad_cust_id(self): with self.assertRaises(Exception): self.Cust1 = Customer("Smith", "John", "555-123-5555", 12) def test_bad_last_name(self): with self.assertRaises(Exception): self.Cust1 = Customer("Sm99ith", "John", "555-123-5555", 1200) def test_bad_first_name(self): with self.assertRaises(Exception): self.Cust1 = Customer("Smith", "Jo99hn", "555-123-5555", 1200) def test_bad_phone_number(self): with self.assertRaises(Exception): self.Cust1 = Customer("Smith", "John", "555123-5555", 1200) def test_str(self): self.assertEqual( self.Cust1.__str__(), "Customer Id: 1200, first name John, last name Smith, Phone:555-123-5555" )
def add_customer(self, name, adress, number, email): customer = Customer(name, adress, number, email) self.customers.append(customer) self.customer_names.append(name) self.save(customer.__str__()) return customer