def setUp(self):
     self.factory = RequestFactory()
     self.middleware = RequestResponseLoggingMiddleware()
 def setUp(self):
     self.factory = RequestFactory()
     self.middleware = RequestResponseLoggingMiddleware()
class TestMiddleware(TestCase):

    def setUp(self):
        self.factory = RequestFactory()
        self.middleware = RequestResponseLoggingMiddleware()

    def test_simple_url_is_logged(self):
        request = self.factory.get('/api/simpleurl/')
        response = HttpResponse(status=200, content='<html>Hello</html>')
        result = self.middleware.process_response(
            request,
            response=response
        )
        assert result == response

        request = self.factory.get('/api/v2/')
        response = HttpResponse(status=200, content='<html>API</html>')
        result = self.middleware.process_response(
            request,
            response=response
        )
        assert result == response
        self.assertQuerysetEqual(Entry.objects.all(), ['<Log 200 "/api/simpleurl/">',
                                                       '<Log 200 "/api/v2/">'], ordered=False)

    def test_other_urls_are_ignored(self):
        request = self.factory.get('/')
        self.middleware.process_response(
            request,
            response=HttpResponse(status=200, content='<html>Hello</html>')
        )
        self.assertQuerysetEqual(Entry.objects.all(), [])

    def test_get_url_is_recorded(self):
        request = self.factory.get('/login/?utm_source=spam')
        out_result = self.middleware.process_response(
            request,
            response=HttpResponse(status=200, content='<html>Hello</html>')
        )

        self.assertTrue(type(out_result) is HttpResponse)
        self.assertQuerysetEqual(Entry.objects.all(), ['<Log 200 "/login/?utm_source=spam">'])
        log = Entry.objects.all()[0]

        self.assertEqual(log.url, '/login/?utm_source=spam')
        self.assertEqual(log.method, 'GET')
        self.assertEqual(log.status_code, 200)
        self.assertIn('SERVER_PROTOCOL: HTTP/1.1', log.in_headers)
        self.assertIn('Content-Type: text/html; charset=utf-8', log.out_headers)
        self.assertEqual(log.ip, '127.0.0.1')
        self.assertEqual(log.app_name, 'auth')
        self.assertEqual(log.request, '')
        self.assertEqual(log.response, '<html>Hello</html>')
        self.assertLessEqual(log.created_at, datetime.datetime.now())
        self.assertLessEqual(log.updated_at, datetime.datetime.now())

    def test_post_url_is_recorded(self):
        request = self.factory.post('/api/blog/post/create/', {'foo': u'бар'})
        self.middleware.process_response(
            request,
            response=HttpResponse(status=201, content='<html>Created</html>')
        )

        self.assertQuerysetEqual(Entry.objects.all(), ['<Log 201 "/api/blog/post/create/">'])
        log = Entry.objects.all()[0]

        self.assertEqual(log.url, '/api/blog/post/create/')
        self.assertEqual(log.app_name, '')
        self.assertEqual(log.method, 'POST')
        self.assertEqual(log.status_code, 201)
        self.assertEqual(log.data, u'foo: бар')
        self.assertEqual(log.response, '<html>Created</html>')

    def test_put_url_is_recorded(self):
        request = self.factory.put('/api/json_handle/',
                                   content_type='application/json',
                                   data=json.dumps({'text': u'Привет, котики!',
                                                    'is_cool': None}))
        self.middleware.process_response(
            request,
            response=HttpResponse(status=404, content='<html>404</html>')
        )

        self.assertQuerysetEqual(Entry.objects.all(), ['<Log 404 "/api/json_handle/">'])
        log = Entry.objects.all()[0]

        self.assertEqual(log.url, '/api/json_handle/')
        self.assertEqual(log.method, 'PUT')
        self.assertEqual(log.status_code, 404)
        assert set(log.data.split('\n')) == set((u"text: Привет, котики!", u"is_cool: None"))
        self.assertEqual(log.response, '<html>404</html>')

    def test_delete_url_is_recorded(self):
        request = self.factory.delete('/api/image/',)
        self.middleware.process_response(
            request,
            response=HttpResponse(status=204, content=u'Удалено')
        )

        self.assertQuerysetEqual(Entry.objects.all(), ['<Log 204 "/api/image/">'])
        log = Entry.objects.all()[0]

        self.assertEqual(log.url, '/api/image/')
        self.assertEqual(log.method, 'DELETE')
        self.assertEqual(log.status_code, 204)
        self.assertEqual(log.response, u'Удалено')

    def test_binary_files_not_saved(self):
        with open(os.path.join('test_app', 'django-logo-positive.png'), 'rb') as fp:
            request = self.factory.post('/api/editor/42/image/', {'hello': 'dolly', 'file': fp})
        self.middleware.process_response(
            request,
            response=HttpResponse(status=200, content='')
        )
        log = Entry.objects.all()[0]
        self.assertEqual(log.request, '')
        assert set(log.data.split('\n')) == set((u"hello: dolly", u"file: django-logo-positive.png"))

    def test_stream_requests_dont_break_logging(self):
        request = self.factory.post('/api/dummy/',
                                    json.dumps({'hello': 'dolly'}),
                                    content_type='application/json')
        request._read_started = True  # emulating stream request
        self.middleware.process_response(
            request,
            response=HttpResponse(status=200, content='')
        )
        self.assertQuerysetEqual(Entry.objects.all(), ['<Log 200 "/api/dummy/">'])
        log = Entry.objects.all()[0]
        self.assertEqual(log.request, "--can't-be-loggged--")
        self.assertEqual(log.data, "")
