class FetchSnippetsMiddlewareTests(TestCase): def setUp(self): self.middleware = FetchSnippetsMiddleware() @patch('snippets.base.middleware.resolve') @patch('snippets.base.middleware.fetch_snippets') def test_resolve_match(self, fetch_snippets, resolve): """ If resolve returns a match to the fetch_snippets view, return the result of the view. """ request = Mock() result = resolve.return_value result.func = fetch_snippets result.args = (1, 'asdf') result.kwargs = {'blah': 5} eq_(self.middleware.process_request(request), fetch_snippets.return_value) fetch_snippets.assert_called_with(request, 1, 'asdf', blah=5) @patch('snippets.base.middleware.resolve') @patch('snippets.base.middleware.fetch_snippets') def test_resolve_no_match(self, fetch_snippets, resolve): """ If resolve doesn't return a match to the fetch_snippets view, return None. """ request = Mock() result = resolve.return_value result.func = lambda request: 'asdf' eq_(self.middleware.process_request(request), None)
class FetchSnippetsMiddlewareTests(TestCase): def setUp(self): self.middleware = FetchSnippetsMiddleware() @patch('snippets.base.middleware.resolve') @patch('snippets.base.middleware.fetch_snippets') def test_resolve_fetch_snippets_match(self, fetch_snippets, resolve): """ If resolve returns a match to the fetch_snippets view, return the result of the view. """ request = Mock() result = resolve.return_value result.func = fetch_snippets result.args = (1, 'asdf') result.kwargs = {'blah': 5} self.assertEqual(self.middleware.process_request(request), fetch_snippets.return_value) fetch_snippets.assert_called_with(request, 1, 'asdf', blah=5) @patch('snippets.base.middleware.resolve') @patch('snippets.base.middleware.fetch_json_snippets') def test_resolve_fetch_json_snippets_match(self, fetch_json_snippets, resolve): """ If resolve returns a match to the fetch_json_snippets view, return the result of the view. """ request = Mock() result = resolve.return_value result.func = fetch_json_snippets result.args = (1, 'asdf') result.kwargs = {'blah': 5} self.assertEqual(self.middleware.process_request(request), fetch_json_snippets.return_value) fetch_json_snippets.assert_called_with(request, 1, 'asdf', blah=5) @patch('snippets.base.middleware.resolve') @patch('snippets.base.middleware.fetch_snippets') def test_resolve_no_match(self, fetch_snippets, resolve): """ If resolve doesn't return a match to the fetch_snippets view, return None. """ request = Mock() result = resolve.return_value result.func = lambda request: 'asdf' self.assertEqual(self.middleware.process_request(request), None) def test_unknown_url(self): request = RequestFactory().get('/admin') self.assertEqual(self.middleware.process_request(request), None)
def setUp(self): self.middleware = FetchSnippetsMiddleware()
def setUp(self): self.get_response_mock = Mock() self.middleware = FetchSnippetsMiddleware(self.get_response_mock)