From 5b03540601bc337205ec4486e19537ba58500029 Mon Sep 17 00:00:00 2001 From: Jake Howard Date: Wed, 12 Dec 2018 23:19:05 +0000 Subject: [PATCH] Test multiple hostname lookup --- tests/test_router/test_utils.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/test_router/test_utils.py b/tests/test_router/test_utils.py index 457b812..55b837f 100644 --- a/tests/test_router/test_utils.py +++ b/tests/test_router/test_utils.py @@ -20,6 +20,16 @@ class ProcessForHostnameTestCase(BaseTestCase): self.assertIsNone(proc.poll()) self.assertEqual(utils.get_process_for_hostname("localhost").pid, proc.pid) + def test_finds_process_with_multiple_hostnames(self): + proc = subprocess.Popen( + "yes", + stdout=subprocess.PIPE, + env={**os.environ, utils.HOSTNAME_ENV_VAR: "localhost,test.local"}, + ) + self.assertIsNone(proc.poll()) + self.assertEqual(utils.get_process_for_hostname("localhost").pid, proc.pid) + self.assertEqual(utils.get_process_for_hostname("test.local").pid, proc.pid) + class HostnameFromRequestTestCase(BaseTestCase): def get_request_for_hostname(self, hostname):