1 /*
2 * Copyright (C) 2016 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 "androidfw/AttributeFinder.h"
18
19 #include "android-base/macros.h"
20 #include "gtest/gtest.h"
21
22 namespace android {
23
24 class MockAttributeFinder
25 : public BackTrackingAttributeFinder<MockAttributeFinder, int> {
26 public:
MockAttributeFinder(const uint32_t * attrs,int len)27 MockAttributeFinder(const uint32_t* attrs, int len)
28 : BackTrackingAttributeFinder(0, len) {
29 attrs_ = new uint32_t[len];
30 memcpy(attrs_, attrs, sizeof(*attrs) * len);
31 }
32
~MockAttributeFinder()33 ~MockAttributeFinder() { delete attrs_; }
34
GetAttribute(const int index) const35 inline uint32_t GetAttribute(const int index) const { return attrs_[index]; }
36
37 private:
38 uint32_t* attrs_;
39 };
40
41 static const uint32_t kSortedAttributes[] = {0x01010000, 0x01010001, 0x01010002,
42 0x01010004, 0x02010001, 0x02010010,
43 0x7f010001};
44
45 static const uint32_t kPackageUnsortedAttributes[] = {
46 0x02010001, 0x02010010, 0x01010000, 0x01010001,
47 0x01010002, 0x01010004, 0x7f010001};
48
49 static const uint32_t kSinglePackageAttributes[] = {0x7f010007, 0x7f01000a,
50 0x7f01000d, 0x00000000};
51
TEST(AttributeFinderTest,IteratesSequentially)52 TEST(AttributeFinderTest, IteratesSequentially) {
53 const int end = arraysize(kSortedAttributes);
54 MockAttributeFinder finder(kSortedAttributes, end);
55
56 EXPECT_EQ(0, finder.Find(0x01010000));
57 EXPECT_EQ(1, finder.Find(0x01010001));
58 EXPECT_EQ(2, finder.Find(0x01010002));
59 EXPECT_EQ(3, finder.Find(0x01010004));
60 EXPECT_EQ(4, finder.Find(0x02010001));
61 EXPECT_EQ(5, finder.Find(0x02010010));
62 EXPECT_EQ(6, finder.Find(0x7f010001));
63 EXPECT_EQ(end, finder.Find(0x7f010002));
64 }
65
TEST(AttributeFinderTest,PackagesAreOutOfOrder)66 TEST(AttributeFinderTest, PackagesAreOutOfOrder) {
67 const int end = arraysize(kSortedAttributes);
68 MockAttributeFinder finder(kSortedAttributes, end);
69
70 EXPECT_EQ(6, finder.Find(0x7f010001));
71 EXPECT_EQ(end, finder.Find(0x7f010002));
72 EXPECT_EQ(4, finder.Find(0x02010001));
73 EXPECT_EQ(5, finder.Find(0x02010010));
74 EXPECT_EQ(0, finder.Find(0x01010000));
75 EXPECT_EQ(1, finder.Find(0x01010001));
76 EXPECT_EQ(2, finder.Find(0x01010002));
77 EXPECT_EQ(3, finder.Find(0x01010004));
78 }
79
TEST(AttributeFinderTest,SomeAttributesAreNotFound)80 TEST(AttributeFinderTest, SomeAttributesAreNotFound) {
81 const int end = arraysize(kSortedAttributes);
82 MockAttributeFinder finder(kSortedAttributes, end);
83
84 EXPECT_EQ(0, finder.Find(0x01010000));
85 EXPECT_EQ(1, finder.Find(0x01010001));
86 EXPECT_EQ(2, finder.Find(0x01010002));
87 EXPECT_EQ(end, finder.Find(0x01010003));
88 EXPECT_EQ(3, finder.Find(0x01010004));
89 EXPECT_EQ(end, finder.Find(0x01010005));
90 EXPECT_EQ(end, finder.Find(0x01010006));
91 EXPECT_EQ(4, finder.Find(0x02010001));
92 EXPECT_EQ(end, finder.Find(0x02010002));
93 }
94
TEST(AttributeFinderTest,FindAttributesInPackageUnsortedAttributeList)95 TEST(AttributeFinderTest, FindAttributesInPackageUnsortedAttributeList) {
96 const int end = arraysize(kPackageUnsortedAttributes);
97 MockAttributeFinder finder(kPackageUnsortedAttributes, end);
98
99 EXPECT_EQ(2, finder.Find(0x01010000));
100 EXPECT_EQ(3, finder.Find(0x01010001));
101 EXPECT_EQ(4, finder.Find(0x01010002));
102 EXPECT_EQ(end, finder.Find(0x01010003));
103 EXPECT_EQ(5, finder.Find(0x01010004));
104 EXPECT_EQ(end, finder.Find(0x01010005));
105 EXPECT_EQ(end, finder.Find(0x01010006));
106 EXPECT_EQ(0, finder.Find(0x02010001));
107 EXPECT_EQ(end, finder.Find(0x02010002));
108 EXPECT_EQ(1, finder.Find(0x02010010));
109 EXPECT_EQ(6, finder.Find(0x7f010001));
110 }
111
TEST(AttributeFinderTest,FindAttributesInSinglePackageAttributeList)112 TEST(AttributeFinderTest, FindAttributesInSinglePackageAttributeList) {
113 const int end = arraysize(kSinglePackageAttributes);
114 MockAttributeFinder finder(kSinglePackageAttributes, end);
115
116 EXPECT_EQ(end, finder.Find(0x010100f4));
117 EXPECT_EQ(end, finder.Find(0x010100f5));
118 EXPECT_EQ(end, finder.Find(0x010100f6));
119 EXPECT_EQ(end, finder.Find(0x010100f7));
120 EXPECT_EQ(end, finder.Find(0x010100f8));
121 EXPECT_EQ(end, finder.Find(0x010100fa));
122 EXPECT_EQ(0, finder.Find(0x7f010007));
123 }
124
125 } // namespace android
126