def post(self): parser = reqparse.RequestParser() parser.add_argument('company_name', required=True) parser.add_argument('email', required=True) parser.add_argument('password', required=True) parser.add_argument('address_1') parser.add_argument('address_2') parser.add_argument('city') parser.add_argument('province') parser.add_argument('zipcode') parser.add_argument('country') parser.add_argument('phone') parser.add_argument('state', default='NOT_VERIFIED' if os.environ.get('TEST') is None else 'ACTIVE') company_args = parser.parse_args() if company_args.get('state') not in COMPANY_STATES: error_dict = { 'error_message': f'Invalid state {company_args.get("state")}', } LOGGER.error(error_dict) return error_dict, 400 try: return Company.create(**company_args) except IntegrityError as e: error_dict = { 'error_message': e, } LOGGER.error(error_dict) return error_dict, 400 except Exception as e: LOGGER.error(e)
def test_create_1(self): test_company_1 = { 'company_name': 'test', 'email': '*****@*****.**', 'password': '******', 'state': 'TEST', 'address_1': 'test', 'city': 'toronto', 'province': 'ON', 'zipcode': 'xxxxxx', 'country': 'canada', 'phone': '123456789' } return_result = Company.create(**test_company_1) self.assertEqual(test_company_1.get('company_name'), return_result.company_name) self.assertEqual(test_company_1.get('email'), return_result.email) self.assertEqual(test_company_1.get('password'), return_result.password) self.assertEqual(test_company_1.get('state'), return_result.state) self.assertEqual(test_company_1.get('address_1'), return_result.address_1) self.assertEqual(test_company_1.get('address_2'), return_result.address_2) self.assertEqual(test_company_1.get('city'), return_result.city) self.assertEqual(test_company_1.get('province'), return_result.province) self.assertEqual(test_company_1.get('zipcode'), return_result.zipcode) self.assertEqual(test_company_1.get('country'), return_result.country) self.assertEqual(test_company_1.get('phone'), return_result.phone) get_result = Company.get(id=return_result.id) self.assertEqual(test_company_1.get('company_name'), get_result.company_name) self.assertEqual(test_company_1.get('email'), get_result.email) self.assertEqual(test_company_1.get('password'), get_result.password) self.assertEqual(test_company_1.get('state'), get_result.state) self.assertEqual(test_company_1.get('address_1'), get_result.address_1) self.assertEqual(test_company_1.get('address_2'), get_result.address_2) self.assertEqual(test_company_1.get('city'), get_result.city) self.assertEqual(test_company_1.get('province'), get_result.province) self.assertEqual(test_company_1.get('zipcode'), get_result.zipcode) self.assertEqual(test_company_1.get('country'), get_result.country) self.assertEqual(test_company_1.get('phone'), get_result.phone)
def test_create_2(self): test_company_1 = { 'company_name': 'test_1', 'email': '*****@*****.**', 'password': '******', 'state': 'TEST', 'address_1': 'test_1', 'city': 'toronto', 'province': 'ON', 'zipcode': 'xxxxxx', 'country': 'canada', 'phone': '123456789' } test_company_2 = { 'company_name': 'test_2', 'email': '*****@*****.**', 'password': '******', 'state': 'TEST', 'address_1': 'test_2', 'city': 'waterloo', 'province': 'ON', 'zipcode': 'xxxxxx', 'country': 'canada', 'phone': '987654321' } return_result_1 = Company.create(**test_company_1) self.assertEqual(test_company_1.get('company_name'), return_result_1.company_name) self.assertEqual(test_company_1.get('email'), return_result_1.email) self.assertEqual(test_company_1.get('password'), return_result_1.password) self.assertEqual(test_company_1.get('state'), return_result_1.state) self.assertEqual(test_company_1.get('address_1'), return_result_1.address_1) self.assertEqual(test_company_1.get('address_2'), return_result_1.address_2) self.assertEqual(test_company_1.get('city'), return_result_1.city) self.assertEqual(test_company_1.get('province'), return_result_1.province) self.assertEqual(test_company_1.get('zipcode'), return_result_1.zipcode) self.assertEqual(test_company_1.get('country'), return_result_1.country) self.assertEqual(test_company_1.get('phone'), return_result_1.phone) return_result_2 = Company.create(**test_company_2) self.assertEqual(test_company_2.get('company_name'), return_result_2.company_name) self.assertEqual(test_company_2.get('email'), return_result_2.email) self.assertEqual(test_company_2.get('password'), return_result_2.password) self.assertEqual(test_company_2.get('state'), return_result_2.state) self.assertEqual(test_company_2.get('address_1'), return_result_2.address_1) self.assertEqual(test_company_2.get('address_2'), return_result_2.address_2) self.assertEqual(test_company_2.get('city'), return_result_2.city) self.assertEqual(test_company_2.get('province'), return_result_2.province) self.assertEqual(test_company_2.get('zipcode'), return_result_2.zipcode) self.assertEqual(test_company_2.get('country'), return_result_2.country) self.assertEqual(test_company_2.get('phone'), return_result_2.phone) get_result_1 = Company.get(id=return_result_1.id) self.assertEqual(test_company_1.get('company_name'), get_result_1.company_name) self.assertEqual(test_company_1.get('email'), get_result_1.email) self.assertEqual(test_company_1.get('password'), get_result_1.password) self.assertEqual(test_company_1.get('state'), get_result_1.state) self.assertEqual(test_company_1.get('address_1'), get_result_1.address_1) self.assertEqual(test_company_1.get('address_2'), get_result_1.address_2) self.assertEqual(test_company_1.get('city'), get_result_1.city) self.assertEqual(test_company_1.get('province'), get_result_1.province) self.assertEqual(test_company_1.get('zipcode'), get_result_1.zipcode) self.assertEqual(test_company_1.get('country'), get_result_1.country) self.assertEqual(test_company_1.get('phone'), get_result_1.phone) get_result_2 = Company.get(id=return_result_2.id) self.assertEqual(test_company_2.get('email'), get_result_2.email) self.assertEqual(test_company_2.get('password'), get_result_2.password) self.assertEqual(test_company_2.get('state'), get_result_2.state) self.assertEqual(test_company_2.get('address_1'), get_result_2.address_1) self.assertEqual(test_company_2.get('address_2'), get_result_2.address_2) self.assertEqual(test_company_2.get('city'), get_result_2.city) self.assertEqual(test_company_2.get('province'), get_result_2.province) self.assertEqual(test_company_2.get('zipcode'), get_result_2.zipcode) self.assertEqual(test_company_2.get('country'), get_result_2.country) self.assertEqual(test_company_2.get('phone'), get_result_2.phone)
def addCompanies(): print "Adding companies..." Company.create(name="Sprint", ticker="S") Company.create(name="Bank of America", ticker="BAC") Company.create(name="Pfizer", ticker="PFE") Company.create(name="Advanced Micro Devices", ticker="AMD") Company.create(name="General Electric", ticker="GE") Company.create(name="Ambev ADS", ticker="ABEV") Company.create(name="Walgreen", ticker="WAG") Company.create(name="Ford Motor", ticker="F") Company.create(name="Rite Aid", ticker="RAD") Company.create(name="Taiwan Semiconductor Manufacturing ADS", ticker="TSM") Company.create(name="Petroleo Brasileiro ADS", ticker="PBR") Company.create(name="AT&T", ticker="T") Company.create(name="Verizon Communications", ticker="VZ") Company.create(name="Boston Scientific", ticker="BSX") Company.create(name="Twitter", ticker="TWTR") Company.create(name="SandRidge Energy", ticker="SD") Company.create(name="Wells Fargo", ticker="WFC") Company.create(name="Delta Air Lines", ticker="DAL") Company.create(name="Alcoa", ticker="AA") Company.create(name="Coca-Cola", ticker="KO") Company.create(name="Regions Financial", ticker="RF") Company.create(name="AbbVie", ticker="ABBV") Company.create(name="Post Holdings", ticker="POST") Company.create(name="JPMorgan Chase", ticker="JPM") Company.create(name="Coach Inc.", ticker="COH") Company.create(name="Western Union", ticker="WU") Company.create(name="CBS Cl B", ticker="CBS") Company.create(name="Citigroup", ticker="C") Company.create(name="Molycorp", ticker="MCP") Company.create(name="J.C. Penney", ticker="JCP") Company.create(name="Vale ADS", ticker="VALE") Company.create(name="EMC", ticker="EMC") Company.create(name="Exxon Mobil", ticker="XOM") Company.create(name="Nokia ADS", ticker="NOK") Company.create(name="BP ADS", ticker="BP") Company.create(name="Kinross Gold", ticker="KGC") Company.create(name="Medtronic", ticker="MDT") Company.create(name="General Motors", ticker="GM") Company.create(name="SunEdison", ticker="SUNE") Company.create(name="MGIC Investment", ticker="MTG") Company.create(name="T-Mobile US", ticker="TMUS") Company.create(name="Oracle", ticker="ORCL") Company.create(name="Alpha Natural Resources", ticker="ANR") Company.create(name="Talisman Energy", ticker="TLM") Company.create(name="Barrick Gold", ticker="ABX") Company.create(name="Morgan Stanley", ticker="MS") Company.create(name="Petroleo Brasileiro ADS A", ticker="PBRA") Company.create(name="Itau Unibanco Holding ADS", ticker="ITUB") Company.create(name="Bank of New York Mellon", ticker="BK") Company.create(name="KeyCorp", ticker="KEY") Company.create(name="American International Group", ticker="AIG") Company.create(name="Gap", ticker="GPS") Company.create(name="Time Warner", ticker="TWX") Company.create(name="D.R. Horton", ticker="DHI") Company.create(name="Dow Chemical", ticker="DOW") Company.create(name="Banco Santander ADS", ticker="SAN") Company.create(name="Paragon Offshore", ticker="PGN") Company.create(name="Las Vegas Sands", ticker="LVS") Company.create(name="Quicksilver Resources", ticker="KWK") Company.create(name="Lowe's", ticker="LOW") Company.create(name="Covidien", ticker="COV") Company.create(name="PPL", ticker="PPL") Company.create(name="Home Depot", ticker="HD") Company.create(name="Gafisa ADS", ticker="GFA") Company.create(name="Merck&Co", ticker="MRK") Company.create(name="Marathon Oil", ticker="MRO") Company.create(name="International Game Technology", ticker="IGT") Company.create(name="United Continental Holdings", ticker="UAL") Company.create(name="Ryerson Holding", ticker="RYI") Company.create(name="Magnum Hunter Resources", ticker="MHR") Company.create(name="Chesapeake Energy", ticker="CHK") Company.create(name="McDonald's", ticker="MCD") Company.create(name="NRG Energy", ticker="NRG") Company.create(name="Genworth Financial Cl A", ticker="GNW") Company.create(name="MGM Resorts International", ticker="MGM") Company.create(name="Walt Disney", ticker="DIS") Company.create(name="SouFun Holdings", ticker="SFUN") Company.create(name="Youku Tudou ADS", ticker="YOKU") Company.create(name="Dollar General", ticker="DG") Company.create(name="Procter&Gamble", ticker="PG") Company.create(name="U.S. Bancorp", ticker="USB") Company.create(name="Xerox", ticker="XRX") Company.create(name="Corning", ticker="GLW") Company.create(name="Exelon", ticker="EXC") Company.create(name="Yamana Gold", ticker="AUY") Company.create(name="Cemex ADS", ticker="CX") Company.create(name="Charles Schwab", ticker="SCHW") Company.create(name="Millennial Media", ticker="MM") Company.create(name="Freeport-McMoRan", ticker="FCX") Company.create(name="Interpublic Group", ticker="IPG") Company.create(name="Halliburton", ticker="HAL") Company.create(name="Hewlett-Packard", ticker="HPQ") Company.create(name="Transocean", ticker="RIG") Company.create(name="Johnson&Johnson", ticker="JNJ") Company.create(name="United States Steel", ticker="X") Company.create(name="Halcon Resources", ticker="HK") Company.create(name="Hertz Global Holdings", ticker="HTZ") Company.create(name="Hartford Financial Services Group", ticker="HIG") Company.create(name="Office Depot", ticker="ODP") Company.create(name="Annaly Capital Management", ticker="NLY")