Add unittests.
Change-Id: I55e7397c44605719bb01aa1b6d81eed2fa743cf4
diff --git a/public/config_test.py b/public/config_test.py
new file mode 100644
index 0000000..554e751
--- /dev/null
+++ b/public/config_test.py
@@ -0,0 +1,160 @@
+#!/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()