def test_without_id_none_companies(self, mock_request, mock_Company):
mock_request.args.get.return_value = None
mock_Company.query.all.return_value = None
response = find(user)
assert mock_request.args.get.called
assert mock_Company.query.all.called
assert response is None
def test_with_id_none_company(self, mock_request, mock_Company):
mock_request.args.get.return_value = {'company_id': 'company_id'}
mock_Company.query.filter_by().first.return_value = None
response = find(user)
assert mock_request.args.get.called
assert mock_Company.query.filter_by().first.called
assert response is None
def test_with_id_success(self, mock_request, mock_Company, mock_Product):
mock_request.args.get.return_value = {'company_id': 'company_id'}
mock_Company.query.filter_by().first.return_value = company
mock_Product.query.filter_by().all.return_value = [product, product]
response = find(user)
assert mock_request.args.get.called
assert mock_Company.query.filter_by().first.called
assert mock_Product.query.filter_by().all.called
assert company_products_validate.validate(response)
def test_without_id_success(self, mock_request, mock_Company,
mock_Product):
mock_request.args.get.return_value = None
mock_Company.query.all.return_value = [company, company, company]
mock_Product.query.filter_by().all.return_value = [product, product]
response = find(user)
assert mock_request.args.get.called
assert mock_Company.query.all.called
assert mock_Product.query.filter_by().all.called
assert companies_products_validate.validate(response)
def test_without_id_none_products(self, mock_request, mock_Company,
mock_Product):
mock_request.args.get.return_value = None
mock_Company.query.all.return_value = [company, company, company]
mock_Product.query.filter_by().all.return_value = None
response = find(user)
assert mock_request.args.get.called
assert mock_Company.query.all.called
assert mock_Product.query.filter_by().all.called
assert companies_products_validate.validate(response)
for _company in response:
assert not _company['products']