blob: 0f602c4259c1b90e4921132b94c0c85de64e394c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
#!/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()
|