Tri Vo | 29ac182 | 2016-10-01 17:06:29 -0700 | [diff] [blame] | 1 | #!/usr/bin/env python |
| 2 | # |
| 3 | # Copyright 2016 - The Android Open Source Project |
| 4 | # |
| 5 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 6 | # you may not use this file except in compliance with the License. |
| 7 | # You may obtain a copy of the License at |
| 8 | # |
| 9 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 10 | # |
| 11 | # Unless required by applicable law or agreed to in writing, software |
| 12 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 13 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 14 | # See the License for the specific language governing permissions and |
| 15 | # limitations under the License. |
| 16 | |
| 17 | """Tests for acloud.public.config.""" |
| 18 | |
| 19 | import mock |
| 20 | |
| 21 | import unittest |
| 22 | from acloud.internal.proto import internal_config_pb2 |
| 23 | from acloud.internal.proto import user_config_pb2 |
| 24 | from acloud.public import config |
| 25 | from acloud.public import errors |
| 26 | |
| 27 | |
| 28 | class AcloudConfigManagerTest(unittest.TestCase): |
| 29 | """Test acloud configuration manager.""" |
| 30 | |
| 31 | USER_CONFIG = """ |
| 32 | service_account_name: "fake@developer.gserviceaccount.com" |
| 33 | service_account_private_key_path: "/path/to/service/account/key" |
| 34 | project: "fake-project" |
| 35 | zone: "us-central1-f" |
| 36 | machine_type: "n1-standard-1" |
| 37 | network: "default" |
| 38 | ssh_private_key_path: "/path/to/ssh/key" |
| 39 | storage_bucket_name: "fake_bucket" |
| 40 | orientation: "portrait" |
| 41 | resolution: "1200x1200x1200x1200" |
| 42 | client_id: "fake_client_id" |
| 43 | client_secret: "fake_client_secret" |
| 44 | metadata_variable { |
| 45 | key: "metadata_1" |
| 46 | value: "metadata_value_1" |
| 47 | } |
| 48 | """ |
| 49 | |
| 50 | INTERNAL_CONFIG = """ |
| 51 | min_machine_size: "n1-standard-1" |
| 52 | disk_image_name: "avd-system.tar.gz" |
| 53 | disk_image_mime_type: "application/x-tar" |
| 54 | disk_image_extension: ".tar.gz" |
| 55 | disk_raw_image_name: "disk.raw" |
| 56 | disk_raw_image_extension: ".img" |
| 57 | creds_cache_file: ".fake_oauth2.dat" |
| 58 | user_agent: "fake_user_agent" |
| 59 | |
| 60 | default_usr_cfg { |
| 61 | machine_type: "n1-standard-1" |
| 62 | network: "default" |
| 63 | metadata_variable { |
| 64 | key: "metadata_1" |
| 65 | value: "metadata_value_1" |
| 66 | } |
| 67 | |
| 68 | metadata_variable { |
| 69 | key: "metadata_2" |
| 70 | value: "metadata_value_2" |
| 71 | } |
| 72 | } |
| 73 | |
| 74 | device_resolution_map { |
| 75 | key: "nexus5" |
| 76 | value: "1080x1920x32x480" |
| 77 | } |
| 78 | |
| 79 | device_default_orientation_map { |
| 80 | key: "nexus5" |
| 81 | value: "portrait" |
| 82 | } |
| 83 | |
| 84 | valid_branch_and_min_build_id { |
| 85 | key: "git_jb-gce-dev" |
| 86 | value: 0 |
| 87 | } |
| 88 | """ |
| 89 | |
| 90 | def setUp(self): |
| 91 | self.config_file = mock.MagicMock() |
| 92 | |
| 93 | def testLoadUserConfig(self): |
| 94 | """Test loading user config.""" |
| 95 | self.config_file.read.return_value = self.USER_CONFIG |
| 96 | cfg = config.AcloudConfigManager.LoadConfigFromProtocolBuffer( |
| 97 | self.config_file, user_config_pb2.UserConfig) |
| 98 | self.assertEquals(cfg.service_account_name, |
| 99 | "fake@developer.gserviceaccount.com") |
| 100 | self.assertEquals(cfg.service_account_private_key_path, |
| 101 | "/path/to/service/account/key") |
| 102 | self.assertEquals(cfg.project, "fake-project") |
| 103 | self.assertEquals(cfg.zone, "us-central1-f") |
| 104 | self.assertEquals(cfg.machine_type, "n1-standard-1") |
| 105 | self.assertEquals(cfg.network, "default") |
| 106 | self.assertEquals(cfg.ssh_private_key_path, "/path/to/ssh/key") |
| 107 | self.assertEquals(cfg.storage_bucket_name, "fake_bucket") |
| 108 | self.assertEquals(cfg.orientation, "portrait") |
| 109 | self.assertEquals(cfg.resolution, "1200x1200x1200x1200") |
| 110 | self.assertEquals(cfg.client_id, "fake_client_id") |
| 111 | self.assertEquals(cfg.client_secret, "fake_client_secret") |
| 112 | self.assertEquals({key: val |
| 113 | for key, val in cfg.metadata_variable.iteritems()}, |
| 114 | {"metadata_1": "metadata_value_1"}) |
| 115 | |
| 116 | def testLoadInternalConfig(self): |
| 117 | """Test loading internal config.""" |
| 118 | self.config_file.read.return_value = self.INTERNAL_CONFIG |
| 119 | cfg = config.AcloudConfigManager.LoadConfigFromProtocolBuffer( |
| 120 | self.config_file, internal_config_pb2.InternalConfig) |
| 121 | self.assertEquals(cfg.min_machine_size, "n1-standard-1") |
| 122 | self.assertEquals(cfg.disk_image_name, "avd-system.tar.gz") |
| 123 | self.assertEquals(cfg.disk_image_mime_type, "application/x-tar") |
| 124 | self.assertEquals(cfg.disk_image_extension, ".tar.gz") |
| 125 | self.assertEquals(cfg.disk_raw_image_name, "disk.raw") |
| 126 | self.assertEquals(cfg.disk_raw_image_extension, ".img") |
| 127 | self.assertEquals(cfg.creds_cache_file, ".fake_oauth2.dat") |
| 128 | self.assertEquals(cfg.user_agent, "fake_user_agent") |
| 129 | self.assertEquals(cfg.default_usr_cfg.machine_type, "n1-standard-1") |
| 130 | self.assertEquals(cfg.default_usr_cfg.network, "default") |
| 131 | self.assertEquals({ |
| 132 | key: val |
| 133 | for key, val in cfg.default_usr_cfg.metadata_variable.iteritems() |
| 134 | }, {"metadata_1": "metadata_value_1", |
| 135 | "metadata_2": "metadata_value_2"}) |
| 136 | self.assertEquals( |
| 137 | {key: val |
| 138 | for key, val in cfg.device_resolution_map.iteritems()}, |
| 139 | {"nexus5": "1080x1920x32x480"}) |
| 140 | device_resolution = { |
| 141 | key: val |
| 142 | for key, val in cfg.device_default_orientation_map.iteritems() |
| 143 | } |
| 144 | self.assertEquals(device_resolution, {"nexus5": "portrait"}) |
| 145 | valid_branch_and_min_build_id = { |
| 146 | key: val |
| 147 | for key, val in cfg.valid_branch_and_min_build_id.iteritems() |
| 148 | } |
| 149 | self.assertEquals(valid_branch_and_min_build_id, {"git_jb-gce-dev": 0}) |
| 150 | |
| 151 | def testLoadConfigFails(self): |
| 152 | """Test loading a bad file.""" |
| 153 | self.config_file.read.return_value = "malformed text" |
| 154 | with self.assertRaises(errors.ConfigError): |
| 155 | config.AcloudConfigManager.LoadConfigFromProtocolBuffer( |
| 156 | self.config_file, internal_config_pb2.InternalConfig) |
| 157 | |
| 158 | |
| 159 | if __name__ == "__main__": |
| 160 | unittest.main() |