Merge pull request #228 from jrziviani/master
Enable unix socket to work with token plugin
This commit is contained in:
commit
72ce5c1ae8
|
@ -93,6 +93,16 @@ class ProxyRequestHandlerTestCase(unittest.TestCase):
|
||||||
self.assertEqual(host, "some host")
|
self.assertEqual(host, "some host")
|
||||||
self.assertEqual(port, "some port")
|
self.assertEqual(port, "some port")
|
||||||
|
|
||||||
|
def test_get_target_unix_socket(self):
|
||||||
|
class TestPlugin(token_plugins.BasePlugin):
|
||||||
|
def lookup(self, token):
|
||||||
|
return ("unix_socket", "/tmp/socket")
|
||||||
|
|
||||||
|
_, socket = self.handler.get_target(
|
||||||
|
TestPlugin(None), self.handler.path)
|
||||||
|
|
||||||
|
self.assertEqual(socket, "/tmp/socket")
|
||||||
|
|
||||||
def test_get_target_raises_error_on_unknown_token(self):
|
def test_get_target_raises_error_on_unknown_token(self):
|
||||||
class TestPlugin(token_plugins.BasePlugin):
|
class TestPlugin(token_plugins.BasePlugin):
|
||||||
def lookup(self, token):
|
def lookup(self, token):
|
||||||
|
|
|
@ -49,7 +49,13 @@ Traffic Legend:
|
||||||
|
|
||||||
def validate_connection(self):
|
def validate_connection(self):
|
||||||
if self.server.token_plugin:
|
if self.server.token_plugin:
|
||||||
(self.server.target_host, self.server.target_port) = self.get_target(self.server.token_plugin, self.path)
|
host, port = self.get_target(self.server.token_plugin, self.path)
|
||||||
|
if host == 'unix_socket':
|
||||||
|
self.server.unix_target = port
|
||||||
|
|
||||||
|
else:
|
||||||
|
self.server.target_host = host
|
||||||
|
self.server.target_port = port
|
||||||
|
|
||||||
if self.server.auth_plugin:
|
if self.server.auth_plugin:
|
||||||
try:
|
try:
|
||||||
|
|
Loading…
Reference in New Issue