def test_db_connection(self):
""" Test DB connection """
Promotion.disconnect()
promotion = PromotionFactory()
resp = self.app.post('/promotions',
data=json.dumps(
dict(code=promotion.code,
percentage=promotion.percentage,
expiry_date=promotion.expiry_date,
start_date=promotion.start_date,
products=promotion.products)),
content_type='application/json')
self.assertEqual(resp.status_code,
status.HTTP_500_INTERNAL_SERVER_ERROR)
Promotion.connect()
resp = self.app.post('/promotions',
data=json.dumps(
dict(code=promotion.code,
percentage=promotion.percentage,
expiry_date=promotion.expiry_date,
start_date=promotion.start_date,
products=promotion.products)),
content_type='application/json')
self.assertEqual(resp.status_code, status.HTTP_201_CREATED)
def test_disconnect(self):
""" Test Disconnet """
Promotion.disconnect()
promotion = Promotion("A002", "dog", False)
self.assertRaises(AttributeError, promotion.save)
