def setUp(self):
super(TestPlayStats, self).setUp()
dbconfig['lastfm.api_key'] = 'SEKRET_LASTFM_KEY'
PlayCountSnapshot(track_id='1',
play_count=5,
artist_name='Tame Impala',
album_title='Lonerism',
label='label1').put()
PlayCountSnapshot(track_id='1',
play_count=5,
artist_name='Tame Impala',
album_title='Lonerism',
label='label1').put()
PlayCountSnapshot(track_id='2',
play_count=20,
artist_name='Taken By Trees',
album_title='Dreams',
label='label2').put()
def test_this_week(self):
p = PlayCountSnapshot(
track_id='3',
play_count=1,
artist_name='Ignore Me',
album_title='Inore This',
label='...',
# This was entered older than a week.
established=datetime.now() - timedelta(days=8))
p.put()
res = self.request('/api/stats')
weekly = res['this_week']['releases']
eq_(weekly[0]['artist'], 'Taken By Trees')
eq_(weekly[0]['release'], 'Dreams')
eq_(weekly[0]['label'], 'label2')
eq_(weekly[0]['play_count'], 20)
assert 'id' in weekly[0]
eq_(weekly[1]['artist'], 'Tame Impala')
eq_(weekly[1]['release'], 'Lonerism')
eq_(weekly[1]['label'], 'label1')
eq_(weekly[1]['play_count'], 5)
eq_(len(weekly), 2, weekly)
