Python HeaderControlMiddleware Examples

Example #1

File: test_middleware.py Project: jlrivera81/incr-228

 def setUp(self):
     super(TestHeaderControlMiddlewareProcessResponse, self).setUp()
     self.middleware = HeaderControlMiddleware()

Example #2

File: test_middleware.py Project: jlrivera81/incr-228

class TestHeaderControlMiddlewareProcessResponse(TestCase):
    """Test the `header_control` middleware. """
    def setUp(self):
        super(TestHeaderControlMiddlewareProcessResponse, self).setUp()
        self.middleware = HeaderControlMiddleware()

    def test_doesnt_barf_if_not_modifying_anything(self):
        fake_request = HttpRequest()

        fake_response = HttpResponse()
        fake_response['Vary'] = 'Cookie'
        fake_response['Accept-Encoding'] = 'gzip'

        result = self.middleware.process_response(fake_request, fake_response)
        self.assertEquals('Cookie', result['Vary'])
        self.assertEquals('gzip', result['Accept-Encoding'])

    def test_doesnt_barf_removing_nonexistent_headers(self):
        fake_request = HttpRequest()

        fake_response = HttpResponse()
        fake_response['Vary'] = 'Cookie'
        fake_response['Accept-Encoding'] = 'gzip'
        remove_headers_from_response(fake_response, 'Vary', 'FakeHeaderWeeee')

        result = self.middleware.process_response(fake_request, fake_response)
        self.assertNotIn('Vary', result)
        self.assertEquals('gzip', result['Accept-Encoding'])

    def test_removes_intended_headers(self):
        fake_request = HttpRequest()

        fake_response = HttpResponse()
        fake_response['Vary'] = 'Cookie'
        fake_response['Accept-Encoding'] = 'gzip'
        remove_headers_from_response(fake_response, 'Vary')

        result = self.middleware.process_response(fake_request, fake_response)
        self.assertNotIn('Vary', result)
        self.assertEquals('gzip', result['Accept-Encoding'])

    def test_forces_intended_header(self):
        fake_request = HttpRequest()

        fake_response = HttpResponse()
        fake_response['Vary'] = 'Cookie'
        fake_response['Accept-Encoding'] = 'gzip'
        force_header_for_response(fake_response, 'Vary', 'Origin')

        result = self.middleware.process_response(fake_request, fake_response)
        self.assertEquals('Origin', result['Vary'])
        self.assertEquals('gzip', result['Accept-Encoding'])

    def test_does_not_mangle_undecorated_response(self):
        fake_request = HttpRequest()

        fake_response = HttpResponse()
        fake_response['Vary'] = 'Cookie'
        fake_response['Accept-Encoding'] = 'gzip'

        result = self.middleware.process_response(fake_request, fake_response)
        self.assertEquals('Cookie', result['Vary'])
        self.assertEquals('gzip', result['Accept-Encoding'])

Example #3

 def setUp(self):
     super().setUp()
     self.middleware = HeaderControlMiddleware()

Example #4

File: test_middleware.py Project: sliva/edx-platform

 def setUp(self):
     super(TestHeaderControlMiddlewareProcessResponse, self).setUp()  # lint-amnesty, pylint: disable=super-with-arguments
     self.middleware = HeaderControlMiddleware()

Example #5

File: test_middleware.py Project: Colin-Fredericks/edx-platform

