def test_competition_json(self): company = CompanyModel('GOL') expected = { 'id': company.id, 'name': company.name, } self.assertEqual( company.json(), expected, "The JSON export of the company is incorrect. Received {}, expected {}." .format(company.json(), expected))
def post(self, name): if CompanyModel.find_by_name(name): return {'message': "A store with name '{}' already exists.".format(name)}, 400 company = CompanyModel(name) try: company.save_to_db() except: return {"message": "An error occurred creating the store."}, 500 return company.json(), 201
def post(self): data = self.parser.parse_args() companyName = data['companyName'] if CompanyModel.find_by_name(companyName): return { 'message': "An item with name '{}' already exists.".format(companyName) }, 400 company = CompanyModel(companyName, **data) try: company.save_to_db() except: return { "message": "An error occurred while inserting the item." }, 500 return company.json(), 201