You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
311 lines
12 KiB
311 lines
12 KiB
#!/usr/bin/env python
|
|
#
|
|
# Copyright 2016 - The Android Open Source Project
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
"""Tests for acloud.public.config."""
|
|
import unittest
|
|
import os
|
|
import tempfile
|
|
|
|
from unittest import mock
|
|
|
|
import six
|
|
|
|
# pylint: disable=no-name-in-module,import-error
|
|
from acloud import errors
|
|
from acloud.internal.proto import internal_config_pb2
|
|
from acloud.internal.proto import user_config_pb2
|
|
from acloud.public import config
|
|
|
|
|
|
class AcloudConfigManagerTest(unittest.TestCase):
|
|
"""Test acloud configuration manager."""
|
|
|
|
USER_CONFIG = """
|
|
service_account_name: "fake@developer.gserviceaccount.com"
|
|
service_account_private_key_path: "/path/to/service/account/key"
|
|
service_account_json_private_key_path: "/path/to/service/account/json_key"
|
|
project: "fake-project"
|
|
zone: "us-central1-f"
|
|
machine_type: "n1-standard-1"
|
|
network: "default"
|
|
ssh_private_key_path: "/path/to/ssh/key"
|
|
storage_bucket_name: "fake_bucket"
|
|
orientation: "portrait"
|
|
resolution: "1200x1200x1200x1200"
|
|
client_id: "fake_client_id"
|
|
client_secret: "fake_client_secret"
|
|
extra_args_ssh_tunnel: "fake_extra_args_ssh_tunnel"
|
|
metadata_variable {
|
|
key: "metadata_1"
|
|
value: "metadata_value_1"
|
|
}
|
|
hw_property: "cpu:3,resolution:1080x1920,dpi:480,memory:4g,disk:10g"
|
|
extra_scopes: "scope1"
|
|
extra_scopes: "scope2"
|
|
betty_image: "fake_betty_image"
|
|
"""
|
|
|
|
INTERNAL_CONFIG = """
|
|
min_machine_size: "n1-standard-1"
|
|
disk_image_name: "avd-system.tar.gz"
|
|
disk_image_mime_type: "application/x-tar"
|
|
disk_image_extension: ".tar.gz"
|
|
disk_raw_image_name: "disk.raw"
|
|
disk_raw_image_extension: ".img"
|
|
creds_cache_file: ".fake_oauth2.dat"
|
|
user_agent: "fake_user_agent"
|
|
kernel_build_target: "kernel"
|
|
emulator_build_target: "sdk_tools_linux"
|
|
|
|
default_usr_cfg {
|
|
machine_type: "n1-standard-1"
|
|
network: "default"
|
|
stable_host_image_name: "fake_stable_host_image_name"
|
|
stable_host_image_project: "fake_stable_host_image_project"
|
|
stable_goldfish_host_image_name: "fake_stable_goldfish_host_image_name"
|
|
stable_goldfish_host_image_project: "fake_stable_goldfish_host_image_project"
|
|
instance_name_pattern: "fake_instance_name_pattern"
|
|
stable_cheeps_host_image_name: "fake_stable_cheeps_host_image_name"
|
|
stable_cheeps_host_image_project: "fake_stable_cheeps_host_image_project"
|
|
metadata_variable {
|
|
key: "metadata_1"
|
|
value: "metadata_value_1"
|
|
}
|
|
|
|
metadata_variable {
|
|
key: "metadata_2"
|
|
value: "metadata_value_2"
|
|
}
|
|
}
|
|
|
|
device_resolution_map {
|
|
key: "nexus5"
|
|
value: "1080x1920x32x480"
|
|
}
|
|
|
|
device_default_orientation_map {
|
|
key: "nexus5"
|
|
value: "portrait"
|
|
}
|
|
|
|
valid_branch_and_min_build_id {
|
|
key: "aosp-master"
|
|
value: 0
|
|
}
|
|
|
|
common_hw_property_map {
|
|
key: "phone"
|
|
value: "cpu:2,resolution:1080x1920,dpi:420,memory:4g,disk:8g"
|
|
}
|
|
|
|
common_hw_property_map {
|
|
key: "auto"
|
|
value: "cpu:4,resolution:1280x800,dpi:160,memory:4g"
|
|
}
|
|
"""
|
|
|
|
def setUp(self):
|
|
self.config_file = mock.MagicMock()
|
|
# initial config with test config.
|
|
self.config_file.read.return_value = self.INTERNAL_CONFIG
|
|
internal_cfg = config.AcloudConfigManager.LoadConfigFromProtocolBuffer(
|
|
self.config_file, internal_config_pb2.InternalConfig)
|
|
self.config_file.read.return_value = self.USER_CONFIG
|
|
usr_cfg = config.AcloudConfigManager.LoadConfigFromProtocolBuffer(
|
|
self.config_file, user_config_pb2.UserConfig)
|
|
self.cfg = config.AcloudConfig(usr_cfg, internal_cfg)
|
|
|
|
# pylint: disable=no-member
|
|
def testLoadUserConfig(self):
|
|
"""Test loading user config."""
|
|
self.config_file.read.return_value = self.USER_CONFIG
|
|
cfg = config.AcloudConfigManager.LoadConfigFromProtocolBuffer(
|
|
self.config_file, user_config_pb2.UserConfig)
|
|
self.assertEqual(cfg.service_account_name,
|
|
"fake@developer.gserviceaccount.com")
|
|
self.assertEqual(cfg.service_account_private_key_path,
|
|
"/path/to/service/account/key")
|
|
self.assertEqual(cfg.service_account_json_private_key_path,
|
|
"/path/to/service/account/json_key")
|
|
self.assertEqual(cfg.project, "fake-project")
|
|
self.assertEqual(cfg.zone, "us-central1-f")
|
|
self.assertEqual(cfg.machine_type, "n1-standard-1")
|
|
self.assertEqual(cfg.network, "default")
|
|
self.assertEqual(cfg.ssh_private_key_path, "/path/to/ssh/key")
|
|
self.assertEqual(cfg.storage_bucket_name, "fake_bucket")
|
|
self.assertEqual(cfg.orientation, "portrait")
|
|
self.assertEqual(cfg.resolution, "1200x1200x1200x1200")
|
|
self.assertEqual(cfg.client_id, "fake_client_id")
|
|
self.assertEqual(cfg.client_secret, "fake_client_secret")
|
|
self.assertEqual(cfg.extra_args_ssh_tunnel, "fake_extra_args_ssh_tunnel")
|
|
self.assertEqual(
|
|
{key: val for key, val in six.iteritems(cfg.metadata_variable)},
|
|
{"metadata_1": "metadata_value_1"})
|
|
self.assertEqual(cfg.hw_property,
|
|
"cpu:3,resolution:1080x1920,dpi:480,memory:4g,"
|
|
"disk:10g")
|
|
self.assertEqual(cfg.extra_scopes, ["scope1", "scope2"])
|
|
self.assertEqual(cfg.betty_image, "fake_betty_image")
|
|
|
|
# pylint: disable=protected-access
|
|
@mock.patch("os.makedirs")
|
|
@mock.patch("os.path.exists")
|
|
def testLoadUserConfigLogic(self, mock_file_exist, _mock_makedirs):
|
|
"""Test load user config logic.
|
|
|
|
Load user config with some special design.
|
|
1. User specified user config:
|
|
If user config didn't exist: Raise exception.
|
|
2. User didn't specify user config, use default config:
|
|
If default config didn't exist: Initialize empty data.
|
|
"""
|
|
config_specify = config.AcloudConfigManager(self.config_file)
|
|
self.config_file.read.return_value = self.USER_CONFIG
|
|
self.assertEqual(config_specify.user_config_path, self.config_file)
|
|
mock_file_exist.return_value = False
|
|
with self.assertRaises(errors.ConfigError):
|
|
config_specify.Load()
|
|
# Test default config
|
|
config_unspecify = config.AcloudConfigManager(None)
|
|
cfg = config_unspecify.Load()
|
|
self.assertEqual(config_unspecify.user_config_path,
|
|
config.GetDefaultConfigFile())
|
|
self.assertEqual(cfg.project, "")
|
|
self.assertEqual(cfg.zone, "")
|
|
|
|
# Test default user config exist
|
|
mock_file_exist.return_value = True
|
|
# Write the config data into a tmp file and have GetDefaultConfigFile
|
|
# return that.
|
|
_, temp_cfg_file_path = tempfile.mkstemp()
|
|
with open(temp_cfg_file_path, "w") as cfg_file:
|
|
cfg_file.writelines(self.USER_CONFIG)
|
|
default_patcher = mock.patch.object(config, "GetDefaultConfigFile",
|
|
return_value=temp_cfg_file_path)
|
|
default_patcher.start()
|
|
try:
|
|
config_exist = config.AcloudConfigManager(None)
|
|
cfg = config_exist.Load()
|
|
self.assertEqual(cfg.project, "fake-project")
|
|
self.assertEqual(cfg.zone, "us-central1-f")
|
|
self.assertEqual(cfg.client_id, "fake_client_id")
|
|
self.assertEqual(cfg.client_secret, "fake_client_secret")
|
|
finally:
|
|
# Delete tmp file
|
|
os.remove(temp_cfg_file_path)
|
|
default_patcher.stop()
|
|
|
|
def testLoadInternalConfig(self):
|
|
"""Test loading internal config."""
|
|
self.config_file.read.return_value = self.INTERNAL_CONFIG
|
|
cfg = config.AcloudConfigManager.LoadConfigFromProtocolBuffer(
|
|
self.config_file, internal_config_pb2.InternalConfig)
|
|
self.assertEqual(cfg.min_machine_size, "n1-standard-1")
|
|
self.assertEqual(cfg.disk_image_name, "avd-system.tar.gz")
|
|
self.assertEqual(cfg.disk_image_mime_type, "application/x-tar")
|
|
self.assertEqual(cfg.disk_image_extension, ".tar.gz")
|
|
self.assertEqual(cfg.disk_raw_image_name, "disk.raw")
|
|
self.assertEqual(cfg.disk_raw_image_extension, ".img")
|
|
self.assertEqual(cfg.creds_cache_file, ".fake_oauth2.dat")
|
|
self.assertEqual(cfg.user_agent, "fake_user_agent")
|
|
self.assertEqual(cfg.default_usr_cfg.machine_type, "n1-standard-1")
|
|
self.assertEqual(cfg.default_usr_cfg.network, "default")
|
|
self.assertEqual({
|
|
key: val
|
|
for key, val in six.iteritems(cfg.default_usr_cfg.metadata_variable)
|
|
}, {
|
|
"metadata_1": "metadata_value_1",
|
|
"metadata_2": "metadata_value_2"
|
|
})
|
|
self.assertEqual(
|
|
{key: val for key, val in six.iteritems(cfg.device_resolution_map)},
|
|
{"nexus5": "1080x1920x32x480"})
|
|
device_resolution = {
|
|
key: val
|
|
for key, val in six.iteritems(cfg.device_default_orientation_map)
|
|
}
|
|
self.assertEqual(device_resolution, {"nexus5": "portrait"})
|
|
valid_branch_and_min_build_id = {
|
|
key: val
|
|
for key, val in six.iteritems(cfg.valid_branch_and_min_build_id)
|
|
}
|
|
self.assertEqual(valid_branch_and_min_build_id, {"aosp-master": 0})
|
|
self.assertEqual(cfg.default_usr_cfg.stable_host_image_name,
|
|
"fake_stable_host_image_name")
|
|
self.assertEqual(cfg.default_usr_cfg.stable_host_image_project,
|
|
"fake_stable_host_image_project")
|
|
self.assertEqual(cfg.kernel_build_target, "kernel")
|
|
|
|
# Emulator related
|
|
self.assertEqual(cfg.default_usr_cfg.stable_goldfish_host_image_name,
|
|
"fake_stable_goldfish_host_image_name")
|
|
self.assertEqual(cfg.default_usr_cfg.stable_goldfish_host_image_project,
|
|
"fake_stable_goldfish_host_image_project")
|
|
self.assertEqual(cfg.emulator_build_target, "sdk_tools_linux")
|
|
self.assertEqual(cfg.default_usr_cfg.instance_name_pattern,
|
|
"fake_instance_name_pattern")
|
|
|
|
# Cheeps related
|
|
self.assertEqual(cfg.default_usr_cfg.stable_cheeps_host_image_name,
|
|
"fake_stable_cheeps_host_image_name")
|
|
self.assertEqual(cfg.default_usr_cfg.stable_cheeps_host_image_project,
|
|
"fake_stable_cheeps_host_image_project")
|
|
|
|
# hw property
|
|
self.assertEqual(
|
|
{key: val for key, val in six.iteritems(cfg.common_hw_property_map)},
|
|
{"phone": "cpu:2,resolution:1080x1920,dpi:420,memory:4g,disk:8g",
|
|
"auto": "cpu:4,resolution:1280x800,dpi:160,memory:4g"})
|
|
|
|
def testLoadConfigFails(self):
|
|
"""Test loading a bad file."""
|
|
self.config_file.read.return_value = "malformed text"
|
|
with self.assertRaises(errors.ConfigError):
|
|
config.AcloudConfigManager.LoadConfigFromProtocolBuffer(
|
|
self.config_file, internal_config_pb2.InternalConfig)
|
|
|
|
def testOverrideWithArgs(self):
|
|
"""Test OverrideWithArgs."""
|
|
# test override zone.
|
|
self.cfg.zone = "us-central1-f"
|
|
args = mock.MagicMock()
|
|
args.which = "create"
|
|
args.flavor = "phone"
|
|
args.zone = "us-central1-b"
|
|
self.cfg.OverrideWithArgs(args)
|
|
self.assertEqual(self.cfg.zone, "us-central1-b")
|
|
|
|
def testGetDefaultHwProperty(self):
|
|
"""Test GetDefaultHwProperty."""
|
|
# test with "phone" flavor
|
|
expected = "cpu:2,resolution:1080x1920,dpi:420,memory:4g,disk:8g"
|
|
self.assertEqual(expected, self.cfg.GetDefaultHwProperty("phone"))
|
|
|
|
# test with "auto" flavor
|
|
expected = "cpu:4,resolution:1280x800,dpi:160,memory:4g"
|
|
self.assertEqual(expected, self.cfg.GetDefaultHwProperty("auto"))
|
|
|
|
def testGetMissingFields(self):
|
|
"""Test GetMissingFields."""
|
|
fields = ["project", "zone", "hw_property"]
|
|
self.cfg.hw_property = ""
|
|
expected = ["hw_property"]
|
|
self.assertEqual(expected, self.cfg.GetMissingFields(fields))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|