def testGetReadtStats(self):
pm.update_items_from_pocket(test_items_1['test-user-1'], 'test-user-1')
items, words = pm.PocketItem.getReadStats('test-user-1',
1390586000.3,
1393995600.5)
self.assertEqual(1, items)
self.assertEqual(230, words)
def testGetReadStatusAfterUpdate(self):
pm.update_items_from_pocket(test_items_1['test-user-1'],
'test-user-1')
pm.update_items_from_pocket(test_update_1['test-user-1'],
'test-user-1')
self.assertEqual(13, len(pm.PocketItem.query().fetch(20)))
items, words = pm.PocketItem.getReadStats('test-user-1',
1390586000.3,
1395995600.5)
self.assertEqual(2, items)
self.assertEqual(2135, words)
def testGetAllStats(self):
pm.update_items_from_pocket(test_items_1['test-user-1'],
'test-user-1')
all_stats = pm.PocketItem.getAllStats('test-user-1', 1393912000)
self.assertDictEqual(all_stats,
{
'unread_items': 11,
'unread_words': 12937,
'added_items_delta': 0,
'added_words_delta': 0,
'read_items_delta': 1,
'read_words_delta': 230,
})
def testApiGetStatsData(self):
pm.update_items_from_pocket(test_items_1['test-user-1'], 'test-user-1')
resp = self.testapp.post_json(
'/_ah/spi/PocketMonApi.get_stats',
{'timestamp': 1393912000,})
self.assertDictEqual(resp.json,
{
'unread_items': '11',
'unread_words': '12937',
'added_items_delta': '0',
'added_words_delta': '0',
'read_items_delta': '1',
'read_words_delta': '230',
})
def testUpdateItemsFromPocketApiDict(self):
for username, items_dict in test_items_1.iteritems():
pm.update_items_from_pocket(items_dict, username)
self.assertEqual(13,
len(pm.PocketItem.queryUser('test-user-1').fetch(20)))
