class ProductionTest(UserTest):
def setUp(self):
UserTest.setUp(self)
self.user_production = UserProduction(user_id=self.user.id,
concept="Test Concept",
quantity=10)
self.user_production.add()
def test_should_add_user_production_given_valid_user_production(self):
user_production = UserProduction(user_id=self.user.id,
concept="Test Concept",
quantity=10)
user_production.request.add()
self.assertIn(user_production, self.db.session)
def test_should_return_error_given_empty_quantity(self):
user_production = UserProduction(user_id=self.user.id,
concept="Test Concept",
quantity="")
error = user_production.validation.validate_empty_values()
self.assertNotEqual(error, None)
def test_should_return_error_given_invalid_user_production(self):
user_production = UserProduction(user_id=self.user.id,
concept="",
quantity=10)
error = user_production.validation.validate()
self.assertNotEqual(error, None)
def register_production(user_id):
if request.method == "POST":
production = UserProduction(user_id=user_id,
concept=request.form['concept'],
quantity=request.form['quantity'])
error = production.request.add()
if not error:
return redirect(url_for('user.profile', id=user_id))
flash(error)
return render_template("user/production/register-production.html",
production_heads=production_heads)
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()