class TestHeaderControlMiddlewareProcessResponse(TestCase):
    """Test the `header_control` middleware. """

    def setUp(self):
        super(TestHeaderControlMiddlewareProcessResponse, self).setUp()
        self.middleware = HeaderControlMiddleware()

    def test_doesnt_barf_if_not_modifying_anything(self):
        fake_request = HttpRequest()

        fake_response = HttpResponse()
        fake_response["Vary"] = "Cookie"
        fake_response["Accept-Encoding"] = "gzip"

        result = self.middleware.process_response(fake_request, fake_response)
        self.assertEquals("Cookie", result["Vary"])
        self.assertEquals("gzip", result["Accept-Encoding"])

    def test_doesnt_barf_removing_nonexistent_headers(self):
        fake_request = HttpRequest()

        fake_response = HttpResponse()
        fake_response["Vary"] = "Cookie"
        fake_response["Accept-Encoding"] = "gzip"
        remove_headers_from_response(fake_response, "Vary", "FakeHeaderWeeee")

        result = self.middleware.process_response(fake_request, fake_response)
        self.assertNotIn("Vary", result)
        self.assertEquals("gzip", result["Accept-Encoding"])

    def test_removes_intended_headers(self):
        fake_request = HttpRequest()

        fake_response = HttpResponse()
        fake_response["Vary"] = "Cookie"
        fake_response["Accept-Encoding"] = "gzip"
        remove_headers_from_response(fake_response, "Vary")

        result = self.middleware.process_response(fake_request, fake_response)
        self.assertNotIn("Vary", result)
        self.assertEquals("gzip", result["Accept-Encoding"])

    def test_forces_intended_header(self):
        fake_request = HttpRequest()

        fake_response = HttpResponse()
        fake_response["Vary"] = "Cookie"
        fake_response["Accept-Encoding"] = "gzip"
        force_header_for_response(fake_response, "Vary", "Origin")

        result = self.middleware.process_response(fake_request, fake_response)
        self.assertEquals("Origin", result["Vary"])
        self.assertEquals("gzip", result["Accept-Encoding"])

    def test_does_not_mangle_undecorated_response(self):
        fake_request = HttpRequest()

        fake_response = HttpResponse()
        fake_response["Vary"] = "Cookie"
        fake_response["Accept-Encoding"] = "gzip"

        result = self.middleware.process_response(fake_request, fake_response)
        self.assertEquals("Cookie", result["Vary"])
        self.assertEquals("gzip", result["Accept-Encoding"])

Example #6

class TestHeaderControlMiddlewareProcessResponse(TestCase):
    """Test the `header_control` middleware. """
    def setUp(self):
        super(TestHeaderControlMiddlewareProcessResponse, self).setUp()  # lint-amnesty, pylint: disable=super-with-arguments
        self.middleware = HeaderControlMiddleware()

    def test_doesnt_barf_if_not_modifying_anything(self):
        fake_request = HttpRequest()

        fake_response = HttpResponse()
        fake_response['Vary'] = 'Cookie'
        fake_response['Accept-Encoding'] = 'gzip'

        result = self.middleware.process_response(fake_request, fake_response)
        assert 'Cookie' == result['Vary']
        assert 'gzip' == result['Accept-Encoding']

    def test_doesnt_barf_removing_nonexistent_headers(self):
        fake_request = HttpRequest()

        fake_response = HttpResponse()
        fake_response['Vary'] = 'Cookie'
        fake_response['Accept-Encoding'] = 'gzip'
        remove_headers_from_response(fake_response, 'Vary', 'FakeHeaderWeeee')

        result = self.middleware.process_response(fake_request, fake_response)
        assert 'Vary' not in result
        assert 'gzip' == result['Accept-Encoding']

    def test_removes_intended_headers(self):
        fake_request = HttpRequest()

        fake_response = HttpResponse()
        fake_response['Vary'] = 'Cookie'
        fake_response['Accept-Encoding'] = 'gzip'
        remove_headers_from_response(fake_response, 'Vary')

        result = self.middleware.process_response(fake_request, fake_response)
        assert 'Vary' not in result
        assert 'gzip' == result['Accept-Encoding']

    def test_forces_intended_header(self):
        fake_request = HttpRequest()

        fake_response = HttpResponse()
        fake_response['Vary'] = 'Cookie'
        fake_response['Accept-Encoding'] = 'gzip'
        force_header_for_response(fake_response, 'Vary', 'Origin')

        result = self.middleware.process_response(fake_request, fake_response)
        assert 'Origin' == result['Vary']
        assert 'gzip' == result['Accept-Encoding']

    def test_does_not_mangle_undecorated_response(self):
        fake_request = HttpRequest()

        fake_response = HttpResponse()
        fake_response['Vary'] = 'Cookie'
        fake_response['Accept-Encoding'] = 'gzip'

        result = self.middleware.process_response(fake_request, fake_response)
        assert 'Cookie' == result['Vary']
        assert 'gzip' == result['Accept-Encoding']