Example #1
0
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)
Example #2
0
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)