From b9b269c73f51b1cfb3cbaa5e979efa53202d8289 Mon Sep 17 00:00:00 2001 From: Pierre Ossman Date: Fri, 16 Apr 2021 13:02:07 +0200 Subject: [PATCH] Add unit test for token file with tab --- tests/test_token_plugins.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/tests/test_token_plugins.py b/tests/test_token_plugins.py index 05d5bef..9254256 100644 --- a/tests/test_token_plugins.py +++ b/tests/test_token_plugins.py @@ -37,6 +37,21 @@ class ReadOnlyTokenFileTestCase(unittest.TestCase): self.assertEqual(result[0], "remote_host") self.assertEqual(result[1], "remote_port") + patch('os.path.isdir', MagicMock(return_value=False)) + def test_tabs(self): + plugin = ReadOnlyTokenFile('configfile') + + config = "testhost:\tremote_host:remote_port" + pyopen = mock_open(read_data=config) + + with patch("websockify.token_plugins.open", pyopen): + result = plugin.lookup('testhost') + + pyopen.assert_called_once_with('configfile') + self.assertIsNotNone(result) + self.assertEqual(result[0], "remote_host") + self.assertEqual(result[1], "remote_port") + class JWSTokenTestCase(unittest.TestCase): def test_asymmetric_jws_token_plugin(self): plugin = JWTTokenApi("./tests/fixtures/public.pem")