public/config_test.py

#!/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 mock

import unittest
from acloud.internal.proto import internal_config_pb2
from acloud.internal.proto import user_config_pb2
from acloud.public import config
from acloud.public import errors


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"
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"
metadata_variable {
    key: "metadata_1"
    value: "metadata_value_1"
}
"""

    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"

default_usr_cfg {
    machine_type: "n1-standard-1"
    network: "default"
    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: "git_jb-gce-dev"
    value: 0
}
"""

    def setUp(self):
        self.config_file = mock.MagicMock()

    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.assertEquals(cfg.service_account_name,
                          "fake@developer.gserviceaccount.com")
        self.assertEquals(cfg.service_account_private_key_path,
                          "/path/to/service/account/key")
        self.assertEquals(cfg.project, "fake-project")
        self.assertEquals(cfg.zone, "us-central1-f")
        self.assertEquals(cfg.machine_type, "n1-standard-1")
        self.assertEquals(cfg.network, "default")
        self.assertEquals(cfg.ssh_private_key_path, "/path/to/ssh/key")
        self.assertEquals(cfg.storage_bucket_name, "fake_bucket")
        self.assertEquals(cfg.orientation, "portrait")
        self.assertEquals(cfg.resolution, "1200x1200x1200x1200")
        self.assertEquals(cfg.client_id, "fake_client_id")
        self.assertEquals(cfg.client_secret, "fake_client_secret")
        self.assertEquals({key: val
                           for key, val in cfg.metadata_variable.iteritems()},
                          {"metadata_1": "metadata_value_1"})

    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.assertEquals(cfg.min_machine_size, "n1-standard-1")
        self.assertEquals(cfg.disk_image_name, "avd-system.tar.gz")
        self.assertEquals(cfg.disk_image_mime_type, "application/x-tar")
        self.assertEquals(cfg.disk_image_extension, ".tar.gz")
        self.assertEquals(cfg.disk_raw_image_name, "disk.raw")
        self.assertEquals(cfg.disk_raw_image_extension, ".img")
        self.assertEquals(cfg.creds_cache_file, ".fake_oauth2.dat")
        self.assertEquals(cfg.user_agent, "fake_user_agent")
        self.assertEquals(cfg.default_usr_cfg.machine_type, "n1-standard-1")
        self.assertEquals(cfg.default_usr_cfg.network, "default")
        self.assertEquals({
            key: val
            for key, val in cfg.default_usr_cfg.metadata_variable.iteritems()
        }, {"metadata_1": "metadata_value_1",
            "metadata_2": "metadata_value_2"})
        self.assertEquals(
            {key: val
             for key, val in cfg.device_resolution_map.iteritems()},
            {"nexus5": "1080x1920x32x480"})
        device_resolution = {
            key: val
            for key, val in cfg.device_default_orientation_map.iteritems()
        }
        self.assertEquals(device_resolution, {"nexus5": "portrait"})
        valid_branch_and_min_build_id = {
            key: val
            for key, val in cfg.valid_branch_and_min_build_id.iteritems()
        }
        self.assertEquals(valid_branch_and_min_build_id, {"git_jb-gce-dev": 0})

    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)


if __name__ == "__main__":
    unittest.main()