1 /*
2 * Copyright (C) 2020 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 "apexutil.h"
18
19 #include <unistd.h>
20
21 #include <string>
22 #include <utility>
23 #include <vector>
24
25 #include <android-base/file.h>
26 #include <apex_manifest.pb.h>
27 #include <gmock/gmock.h>
28 #include <gtest/gtest.h>
29
30 using namespace std::literals;
31
32 using ::android::apex::GetActivePackages;
33 using ::android::base::WriteStringToFile;
34 using ::apex::proto::ApexManifest;
35 using ::testing::Contains;
36 using ::testing::Pair;
37 using ::testing::UnorderedElementsAre;
38
39 namespace {
40
CreateApexManifest(std::string apex_name,int version)41 ApexManifest CreateApexManifest(std::string apex_name, int version) {
42 ApexManifest manifest;
43 manifest.set_name(apex_name);
44 manifest.set_version(version);
45 return manifest;
46 }
47
Mkdir(std::string dir_path)48 void Mkdir(std::string dir_path) {
49 if (access(dir_path.c_str(), F_OK) == 0)
50 return;
51 ASSERT_NE(-1, mkdir(dir_path.c_str(), 0755) == -1)
52 << "Failed to create a directory: " << dir_path;
53 }
54
WriteFile(std::string file_path,std::string content)55 void WriteFile(std::string file_path, std::string content) {
56 ASSERT_TRUE(WriteStringToFile(content, file_path))
57 << "Failed to write a file: " << file_path;
58 }
59
60 } // namespace
61
62 namespace apex {
63 namespace proto {
64
65 // simple implementation for testing purpose
operator ==(const ApexManifest & lhs,const ApexManifest & rhs)66 bool operator==(const ApexManifest &lhs, const ApexManifest &rhs) {
67 return lhs.SerializeAsString() == rhs.SerializeAsString();
68 }
69
70 } // namespace proto
71 } // namespace apex
72
TEST(ApexUtil,GetActivePackages)73 TEST(ApexUtil, GetActivePackages) {
74 TemporaryDir td;
75
76 // com.android.foo: valid
77 auto foo_path = td.path + "/com.android.foo"s;
78 auto foo_manifest = CreateApexManifest("com.android.foo", 1);
79 Mkdir(foo_path);
80 WriteFile(foo_path + "/apex_manifest.pb", foo_manifest.SerializeAsString());
81
82 // com.android.foo@1: ignore versioned
83 Mkdir(foo_path + "@1");
84 WriteFile(foo_path + "@1/apex_manifest.pb", foo_manifest.SerializeAsString());
85
86 // com.android.baz: valid
87 auto bar_path = td.path + "/com.android.bar"s;
88 auto bar_manifest = CreateApexManifest("com.android.bar", 2);
89 Mkdir(bar_path);
90 WriteFile(bar_path + "/apex_manifest.pb", bar_manifest.SerializeAsString());
91
92 // invalid: a directory without apex_manifest.pb
93 Mkdir(td.path + "/com.android.baz"s);
94 // invalid: a file
95 WriteFile(td.path + "/com.android.qux.apex"s, "");
96
97 auto apexes = GetActivePackages(td.path);
98 ASSERT_EQ(2U, apexes.size());
99
100 ASSERT_THAT(apexes, UnorderedElementsAre(Pair(foo_path, foo_manifest),
101 Pair(bar_path, bar_manifest)));
102 }