def test_make_request_invalid_method(self, m_requests, m_sleep): """Make a request using an invalid http method.""" request = testing.DummyRequest({mut.URL_KEY: SAMPLE_URL, mut.METHOD_KEY: INVALID_METHOD}) try: mut.make_request(request) self.fail("Should have raised KeyError.") except KeyError as e: self.assertEquals(mut.INVALID_METHOD, e.args[0])
def test_make_request(self, m_requests, m_sleep): """Make a simple request.""" request = testing.DummyRequest({mut.URL_KEY: SAMPLE_URL}) m_response, response_dict = self.mock_response() m_requests.get.return_value = m_response self.assertEqual(response_dict, mut.make_request(request)) m_requests.get.assert_called_with(url=SAMPLE_URL) m_sleep.assert_called_with(mut.SECURITY_SLEEP)
def test_make_request_method(self, m_requests, m_sleep): """Make a request using an alternate http method.""" request = testing.DummyRequest({mut.URL_KEY: SAMPLE_URL, mut.METHOD_KEY: SAMPLE_METHOD}) m_response, response_dict = self.mock_response() m_requests.post.return_value = m_response self.assertEqual(response_dict, mut.make_request(request)) m_requests.post.assert_called_with(url=SAMPLE_URL) m_sleep.assert_called_with(mut.SECURITY_SLEEP)
def test_make_request_headers(self, m_requests, m_sleep): """Make a simple request with parameters.""" request = testing.DummyRequest({mut.URL_KEY: SAMPLE_URL, mut.HEADERS_KEY: json.dumps(SAMPLE_REQUEST_HEADERS)}) m_response, response_dict = self.mock_response() m_requests.get.return_value = m_response self.assertEqual(response_dict, mut.make_request(request)) m_requests.get.assert_called_with(url=SAMPLE_URL, headers=SAMPLE_REQUEST_HEADERS) m_sleep.assert_called_with(mut.SECURITY_SLEEP)
def test_make_request_charset(self, m_requests, m_sleep): """Make a request with a specific character set in the response.""" request = testing.DummyRequest({mut.URL_KEY: SAMPLE_URL}) m_response, response_dict = self.mock_response(headers=CHARSET_HEADERS, body=mock.Mock()) m_requests.get.return_value = m_response m_response.content.decode.return_value = SAMPLE_BODY response_dict["body"] = SAMPLE_BODY #Not very elegant, but it's an edge case self.assertEqual(response_dict, mut.make_request(request)) m_requests.get.assert_called_with(url=SAMPLE_URL) m_response.content.decode.assert_called_with(CHARSET) m_sleep.assert_called_with(mut.SECURITY_SLEEP)
def test_make_request(self): """Do it!""" request = testing.DummyRequest(params={"url": SAMPLE_URL}) response = mut.make_request(request) status = response["status"] headers = response["headers"] body = response["body"] self.assertEquals(SAMPLE_STATUS, status) self.assertEquals(SAMPLE_BODY, body) for key, value in SAMPLE_HEADERS.items(): self.assert_(key in headers) self.assertEquals(value, headers[key])