Add unit test for token file with tab

1 files changed, 15 insertions, 0 deletions

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")