class TrafficTest(TestCase): def setUp(self): self.admin = RequestAdmin(Request, site) self.factory = RequestFactory() def test_traffic(self): request = self.factory.get('/foo') self.admin.traffic(request) @override_settings(USE_I18N=False) def test_traffic_without_i18n(self): del _trans.gettext request = self.factory.get('/foo') self.admin.traffic(request) def test_bad_days(self): request = self.factory.get('/foo', {'days': 'foo'}) self.admin.traffic(request) def test_days(self): for days, intervals in ((9, 10), (30, 16), (365, 13)): request = self.factory.get('/foo', {'days': days}) data = json.loads(self.admin.traffic(request).content.decode()) for single_data in data: self.assertEqual(len(single_data['data']), intervals)
class TrafficTest(TestCase): def setUp(self): self.admin = RequestAdmin(Request, site) self.factory = RequestFactory() def test_traffic(self): request = self.factory.get('/foo') self.admin.traffic(request) def test_bad_days(self): request = self.factory.get('/foo', {'days': 'foo'}) self.admin.traffic(request) def test_days_lt_10(self): request = self.factory.get('/foo', {'days': 9}) self.admin.traffic(request) def test_days_gt_60(self): request = self.factory.get('/foo', {'days': 61}) self.admin.traffic(request)