def test_cache_is_cdn_with_cdn_request(self, mock_get_cdn_user_agents): """ Tests that when a CDN request is made -- i.e. from an edge node back to the origin -- that we classify the request as coming from a CDN. """ mock_get_cdn_user_agents.return_value = 'Amazon CloudFront' request_factory = RequestFactory() browser_request = request_factory.get('/fake', HTTP_USER_AGENT='Amazon CloudFront') is_from_cdn = StaticContentServer.is_cdn_request(browser_request) self.assertEqual(is_from_cdn, True)
def test_cache_is_cdn_with_normal_request(self, mock_get_cdn_user_agents): """ Tests that when a normal request is made -- i.e. from an end user with their browser -- that we don't classify the request as coming from a CDN. """ mock_get_cdn_user_agents.return_value = 'Amazon CloudFront' request_factory = RequestFactory() browser_request = request_factory.get('/fake', HTTP_USER_AGENT='Chrome 1234') is_from_cdn = StaticContentServer.is_cdn_request(browser_request) self.assertEqual(is_from_cdn, False)