blob: 0e87e4d984bc82c8d5cca7c968c9c31d261ab666 [file] [log] [blame] [edit]
#!/usr/bin/python
#
# 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()