コード例 #1
0
ファイル: strategy.py プロジェクト: strycore/django-mithril
    def test_super_user_exempt(self):
        """
            Make sure that superuser is exempt from IP whitelists.
        """

        expected = '10.112.12.12'
        req = self.fake_request(expected)
        req.user.is_superuser = False

        whitelist = Whitelist.objects.create(name='asdf', slug='asdf')
        whitelist.range_set.create(ip='127.0.0.1', cidr=32)

        strat = Strategy()
        strat.return_one = lambda *a, **kw: whitelist.pk
        strat.actions = [['return_one', 'pk']]

        self.assertTrue(
                isinstance(strat.process_view(req, None, (), {}),
                    HttpResponseForbidden)
        )

        req.user.is_superuser = True

        self.assertEqual(
            strat.process_view(req, None, (), {}),
            None
        )
コード例 #2
0
ファイル: strategy.py プロジェクト: strycore/django-mithril
    def test_process_view_passes_curried_mithril_reset_view(self):
        expected = fmt_ip(random.randint(1, 0xFFFFFFFF)) 
        req = self.fake_request(expected)

        strat = Strategy()

        wl = Whitelist.objects.create(
            name='okay',
            slug='okay'
        )

        wl.range_set.create(
            ip='0.0.0.0',
            cidr=32
        )

        strat.return_one = lambda *a, **kw: wl.pk
        strat.actions = [['return_one', 'pk']]

        view = lambda:None
        view.mithril_reset = \
            lambda request, view, args, kwargs: check(
                    request, view, args, kwargs)

        def check(request, target_view, target_args, target_kwargs):
            self.assertTrue(request is req)
            self.assertEqual(target_view, view)
            return expected

        self.assertEqual(
            strat.process_view(req, view, (), {}),
            expected
        )
コード例 #3
0
ファイル: strategy.py プロジェクト: strycore/django-mithril
    def test_process_view_skips_lookups_raising_fielderror(self):
        expected = fmt_ip(random.randint(1, 0xFFFFFFFF)) 
        req = self.fake_request(expected)

        strat = Strategy()

        strat.return_one = lambda *a, **kw: 1
        strat.actions = [['return_one', 'anything']]

        self.assertEqual(
            strat.process_view(req, None, (), {}),
            None
        )
コード例 #4
0
ファイル: strategy.py プロジェクト: gmcquillan/django-mithril
    def test_process_view_skips_methods_returning_no_whitelists(self):
        expected = fmt_ip(random.randint(1, 0xFFFFFFFF)) 
        req = self.fake_request(expected)

        strat = Strategy()

        # there are no whitelists at this point,
        # so a 'pk' lookup will return nothing!
        strat.return_one = lambda *a, **kw: 1
        strat.actions = [['return_one', 'pk']]

        self.assertEqual(
            strat.process_view(req, None, (), {}),
            None
        )
コード例 #5
0
ファイル: strategy.py プロジェクト: strycore/django-mithril
    def test_process_view_passes_forbidden_response_class(self):
        expected = fmt_ip(random.randint(1, 0xFFFFFFFF)) 
        req = self.fake_request(expected)

        strat = Strategy()

        wl = Whitelist.objects.create(
            name='okay',
            slug='okay'
        )

        wl.range_set.create(
            ip='0.0.0.0',
            cidr=32
        )

        strat.return_one = lambda *a, **kw: wl.pk
        strat.actions = [['return_one', 'pk']]
        self.assertTrue(
            isinstance(strat.process_view(req, None, (), {}),
                HttpResponseForbidden)
        )