class TestMiddleware(TestCase):
    def setUp(self):
        self.factory = RequestFactory()
        self.middleware = RequestResponseLoggingMiddleware()

    def test_simple_url_is_logged(self):
        request = self.factory.get('/api/simpleurl/')
        response = HttpResponse(status=200, content='<html>Hello</html>')
        result = self.middleware.process_response(request, response=response)
        assert result == response

        request = self.factory.get('/api/v2/')
        response = HttpResponse(status=200, content='<html>API</html>')
        result = self.middleware.process_response(request, response=response)
        assert result == response
        self.assertQuerysetEqual(
            Entry.objects.all(),
            ['<Log 200 "/api/simpleurl/">', '<Log 200 "/api/v2/">'],
            ordered=False)

    def test_other_urls_are_ignored(self):
        request = self.factory.get('/')
        self.middleware.process_response(request,
                                         response=HttpResponse(
                                             status=200,
                                             content='<html>Hello</html>'))
        self.assertQuerysetEqual(Entry.objects.all(), [])

    def test_get_url_is_recorded(self):
        request = self.factory.get('/login/?utm_source=spam')
        out_result = self.middleware.process_response(
            request,
            response=HttpResponse(status=200, content='<html>Hello</html>'))

        self.assertTrue(type(out_result) is HttpResponse)
        self.assertQuerysetEqual(Entry.objects.all(),
                                 ['<Log 200 "/login/?utm_source=spam">'])
        log = Entry.objects.all()[0]

        self.assertEqual(log.url, '/login/?utm_source=spam')
        self.assertEqual(log.method, 'GET')
        self.assertEqual(log.status_code, 200)
        self.assertIn('SERVER_PROTOCOL: HTTP/1.1', log.in_headers)
        self.assertIn('Content-Type: text/html; charset=utf-8',
                      log.out_headers)
        self.assertEqual(log.ip, '127.0.0.1')
        self.assertEqual(log.app_name, 'auth')
        self.assertEqual(log.request, '')
        self.assertEqual(log.response, '<html>Hello</html>')
        self.assertLessEqual(log.created_at, datetime.datetime.now())
        self.assertLessEqual(log.updated_at, datetime.datetime.now())

    def test_post_url_is_recorded(self):
        request = self.factory.post('/api/blog/post/create/', {'foo': u'бар'})
        self.middleware.process_response(request,
                                         response=HttpResponse(
                                             status=201,
                                             content='<html>Created</html>'))

        self.assertQuerysetEqual(Entry.objects.all(),
                                 ['<Log 201 "/api/blog/post/create/">'])
        log = Entry.objects.all()[0]

        self.assertEqual(log.url, '/api/blog/post/create/')
        self.assertEqual(log.app_name, '')
        self.assertEqual(log.method, 'POST')
        self.assertEqual(log.status_code, 201)
        self.assertEqual(log.data, u'foo: бар')
        self.assertEqual(log.response, '<html>Created</html>')

    def test_put_url_is_recorded(self):
        request = self.factory.put('/api/json_handle/',
                                   content_type='application/json',
                                   data=json.dumps({
                                       'text': u'Привет, котики!',
                                       'is_cool': None
                                   }))
        self.middleware.process_response(request,
                                         response=HttpResponse(
                                             status=404,
                                             content='<html>404</html>'))

        self.assertQuerysetEqual(Entry.objects.all(),
                                 ['<Log 404 "/api/json_handle/">'])
        log = Entry.objects.all()[0]

        self.assertEqual(log.url, '/api/json_handle/')
        self.assertEqual(log.method, 'PUT')
        self.assertEqual(log.status_code, 404)
        assert set(log.data.split('\n')) == set(
            (u"text: Привет, котики!", u"is_cool: None"))
        self.assertEqual(log.response, '<html>404</html>')

    def test_delete_url_is_recorded(self):
        request = self.factory.delete('/api/image/', )
        self.middleware.process_response(request,
                                         response=HttpResponse(
                                             status=204, content=u'Удалено'))

        self.assertQuerysetEqual(Entry.objects.all(),
                                 ['<Log 204 "/api/image/">'])
        log = Entry.objects.all()[0]

        self.assertEqual(log.url, '/api/image/')
        self.assertEqual(log.method, 'DELETE')
        self.assertEqual(log.status_code, 204)
        self.assertEqual(log.response, u'Удалено')

    def test_binary_files_not_saved(self):
        with open(os.path.join('test_app', 'django-logo-positive.png'),
                  'rb') as fp:
            request = self.factory.post('/api/editor/42/image/', {
                'hello': 'dolly',
                'file': fp
            })
        self.middleware.process_response(request,
                                         response=HttpResponse(status=200,
                                                               content=''))
        log = Entry.objects.all()[0]
        self.assertEqual(log.request, '')
        assert set(log.data.split('\n')) == set(
            (u"hello: dolly", u"file: django-logo-positive.png"))

    def test_stream_requests_dont_break_logging(self):
        request = self.factory.post('/api/dummy/',
                                    json.dumps({'hello': 'dolly'}),
                                    content_type='application/json')
        request._read_started = True  # emulating stream request
        self.middleware.process_response(request,
                                         response=HttpResponse(status=200,
                                                               content=''))
        self.assertQuerysetEqual(Entry.objects.all(),
                                 ['<Log 200 "/api/dummy/">'])
        log = Entry.objects.all()[0]
        self.assertEqual(log.request, "--can't-be-loggged--")
        self.assertEqual(log.data, "")