def test_returned_headers(self):
"""Set some headers in the mock and check they are passed back"""
# Arrange
mock = MockResponse()
mock.headers = dict(a=1, b=2)
url = ""
self.backend.responses.add(mock, url)
# Act
self.browser.go(url)
# Assert
self.assertEqual(self.browser.headers, mock.headers)
def mocked_visit(self):
mock = MockResponse()
mock.src = "test data"
mock.http_code = 123
mock.headers = {}
url = ""
method = 'GET'
data = None
headers = None
follow = None
agent = None
retries = None
debug = None
self.backend.responses.add(mock, url, method, data, headers)
self.browser.go(url, method, data, headers, follow, agent, retries, debug)
return url, mock
