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