def test_duplicate_companies(self):
links = scrape_reddit("AAPL", str(ticker_to_name(get_companies(), "SUNE")))
save_reddit_articles(links)
instance1 = Message.objects.all()
save_reddit_articles(links)
instance2 = Message.objects.all()
self.assertEqual(instance1[0], instance2[0])
self.assertEqual(instance1[1], instance2[1])
def test_reddit_save(self):
links = scrape_reddit("AAPL", str(ticker_to_name(get_companies(), "SUNE")))
expected = {
'url': 'http://www.reddit.com/r/investing/comments/40wx7b/sunedison_inc_to_distribute_tesla_motors_inc/?ref=search_posts',
'urls': ['https://www.reddit.com/r/investing/comments/40wx7b/sunedison_inc_to_distribute_tesla_motors_inc/'],
'social_id': '40wx7b',
'created_time': datetime.datetime.utcfromtimestamp(1452764067.0).strftime("%Y-%m-%dT%H:%M:%SZ"),
'content': 'Sunedison Inc To Distribute Tesla Motors Inc Powerwall',
'author': 'MartEden',
'popularity': 27,
'source': 'reddit',
'author_image': 'https://www.redditstatic.com/icon-touch.png',
'focus': 'AAPL',
'symbols': ['AAPL']
}
save_reddit_articles(links)
dbobj = Message.objects.get(social_id=expected['social_id'])
self.assertEqual(dbobj.url, str(expected['url']))
self.assertEqual(dbobj.content, str(expected['content']))
self.assertEqual(dbobj.author, str(expected['author']))
self.assertEqual(dbobj.focus, 'AAPL')
def test_invalid_ticker_type(self):
companies = get_companies()
with self.assertRaises(ValueError):
ticker_to_name(companies, ("ticker", ))
def test_ticker_to_name_bad_ticker(self):
companies = get_companies()
with self.assertRaises(ValueError):
ticker_to_name(companies, "ayyooo")
def test_ticker_to_name_bad_company_data(self):
with self.assertRaises(ValueError):
ticker_to_name("waffles", "MSFT")
def test_ticker_to_name_works(self):
companies = get_companies()
expected = "Microsoft Corporation"
self.assertEqual(ticker_to_name(companies, "MSFT"), expected)