class TestStatsWithMondoDB(unittest.TestCase):
def setUp(self):
self.db = MongoDB(col = 'stats_test')
self.stats = Statistics(self.db)
mock = MockDatetime()
self.expected_date = mock.now()
self.stats._datetime = mock
def test_save_stat(self):
json_doc = {'url':"url", 'datetime': "2014-01-01 14:10:00" }
self.stats.register_access_now("url")
result = self.db.find_one_by('url', "url")
self.assertEqual(result['url'], "url")
self.assertEqual(result['datetime'], str(self.expected_date))
self.db.drop('stats_test')
def test_all_visits(self):
self.stats.register_access_now("url")
visits = self.stats.all_visits()
self.assertEqual(len(visits), 1)
def test_all_visits_with_statis_without_ip(self):
db = mock()
when(db).find_all().thenReturn([{'url':"/", 'datetime': "2014/02/02 14:53"}])
stats = Statistics(db)
visits = stats.all_visits()
self.assertEqual(len(visits), 1)
self.assertEquals("2014/02/02 14:53", visits[0].access_datetime)
self.assertEqual('/', visits[0].url)
def setUp(self):
self.db = MongoDB(col = 'stats_test')
self.stats = Statistics(self.db)
mock = MockDatetime()
self.expected_date = mock.now()
self.stats._datetime = mock
class TestStatistics(unittest.TestCase):
def setUp(self):
self.mock_date = MockDatetime()
self.mock_date.set_datetime("01/01/2014", "12:00")
self.stats = Statistics(MockMongo())
self.stats._datetime = self.mock_date
def test_by_default_statistics_uses_a_datetime(self):
stats = Statistics(None)
self.assertIs(stats._datetime, datetime.datetime)
def test_store_and_retrive_visit(self):
self.stats.register_access_now('/', "0.0.0.0")
visits = self.stats.all_visits()
self.assertEqual(len(visits), 1)
self.assertEquals(str(self.mock_date.dt), visits[0].access_datetime)
self.assertEqual('/', visits[0].url)
def test_all_visits_with_statis_without_ip(self):
db = mock()
when(db).find_all().thenReturn([{'url':"/", 'datetime': "2014/02/02 14:53"}])
stats = Statistics(db)
visits = stats.all_visits()
self.assertEqual(len(visits), 1)
self.assertEquals("2014/02/02 14:53", visits[0].access_datetime)
self.assertEqual('/', visits[0].url)
def test_when_a_visit_comes_from_a_ignore_ip_dont_store_it(self):
self.stats.add_ignore_ip('ignore')
self.stats.register_access_now("url", "ignore")
visits = self.stats.all_visits()
self.assertEqual(len(visits), 0)
def test_when_a_visit_comes_from_one_of_ignore_ips_dont_store_it(self):
self.stats.add_ignore_ip('ignore01')
self.stats.add_ignore_ip('ignore02')
self.stats.register_access_now("ignore01", "ignore01")
self.stats.register_access_now("ignore02", "ignore02")
self.stats.register_access_now("no_ignore", "no_ignore")
visits = self.stats.all_visits()
self.assertEqual(len(visits), 1)
self.assertEqual(visits[0].url, "no_ignore")
def setUp(self):
self.mock_date = MockDatetime()
self.mock_date.set_datetime("01/01/2014", "12:00")
self.stats = Statistics(MockMongo())
self.stats._datetime = self.mock_date
def create_stats():
mockMongo = MockMongo()
stats = Statistics(mockMongo)
stats._datetime = MockDatetime(None)
return stats
