def test_get_url_response_time_almostok(self, mock_requests): # Mock correct response classinst = Measurement("http://example.com", 5) mock_requests.return_value.status_code = 503 mock_requests.return_value.headers = {'Date': 'Wed, 31 Mar 2021 08:47:43 GMT'} mock_requests.return_value.text = "Mocked request response gateway timeout" ret = classinst.get_url_response_time() metricsDict = dict() metricsDict['url'] = "http://example.com" metricsDict['status_code'] = 503 metricsDict['receivedtext'] = "Mocked request response gateway timeout" # do not use assertDictContainsSubset, deprecated self.assertEqual(dict(ret, **metricsDict), ret, "Wrong return from mock") self.assertIn("resptimems", ret, "No response time measurement in return from function when return code != 200")
def test_instanciate_nok_badurl(self, mock_requests): classinst = Measurement("example com", 5) mock_requests.return_value = None ret = classinst.get_url_response_time() self.assertEqual(None, ret, "Bad URL given, failed")