class DataBaseIntegrationTests(BaseTest):
def test_user_model_crud(self):
self.user_model = UserModel(6, 'TestUser', '*****@*****.**')
self.assertIsNone(self.user_model.find_by_name('TestUser'))
self.user_model.insert_into_table(self.user_model.username,
self.user_model.password)
self.assertIsNotNone(self.user_model.find_by_name('TestUser'))
def test_store_model_crud(self):
self.store_model = StoreModel(6, 'TestProduct', 19.99)
self.assertIsNone(self.store_model.find_by_product('TestProduct'))
self.assertIsNotNone(self.store_model.find_by_product('car_1'))
def post(self, product):
product = StoreModel.find_by_product(product)
if product:
return {'message': 'Product already in database!'}
else:
parser = reqparse.RequestParser()
parser.add_argument('product',
type=str,
required=True,
help='This field is mandatory!')
parser.add_argument('price',
type=float,
required=True,
help='This field is mandatory!')
data_payload = parser.parse_args()
StoreModel.add_product(data_payload['product'],
data_payload['price'])
return {'message': 'Product successfully added to database!'}, 201
def get(self, product):
products = StoreModel.find_by_product(product)
if products:
return {'product': [product.json() for product in products]}, 200
else:
return {'message': 'Product not found!'}, 404
