1*9c5db199SXin Li#!/usr/bin/python3 2*9c5db199SXin Li# 3*9c5db199SXin Li# Copyright 2008 Google Inc. All Rights Reserved. 4*9c5db199SXin Li 5*9c5db199SXin Li"""Test for the rpc proxy class.""" 6*9c5db199SXin Li 7*9c5db199SXin Liimport unittest, os 8*9c5db199SXin Liimport common 9*9c5db199SXin Lifrom autotest_lib.cli import rpc 10*9c5db199SXin Lifrom autotest_lib.client.common_lib import global_config 11*9c5db199SXin Lifrom autotest_lib.frontend.afe import rpc_client_lib 12*9c5db199SXin Lifrom autotest_lib.frontend.afe.json_rpc import proxy 13*9c5db199SXin Li 14*9c5db199SXin LiGLOBAL_CONFIG = global_config.global_config 15*9c5db199SXin Li 16*9c5db199SXin Li 17*9c5db199SXin Liclass rpc_unittest(unittest.TestCase): 18*9c5db199SXin Li def setUp(self): 19*9c5db199SXin Li self.old_environ = os.environ.copy() 20*9c5db199SXin Li if 'AUTOTEST_WEB' in os.environ: 21*9c5db199SXin Li del os.environ['AUTOTEST_WEB'] 22*9c5db199SXin Li 23*9c5db199SXin Li 24*9c5db199SXin Li def tearDown(self): 25*9c5db199SXin Li os.environ.clear() 26*9c5db199SXin Li os.environ.update(self.old_environ) 27*9c5db199SXin Li 28*9c5db199SXin Li 29*9c5db199SXin Li def test_get_autotest_server_specific(self): 30*9c5db199SXin Li self.assertEqual('http://foo', rpc.get_autotest_server('foo')) 31*9c5db199SXin Li 32*9c5db199SXin Li 33*9c5db199SXin Li def test_get_autotest_server_none(self): 34*9c5db199SXin Li GLOBAL_CONFIG.override_config_value('SERVER', 'hostname', 'Prince') 35*9c5db199SXin Li self.assertEqual('http://Prince', rpc.get_autotest_server(None)) 36*9c5db199SXin Li 37*9c5db199SXin Li 38*9c5db199SXin Li def test_get_autotest_server_environ(self): 39*9c5db199SXin Li os.environ['AUTOTEST_WEB'] = 'foo-dev' 40*9c5db199SXin Li self.assertEqual('http://foo-dev', rpc.get_autotest_server(None)) 41*9c5db199SXin Li del os.environ['AUTOTEST_WEB'] 42*9c5db199SXin Li 43*9c5db199SXin Li 44*9c5db199SXin Li def test_get_autotest_server_environ_precedence(self): 45*9c5db199SXin Li os.environ['AUTOTEST_WEB'] = 'foo-dev' 46*9c5db199SXin Li self.assertEqual('http://foo', rpc.get_autotest_server('foo')) 47*9c5db199SXin Li del os.environ['AUTOTEST_WEB'] 48*9c5db199SXin Li 49*9c5db199SXin Li 50*9c5db199SXin Liif __name__ == '__main__': 51*9c5db199SXin Li unittest.main() 52