xref: /aosp_15_r20/external/autotest/autotest_lib/cli/rpc_unittest.py (revision 9c5db1993ded3edbeafc8092d69fe5de2ee02df7)
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