def test_process_view_skips_exempt(self): anything = lambda *a: a anything.mithril_exempt = True mw = WhitelistMiddleware() self.assertEqual( mw.process_view(None, anything, None, None), None )
def test_process_view_delegates_to_strategy(self): strategy = FakeStrategy() expected = random.randint(0, 10) def collect(*a, **kw): strategy.args = a strategy.kwargs = kw return expected strategy.process_view = collect mw = WhitelistMiddleware() mw.get_strategy = lambda *a: strategy result = mw.process_view(expected, expected, [], {}) self.assertEqual(result, expected) self.assertEqual(strategy.args, (expected, expected)) self.assertEqual(strategy.kwargs, {})