|
|
|
@ -1,5 +1,6 @@
|
|
|
|
|
import unittest
|
|
|
|
|
import io
|
|
|
|
|
import os
|
|
|
|
|
|
|
|
|
|
from socrate import conf, system
|
|
|
|
|
|
|
|
|
@ -78,3 +79,30 @@ class TestSystem(unittest.TestCase):
|
|
|
|
|
system.resolve_address("1.2.3.4.xip.io:80"),
|
|
|
|
|
"1.2.3.4:80"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
def test_get_host_address_from_environment(self):
|
|
|
|
|
if "TEST_ADDRESS" in os.environ:
|
|
|
|
|
del os.environ["TEST_ADDRESS"]
|
|
|
|
|
if "HOST_TEST" in os.environ:
|
|
|
|
|
del os.environ["HOST_TEST"]
|
|
|
|
|
# if nothing is set, the default must be resolved
|
|
|
|
|
self.assertEqual(
|
|
|
|
|
system.get_host_address_from_environment("TEST", "1.2.3.4.xip.io:80"),
|
|
|
|
|
"1.2.3.4:80"
|
|
|
|
|
)
|
|
|
|
|
# if HOST is set, the HOST must be resolved
|
|
|
|
|
os.environ['HOST_TEST']="1.2.3.5.xip.io:80"
|
|
|
|
|
self.assertEqual(
|
|
|
|
|
system.get_host_address_from_environment("TEST", "1.2.3.4.xip.io:80"),
|
|
|
|
|
"1.2.3.5:80"
|
|
|
|
|
)
|
|
|
|
|
# if ADDRESS is set, the ADDRESS must be returned unresolved
|
|
|
|
|
os.environ['TEST_ADDRESS']="1.2.3.6.xip.io:80"
|
|
|
|
|
self.assertEqual(
|
|
|
|
|
system.get_host_address_from_environment("TEST", "1.2.3.4.xip.io:80"),
|
|
|
|
|
"1.2.3.6.xip.io:80"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
unittest.main()
|
|
|
|
|