From ee88ec845f602640532ba77eb3dee2b4221f1ebf Mon Sep 17 00:00:00 2001 From: wuttinanhi Date: Tue, 8 Nov 2022 02:35:39 +0000 Subject: [PATCH] better mocking --- mock/seed_car.py | 6 +++--- mock/seed_chat.py | 23 +++++++++++++---------- mock/seed_parking_lot.py | 9 ++++++--- mock/seed_reservation.py | 6 +++--- mock/seed_user.py | 23 +++++++++++------------ 5 files changed, 36 insertions(+), 31 deletions(-) diff --git a/mock/seed_car.py b/mock/seed_car.py index ffb8edc..a2cfae0 100644 --- a/mock/seed_car.py +++ b/mock/seed_car.py @@ -13,6 +13,6 @@ def seed_car(): user = UserService.find_by_email("annie@example.com") # mock user car - CarService.add(user, "A11111", "Tesla") - CarService.add(user, "A22222", "Starship") - CarService.add(user, "A33333", "Falcon9") + CarService.add(user, "A-11111", "Tesla Car") + CarService.add(user, "A-22222", "Starship Rocket") + CarService.add(user, "A-33333", "Falcon Nine Rocket") diff --git a/mock/seed_chat.py b/mock/seed_chat.py index 40483b0..968eab2 100644 --- a/mock/seed_chat.py +++ b/mock/seed_chat.py @@ -15,16 +15,19 @@ def seed_chat(): charlie = UserService.find_by_email("charlie@example.com") # mock dummy chat - ChatService.send_chat(annie, bobbie, "1 start Hello Bobbie!") - ChatService.send_chat(bobbie, annie, "2 Hi Annie!") - ChatService.send_chat(annie, bobbie, "3 How are you?") - ChatService.send_chat(bobbie, annie, "4 end I'm good!") - - ChatService.send_chat(bobbie, charlie, "1 start Hello Charlie!") - ChatService.send_chat(charlie, bobbie, "2 end Hi Bobbie!") - - ChatService.send_chat(annie, charlie, "1 start Hello Charlie!") - ChatService.send_chat(charlie, annie, "2 end Hi Annie!") + # annie -> bobbie + ChatService.send_chat(annie, bobbie, "Hello Bobbie!") + ChatService.send_chat(bobbie, annie, "Hi Annie!") + ChatService.send_chat(annie, bobbie, "Can you move your car?") + ChatService.send_chat(bobbie, annie, "Sure!") + + # bobbie -> charlie + ChatService.send_chat(bobbie, charlie, "Hi Charlie!") + ChatService.send_chat(charlie, bobbie, "Hello Bobbie!") + + # annie -> charlie + ChatService.send_chat(annie, charlie, "Hello Charlie!") + ChatService.send_chat(charlie, annie, "Hi Annie!") if __name__ == "__main__": diff --git a/mock/seed_parking_lot.py b/mock/seed_parking_lot.py index 03e95c1..235cdba 100644 --- a/mock/seed_parking_lot.py +++ b/mock/seed_parking_lot.py @@ -8,6 +8,9 @@ def seed_parking_lot(): print("Mocking parking lot...") - ParkingLotService.add("Floor 1", True) - ParkingLotService.add("Floor 2", True) - ParkingLotService.add("Floor 3", False) + for building in range(1, 5): + for floor in range(1, 5): + for lot in range(1, 10): + ParkingLotService.add( + f"Building {building} Floor {floor} Lot {lot}", True + ) diff --git a/mock/seed_reservation.py b/mock/seed_reservation.py index e6f9999..8cb03e5 100644 --- a/mock/seed_reservation.py +++ b/mock/seed_reservation.py @@ -18,15 +18,15 @@ def seed_reservation(): car_1 = CarService.find_by_id(1) parking_lot_1 = ParkingLotService.find_by_id(1) - for i in range(1, 11): - parking_date = datetime(year=2022, month=1, day=1, hour=i, minute=0, second=0) + for i in range(1, 5): + parking_date = datetime.now().replace(hour=0, minute=0, second=0, microsecond=0) # mock reservation reservation = ReservationService.create_reservation( user, car_1, parking_lot_1, parking_date ) - if i < 10: + if i < 4: # end created reservation ReservationService.end_reservation( reservation, reservation.start_time + timedelta(hours=1) diff --git a/mock/seed_user.py b/mock/seed_user.py index 528cffd..0f3aadc 100644 --- a/mock/seed_user.py +++ b/mock/seed_user.py @@ -12,8 +12,8 @@ def seed_user(): "root@example.com", "root-password", "root", - "root-firstname", - "root-lastname", + "Root", + "Admin", "0000000000", "0000000000000", ) @@ -22,9 +22,9 @@ def seed_user(): "annie@example.com", "annie-password", "annie", - "annie-firstname", - "annie-lastname", - "1111111111", + "Annie", + "Carparker", + "0101234567", "1111111111111", ) @@ -32,9 +32,9 @@ def seed_user(): "bobbie@example.com", "bobbie-password", "bobbie", - "bobbie-firstname", - "bobbie-lastname", - "2222222222", + "Bobbie", + "Carparker", + "0201234567", "2222222222222", ) @@ -42,9 +42,8 @@ def seed_user(): "charlie@example.com", "charlie-password", "charlie", - "charlie-firstname", - "charlie-lastname", - "3333333333", + "Charlie", + "Carparker", + "0301234567", "3333333333333", ) -