def test_init_no_cookies(self):
""" Test that basic values get initialized without cookies """
content = "testing 123"
resp = HttpResponse(self.mock_response, content)
self.assertEqual(resp.status_int, self.mock_response.status)
self.assertEqual(resp.status_reason, self.mock_response.reason)
self.assertEqual(resp.status, "%d %s" % (
self.mock_response.status, self.mock_response.reason))
self.assertEqual(resp.body, content)
self.assertEqual(resp.content_type, self.mock_response.get('content-type'))
self.assertEqual(resp.raw_headers, dict(self.mock_response))
def test_assert_status(self):
""" Test that assertStatus returns True if they match, False otherwise """
resp = HttpResponse(self.mock_response)
self.assertTrue(resp.assertStatus(200))
self.assertTrue(resp.assertNotStatus(400))
self.assertFalse(resp.assertStatus(400))
self.assertFalse(resp.assertNotStatus(200))
def test_assert_in_body(self):
""" Test that objects that appear in the body return True """
content = "Testing 123"
resp = HttpResponse(self.mock_response, content)
self.assertTrue(resp.assertInBody("Testing"))
self.assertTrue(resp.assertNotInBody("noteventhere"))
self.assertFalse(resp.assertNotInBody("Testing"))
self.assertFalse(resp.assertInBody("noteventhere"))
def _make_request(self, uri, method, body, headers={}):
"""
Wraps the response and content returned by :mod:`httplib2` into a
:class:`~webunit2.response.HttpResponse` object.
``uri``:
Absolute URI to the resource.
``method``:
Any supported HTTP methods defined in :rfc:`2616`.
``body``:
In the case of POST and PUT requests, this can contain the contents
of the request.
``headers``:
Dictionary of header values to be sent as part of the request.
Returns a :class:`~webunit2.response.HttpResponse` object containing
the request results.
"""
response, content = self._httpobj.request(uri,
method=method,
body=body,
headers=headers)
return HttpResponse(response, content)
def test_assert_cookie(self, mock_parse_cookies):
""" Test that checking for cookies returns the correct values """
mock_parse_cookies.return_value = {
"token": {
"token": "someval",
"httponly": True,
"secure": False,
"path": "/",
"expires": "Wed, 09 Jun 2021 10:18:14 GMT"
}
}
resp = HttpResponse(self.mock_response)
self.assertFalse(resp.assertCookie("not valid"))
self.assertFalse(resp.assertCookie("not valid", "someval"))
self.assertFalse(resp.assertCookie("not valid", "someval", attrs={"secure": False}))
self.assertFalse(resp.assertCookie("not valid", attrs={"secure": False}))
self.assertFalse(resp.assertCookie("token", "not a val", attrs={"Secure": True}))
self.assertFalse(resp.assertCookie("token", attrs={"Secure": True}))
self.assertTrue(resp.assertCookie("token"))
self.assertTrue(resp.assertCookie("token", "someval"))
self.assertTrue(resp.assertCookie("token", "someval", {"Secure": False}))
self.assertTrue(resp.assertCookie("token", attrs={"Secure": False}))
self.assertFalse(resp.assertNotCookie("token"))
self.assertFalse(resp.assertNotCookie("token", "someval"))
self.assertFalse(resp.assertNotCookie("token", "someval", {"Secure": False}))
self.assertFalse(resp.assertNotCookie("token", attrs={"Secure": False}))
self.assertTrue(resp.assertNotCookie("not valid"))
self.assertTrue(resp.assertNotCookie("not valid", "someval"))
self.assertTrue(resp.assertNotCookie("not valid", "someval", attrs={"secure": True}))
self.assertTrue(resp.assertNotCookie("not valid", attrs={"secure": True}))
self.assertTrue(resp.assertNotCookie("token", attrs={"Secure": True}))
def test_assert_header(self):
""" Test that checking for headers returns the correct values """
resp = HttpResponse(self.mock_response)
self.assertTrue(resp.assertHeader('content-type'))
self.assertTrue(resp.assertHeader('content-type', 'text/html'))
self.assertFalse(resp.assertHeader('content-type', 'plain/text'))
self.assertFalse(resp.assertHeader('not a header'))
self.assertFalse(resp.assertHeader('not a header', 'not a value'))
self.assertFalse(resp.assertNotHeader('content-type'))
self.assertFalse(resp.assertNotHeader('content-type', 'text/html'))
self.assertTrue(resp.assertNotHeader('content-type', 'plain/text'))
self.assertTrue(resp.assertNotHeader('not a header'))
self.assertTrue(resp.assertNotHeader('not a header', 'not a value'))
