class TestReceipt(unittest.TestCase):
def setUp(self):
test_date = date(2020, 5, 1)
self.test_receipt = Receipt(test_date, "Home Depot", 55.19, "DISC")
price_orchid = Price(22.98, 9)
orchid = Item("Orchid", price_orchid)
self.test_receipt.add_item(orchid)
price_lamp_holder = Price(3.78, 9)
lamp_holder = Item("Lamp Holder", price_lamp_holder)
self.test_receipt.add_item(lamp_holder)
price_compressed_air = Price(5.98, 9)
compressed_air = Item("Compressed Air Can", price_compressed_air)
self.test_receipt.add_item(compressed_air)
price_softener = Price(5.97, 9, quantity=3)
softener_pellets = Item("Softener Pellets", price_softener)
self.test_receipt.add_item(softener_pellets)
def test_print(self):
self.test_receipt.pretty_print()
return True
def populate_provinces_tbl(session):
canada = Country("CAN", "Canada")
ontario = Province('Ontario','ON', 13)
quebec = Province('Quebec','QC', 14.975)
canada.provinces = [Province('Alberta','AB', 5),
Province('British Columbia','BC', 12),
Province('Manitoba','MB', 13),
Province('New Brunswick','NB', 13),
Province('Newfoundland and Labrador','NL', 13),
Province('Northwest Territories','NT', 5),
Province('Nova Scotia','NS', 15),
Province('Nunavut','NU', 5),
ontario,
Province('Prince Edward Island','PE', 14),
quebec,
Province('Saskatchewan','SK', 10),
Province('Yukon','YT', 5)
]
session.add(canada)
#Create test user
jhooey = User("Jacob", "Hooey", "jhooey", "password")
#Create test Stores
loblaws = Store("Loblaws", "Rideau and Nelson", ontario)
Maxi = Store("Maxi", "Hull St. Joseph", quebec)
herbspice = Store("Herb and Spice Shop", "375 Bank Street", ontario)
#Create test Receipts
loblaws_receipt1 = Receipt(loblaws)
loblaws_receipt2 = Receipt(loblaws, datetime.date.fromordinal(datetime.date.today().toordinal()-1))
loblaws_receipt3 = Receipt(loblaws, datetime.date.fromordinal(datetime.date.today().toordinal()-4))
#Create Test Items
bananas = Item('Bananas', 'yellow fruit', False)
napkins = Item('Napkins', 'paper napkins', True)
#Add items to receipts
loblaws_receipt1.add_item(session, bananas, 2, 0.79)
loblaws_receipt1.add_item(session, napkins, 1, 2.99)
loblaws_receipt2.add_item(session, bananas, 1.54, 0.79)
loblaws_receipt3.add_item(session, bananas, 10.2, 0.59)
loblaws_receipt3.add_item(session, napkins, 3, 1.99)
#Add Receipts to test user
jhooey.add_receipt(loblaws_receipt1)
jhooey.add_receipt(loblaws_receipt2)
jhooey.add_receipt(loblaws_receipt3)
session.add_all([
loblaws,
Maxi,
herbspice,
jhooey,
bananas,
napkins,
Category('Food', 'Stuff you eat'),
Category('Household Supplies', "Stuff you don't eat")
],
)
session.commit()
