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")
