class PrerenderIOTestMethods(TestCase):
    def setUp(self):
        self.backend = PrerenderIO()

    def test_get_response_for_url_missing_url(self):
        self.assertRaises(TypeError, self.backend.get_response_for_url)
        self.assertRaises(ValueError, self.backend.get_response_for_url, None)

    def test_get_response_for_url_valid(self):
        with HTTMock(mock_prerender_response):
            resp = self.backend.get_response_for_url("http://www.example.com")
            self.assertEqual(MOCK_RESPONSE, resp.content)
            for k, v in MOCK_RESPONSE_HEADERS.items():
                self.assertEqual(resp[k], v)

    def test_update_url_with_url_only(self):
        with HTTMock(mock_prerender_recache_response):
            resp = self.backend.update_url(url="http://www.example.com")
            self.assertEqual(resp, True)

    def test_update_url_with_regex_only(self):
        with HTTMock(mock_prerender_recache_response):
            resp = self.backend.update_url(regex="http://www.example.com/*")
            self.assertEqual(resp, True)

    def test_update_url_missing_url_and_regex(self):
        with HTTMock(mock_prerender_recache_response):
            self.assertRaises(ValueError, self.backend.update_url)
class PrerenderTimeout(TestCase):
    @override_settings(PRERENDER_TIMEOUT=5)
    def test_timeout_setting(self):
        self.backend = PrerenderIO()
        self.assertEqual(self.backend._request_kwargs({}), {'timeout': 5})

    @override_settings(PRERENDER_TIMEOUT=5)
    def test_timeout_response(self):
        self.backend = PrerenderIO()
        self.backend.session.get = MagicMock(
            side_effect=requests.exceptions.Timeout())
        resp = self.backend.get_response_for_url("http://www.example.com")
        self.assertEqual(resp.status_code, 408)
class PrerenderTimeout(TestCase):

    @override_settings(PRERENDER_TIMEOUT=5)
    def test_timeout_setting(self):
        self.backend = PrerenderIO()
        self.assertEqual(self.backend._request_kwargs({}), {'timeout': 5})

    @override_settings(PRERENDER_TIMEOUT=5)
    def test_timeout_response(self):
        self.backend = PrerenderIO()
        self.backend.session.get = MagicMock(
            side_effect=requests.exceptions.Timeout()
        )
        resp = self.backend.get_response_for_url("http://www.example.com")
        self.assertEqual(resp.status_code, 408)
 def test_timeout_response(self):
     self.backend = PrerenderIO()
     self.backend.session.get = MagicMock(
         side_effect=requests.exceptions.Timeout()
     )
     resp = self.backend.get_response_for_url("http://www.example.com")
     self.assertEqual(resp.status_code, 408)
class PrerenderIOTestToken(TestCase):
    @override_settings(PRERENDER_TOKEN=None)
    def test_get_token_throws_exception_if_missing(self):
        self.assertRaises(ValueError, PrerenderIO)

    @override_settings(PRERENDER_TOKEN="123124341adfsaf")
    def test_get_token(self):
        self.backend = PrerenderIO()
        # Test function
        self.assertEqual(self.backend._get_token(), "123124341adfsaf")
        # Test __init__
        self.assertEqual(self.backend.token, "123124341adfsaf")
 def setUp(self):
     self.backend = PrerenderIO()
 def test_get_token(self):
     self.backend = PrerenderIO()
     # Test function
     self.assertEqual(self.backend._get_token(), "123124341adfsaf")
     # Test __init__
     self.assertEqual(self.backend.token, "123124341adfsaf")
 def test_timeout_setting(self):
     self.backend = PrerenderIO()
     self.assertEqual(self.backend._request_kwargs({}), {'timeout': 5})
 def test_timeout_response(self):
     self.backend = PrerenderIO()
     self.backend.session.get = MagicMock(
         side_effect=requests.exceptions.Timeout())
     resp = self.backend.get_response_for_url("http://www.example.com")
     self.assertEqual(resp.status_code, 408)
Example #10
0
 def test_timeout_setting(self):
     self.backend = PrerenderIO()
     self.assertEqual(self.backend._request_kwargs({}), {'timeout': 5})