class YamlServerRequestHandler_tests(TestCase): TEST_YAML = """my super test yaml""" TEST_LOCATIONS = """ locations: - monitoring - service-monitoring""" def setUp(self): mock_yaml_locations = MagicMock() mock_yaml_locations.get_yaml.return_value = self.TEST_YAML mock_yaml_locations.get_locations_as_yaml.return_value = self.TEST_LOCATIONS yaml_server.__config__["locations"] = mock_yaml_locations self.patcher = mock.patch('SimpleHTTPServer.SimpleHTTPRequestHandler') self.patcher.start() self.request_handler = YamlServerRequestHandler() self.request_handler.send_response = MagicMock() self.request_handler.headers = "" self.request_handler.send_header = MagicMock() self.request_handler.wfile = MagicMock() self.request_handler.date_time_string = MagicMock() self.request_handler.date_time_string.return_value = "NOW" def tearDown(self): self.patcher.stop() def test_get_locations(self): self.request_handler.path = "/" self.request_handler.do_GET() self.request_handler.send_header.assert_any_call('ETag', 'e363ed10bccdb2842e14498ef956a346a83f0617ddeeb8e991a654691c0ea836') self.request_handler.send_header.assert_any_call('Content-length', len(self.TEST_LOCATIONS)) self.request_handler.send_header.assert_any_call('Content-type', 'application/yaml') self.request_handler.wfile.write.assert_any_call(self.TEST_LOCATIONS) def test_get_yaml(self): self.request_handler.path = "/unittest" self.request_handler.do_GET() self.request_handler.send_header.assert_any_call('ETag', '6ee9c999bead9a4015de1b5108e3ac005a4893aa6e79e207b3be403063f1a296') self.request_handler.send_header.assert_any_call('Content-length', len(self.TEST_YAML)) self.request_handler.send_header.assert_any_call('Content-type', 'application/yaml') self.request_handler.send_header.assert_any_call('Last-Modified', 'NOW') self.request_handler.wfile.write.assert_any_call(self.TEST_YAML)
def setUp(self): mock_yaml_locations = MagicMock() mock_yaml_locations.get_yaml.return_value = self.TEST_YAML mock_yaml_locations.get_locations_as_yaml.return_value = self.TEST_LOCATIONS yaml_server.__config__["locations"] = mock_yaml_locations self.patcher = mock.patch('SimpleHTTPServer.SimpleHTTPRequestHandler') self.patcher.start() self.request_handler = YamlServerRequestHandler() self.request_handler.send_response = MagicMock() self.request_handler.headers = "" self.request_handler.send_header = MagicMock() self.request_handler.wfile = MagicMock() self.request_handler.date_time_string = MagicMock() self.request_handler.date_time_string.return_value = "NOW"