| #!/usr/bin/python2 |
| # |
| # Copyright 2008 Google Inc. All Rights Reserved. |
| |
| """Test for the rpc proxy class.""" |
| |
| import unittest, os |
| import common |
| from autotest_lib.cli import rpc |
| from autotest_lib.client.common_lib import global_config |
| from autotest_lib.frontend.afe import rpc_client_lib |
| from autotest_lib.frontend.afe.json_rpc import proxy |
| |
| GLOBAL_CONFIG = global_config.global_config |
| |
| |
| class rpc_unittest(unittest.TestCase): |
| def setUp(self): |
| self.old_environ = os.environ.copy() |
| if 'AUTOTEST_WEB' in os.environ: |
| del os.environ['AUTOTEST_WEB'] |
| |
| |
| def tearDown(self): |
| os.environ.clear() |
| os.environ.update(self.old_environ) |
| |
| |
| def test_get_autotest_server_specific(self): |
| self.assertEqual('http://foo', rpc.get_autotest_server('foo')) |
| |
| |
| def test_get_autotest_server_none(self): |
| GLOBAL_CONFIG.override_config_value('SERVER', 'hostname', 'Prince') |
| self.assertEqual('http://Prince', rpc.get_autotest_server(None)) |
| |
| |
| def test_get_autotest_server_environ(self): |
| os.environ['AUTOTEST_WEB'] = 'foo-dev' |
| self.assertEqual('http://foo-dev', rpc.get_autotest_server(None)) |
| del os.environ['AUTOTEST_WEB'] |
| |
| |
| def test_get_autotest_server_environ_precedence(self): |
| os.environ['AUTOTEST_WEB'] = 'foo-dev' |
| self.assertEqual('http://foo', rpc.get_autotest_server('foo')) |
| del os.environ['AUTOTEST_WEB'] |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |