62 lines
1.9 KiB
Python
Executable File
62 lines
1.9 KiB
Python
Executable File
#!/usr/bin/python2
|
|
#
|
|
# Copyright Gregory P. Smith, Google Inc 2008
|
|
# Released under the GPL v2
|
|
|
|
"""Tests for server.frontend."""
|
|
|
|
#pylint: disable=missing-docstring
|
|
|
|
import os, unittest
|
|
import common
|
|
from autotest_lib.client.common_lib import global_config
|
|
from autotest_lib.client.common_lib import utils
|
|
from autotest_lib.client.common_lib.test_utils import mock
|
|
from autotest_lib.frontend.afe import rpc_client_lib
|
|
from autotest_lib.server import frontend
|
|
|
|
GLOBAL_CONFIG = global_config.global_config
|
|
|
|
|
|
class BaseRpcClientTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.god = mock.mock_god()
|
|
self.god.mock_up(rpc_client_lib, 'rpc_client_lib')
|
|
self.god.stub_function(utils, 'send_email')
|
|
self._saved_environ = dict(os.environ)
|
|
if 'AUTOTEST_WEB' in os.environ:
|
|
del os.environ['AUTOTEST_WEB']
|
|
|
|
|
|
def tearDown(self):
|
|
self.god.unstub_all()
|
|
os.environ.clear()
|
|
os.environ.update(self._saved_environ)
|
|
|
|
|
|
class RpcClientTest(BaseRpcClientTest):
|
|
def test_init(self):
|
|
os.environ['LOGNAME'] = 'unittest-user'
|
|
GLOBAL_CONFIG.override_config_value('SERVER', 'hostname', 'test-host')
|
|
rpc_client_lib.add_protocol.expect_call('test-host').and_return(
|
|
'http://test-host')
|
|
rpc_client_lib.get_proxy.expect_call(
|
|
'http://test-host/path',
|
|
headers={'AUTHORIZATION': 'unittest-user'})
|
|
frontend.RpcClient('/path', None, None, None, None, None)
|
|
self.god.check_playback()
|
|
|
|
|
|
class CrosVersionFormatTestCase(unittest.TestCase):
|
|
def test_format_cros_image_name(self):
|
|
test_board = 'fubar-board'
|
|
test_version = 'R99-20000.15.0'
|
|
image_name = frontend.format_cros_image_name(
|
|
test_board, test_version)
|
|
self.assertIn(test_board, image_name)
|
|
self.assertIn(test_version, image_name)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|