class TestQuerySetMiddleware(TestCase): def setUp(self): self.middleware = QuerySetMiddleware() self.request = Mock() self.request.session = {} def test_query(self): self.request.configure_mock( **{ 'GET': { 'MODEL_PATH': 'django.contrib.auth.models.User', 'ARGS': '' } } ) john = User(username='******', email='*****@*****.**') john.save() response = self.middleware.process_request(self.request) self.assertEqual(200, response.status_code) self.assertEqual( 'john', json.loads(response.content.decode('utf-8'))[0].get('username') ) self.assertEqual( '*****@*****.**', json.loads(response.content.decode('utf-8'))[0].get('email') ) def test_empty_query(self): self.request.configure_mock( **{ 'GET': { 'MODEL_PATH': 'django.contrib.auth.models.User', 'QUERY_ARGS': '{"username": "******"}' } } ) john = User(username='******', email='*****@*****.**') john.save() response = self.middleware.process_request(self.request) self.assertEqual(200, response.status_code) self.assertEqual( 0, len(json.loads(response.content.decode('utf-8'))), 'Query result should be empty when querying for username "jane"' )
def setUp(self): self.middleware = QuerySetMiddleware() self.request = Mock() self.request.session = {}