def setUp(self):
UserTest.setUp(self)
self.user_production = UserProduction(
user_id=self.user.id,
concept="Test Production",
quantity=10,
date=date.today()
)
self.user_production.add()
user_production2 = UserProduction(
user_id=self.user.id,
concept="Test Production",
quantity=10,
date=date.today() + timedelta(days=10)
)
user_production2.add()
self.user_activity = UserActivity(
user_id=self.user.id,
title="Test Activity"
)
self.user_activity.add()
self.order = Order(
user_id=self.user.id,
title="Test Activity",
status="Cancelada",
due_date=date.today()
)
self.order.add()
order2 = Order(
user_id=self.user.id,
title="Test Activity",
status="Pendiente",
due_date=date.today() + timedelta(days=10)
)
order2.add()
class TestGetFinishedWeekOrders(UserScheduleTest):
def setUp(self):
UserScheduleTest.setUp(self)
self.finished_order = Order(
user_id=self.user.id,
title="Test Activity",
status="Completada",
due_date=date.today()
)
self.finished_order.add()
def test_should_return_all_finished_week_orders_given_date(self):
week_finished_orders = self.user.schedule.get_finished_week_orders(
date.today()
)
self.assertEqual(week_finished_orders, [self.finished_order])
def test_should_not_return_finished_orders_given_no_orders_scheduled_for_given_week(self):
week_finished_orders = self.user.schedule.get_finished_week_orders(
date.today() - timedelta(days=10)
)
self.assertEqual(week_finished_orders, [])
class OrderTest(Test):
def setUp(self):
Test.setUp(self)
self.user = User(
username="******",
password="******"
)
self.user.add()
self.order = Order(
user_id=self.user.id,
title="Test Title",
description="Test Description",
due_date=date.today()
)
self.order.add()