1 //
2 // Copyright (C) 2015 The Android Open Source Project
3 //
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
7 //
8 // http://www.apache.org/licenses/LICENSE-2.0
9 //
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
15 //
16
17 #include "update_engine/payload_generator/payload_generation_config.h"
18
19 #include <utility>
20
21 #include <gtest/gtest.h>
22
23 namespace chromeos_update_engine {
24
25 class PayloadGenerationConfigTest : public ::testing::Test {};
26
TEST_F(PayloadGenerationConfigTest,SimpleLoadPostInstallConfigTest)27 TEST_F(PayloadGenerationConfigTest, SimpleLoadPostInstallConfigTest) {
28 ImageConfig image_config;
29 image_config.partitions.emplace_back("root");
30 brillo::KeyValueStore store;
31 EXPECT_TRUE(
32 store.LoadFromString("RUN_POSTINSTALL_root=true\n"
33 "POSTINSTALL_PATH_root=postinstall\n"
34 "FILESYSTEM_TYPE_root=ext4\n"
35 "POSTINSTALL_OPTIONAL_root=true"));
36 EXPECT_TRUE(image_config.LoadPostInstallConfig(store));
37 EXPECT_FALSE(image_config.partitions[0].postinstall.IsEmpty());
38 EXPECT_EQ(true, image_config.partitions[0].postinstall.run);
39 EXPECT_EQ("postinstall", image_config.partitions[0].postinstall.path);
40 EXPECT_EQ("ext4", image_config.partitions[0].postinstall.filesystem_type);
41 EXPECT_TRUE(image_config.partitions[0].postinstall.optional);
42 }
43
TEST_F(PayloadGenerationConfigTest,LoadPostInstallConfigNameMismatchTest)44 TEST_F(PayloadGenerationConfigTest, LoadPostInstallConfigNameMismatchTest) {
45 ImageConfig image_config;
46 image_config.partitions.emplace_back("system");
47 brillo::KeyValueStore store;
48 EXPECT_TRUE(
49 store.LoadFromString("RUN_POSTINSTALL_root=true\n"
50 "POSTINSTALL_PATH_root=postinstall\n"
51 "FILESYSTEM_TYPE_root=ext4"));
52 EXPECT_FALSE(image_config.LoadPostInstallConfig(store));
53 EXPECT_TRUE(image_config.partitions[0].postinstall.IsEmpty());
54 }
55
TEST_F(PayloadGenerationConfigTest,LoadDynamicPartitionMetadataTest)56 TEST_F(PayloadGenerationConfigTest, LoadDynamicPartitionMetadataTest) {
57 ImageConfig image_config;
58 brillo::KeyValueStore store;
59 ASSERT_TRUE(
60 store.LoadFromString("super_partition_groups=group_a group_b\n"
61 "group_a_size=3221225472\n"
62 "group_a_partition_list=system system_ext\n"
63 "group_b_size=2147483648\n"
64 "group_b_partition_list=vendor\n"));
65 EXPECT_TRUE(image_config.LoadDynamicPartitionMetadata(store));
66 ASSERT_NE(nullptr, image_config.dynamic_partition_metadata);
67
68 ASSERT_EQ(2, image_config.dynamic_partition_metadata->groups_size());
69
70 const auto& group_a = image_config.dynamic_partition_metadata->groups(0);
71 EXPECT_EQ("group_a", group_a.name());
72 EXPECT_EQ(3221225472u, group_a.size());
73 ASSERT_EQ(2, group_a.partition_names_size());
74 EXPECT_EQ("system", group_a.partition_names(0));
75 EXPECT_EQ("system_ext", group_a.partition_names(1));
76
77 const auto& group_b = image_config.dynamic_partition_metadata->groups(1);
78 EXPECT_EQ("group_b", group_b.name());
79 EXPECT_EQ(2147483648u, group_b.size());
80 ASSERT_EQ(1, group_b.partition_names_size());
81 EXPECT_EQ("vendor", group_b.partition_names(0));
82 }
83
TEST_F(PayloadGenerationConfigTest,LoadDynamicPartitionMetadataMissingSizeTest)84 TEST_F(PayloadGenerationConfigTest,
85 LoadDynamicPartitionMetadataMissingSizeTest) {
86 ImageConfig image_config;
87 brillo::KeyValueStore store;
88 ASSERT_TRUE(
89 store.LoadFromString("super_partition_groups=foo\n"
90 "foo_partition_list=baz\n"));
91 EXPECT_FALSE(image_config.LoadDynamicPartitionMetadata(store));
92 EXPECT_EQ(nullptr, image_config.dynamic_partition_metadata);
93 }
94
TEST_F(PayloadGenerationConfigTest,LoadDynamicPartitionMetadataBadSizeTest)95 TEST_F(PayloadGenerationConfigTest, LoadDynamicPartitionMetadataBadSizeTest) {
96 ImageConfig image_config;
97 brillo::KeyValueStore store;
98 ASSERT_TRUE(
99 store.LoadFromString("super_partition_groups=foo\n"
100 "foo_size=bar\n"
101 "foo_partition_list=baz\n"));
102 EXPECT_FALSE(image_config.LoadDynamicPartitionMetadata(store));
103 EXPECT_EQ(nullptr, image_config.dynamic_partition_metadata);
104 }
105
TEST_F(PayloadGenerationConfigTest,ValidateDynamicPartitionMetadata)106 TEST_F(PayloadGenerationConfigTest, ValidateDynamicPartitionMetadata) {
107 ImageConfig image_config;
108
109 PartitionConfig system("system");
110 system.size = 2147483648u;
111 PartitionConfig system_ext("system_ext");
112 system_ext.size = 1073741824u;
113
114 image_config.partitions.push_back(std::move(system));
115 image_config.partitions.push_back(std::move(system_ext));
116
117 brillo::KeyValueStore store;
118 ASSERT_TRUE(
119 store.LoadFromString("super_partition_groups=foo\n"
120 "foo_size=3221225472\n"
121 "foo_partition_list=system system_ext\n"));
122 EXPECT_TRUE(image_config.LoadDynamicPartitionMetadata(store));
123 EXPECT_NE(nullptr, image_config.dynamic_partition_metadata);
124
125 EXPECT_TRUE(image_config.ValidateDynamicPartitionMetadata());
126 }
127
TEST_F(PayloadGenerationConfigTest,ValidateDynamicPartitionMetadataTooBig)128 TEST_F(PayloadGenerationConfigTest, ValidateDynamicPartitionMetadataTooBig) {
129 ImageConfig image_config;
130
131 PartitionConfig system("system");
132 system.size = 4294967296u;
133 image_config.partitions.push_back(std::move(system));
134
135 brillo::KeyValueStore store;
136 ASSERT_TRUE(
137 store.LoadFromString("super_partition_groups=foo\n"
138 "foo_size=3221225472\n"
139 "foo_partition_list=system\n"));
140 EXPECT_TRUE(image_config.LoadDynamicPartitionMetadata(store));
141 EXPECT_NE(nullptr, image_config.dynamic_partition_metadata);
142
143 EXPECT_FALSE(image_config.ValidateDynamicPartitionMetadata());
144 }
145 } // namespace chromeos_update_engine
146