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)