1 /*
2  * Copyright (C) 2013 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 "dedupe_set.h"
18 
19 #include <algorithm>
20 #include <cstdio>
21 #include <vector>
22 
23 #include "base/array_ref.h"
24 #include "dedupe_set-inl.h"
25 #include "gtest/gtest.h"
26 #include "thread-inl.h"
27 
28 namespace art {
29 
30 class DedupeSetTestHashFunc {
31  public:
operator ()(const ArrayRef<const uint8_t> & array) const32   size_t operator()(const ArrayRef<const uint8_t>& array) const {
33     size_t hash = 0;
34     for (uint8_t c : array) {
35       hash += c;
36       hash += hash << 10;
37       hash += hash >> 6;
38     }
39     return hash;
40   }
41 };
42 
43 class DedupeSetTestAlloc {
44  public:
Copy(const ArrayRef<const uint8_t> & src)45   const std::vector<uint8_t>* Copy(const ArrayRef<const uint8_t>& src) {
46     return new std::vector<uint8_t>(src.begin(), src.end());
47   }
48 
Destroy(const std::vector<uint8_t> * key)49   void Destroy(const std::vector<uint8_t>* key) {
50     delete key;
51   }
52 };
53 
TEST(DedupeSetTest,Test)54 TEST(DedupeSetTest, Test) {
55   Thread* self = Thread::Current();
56   DedupeSetTestAlloc alloc;
57   DedupeSet<ArrayRef<const uint8_t>,
58             std::vector<uint8_t>,
59             DedupeSetTestAlloc,
60             size_t,
61             DedupeSetTestHashFunc> deduplicator("test", alloc);
62   const std::vector<uint8_t>* array1;
63   {
64     uint8_t raw_test1[] = { 10u, 20u, 30u, 45u };
65     ArrayRef<const uint8_t> test1(raw_test1);
66     array1 = deduplicator.Add(self, test1);
67     ASSERT_NE(array1, nullptr);
68     ASSERT_TRUE(std::equal(test1.begin(), test1.end(), array1->begin()));
69   }
70 
71   const std::vector<uint8_t>* array2;
72   {
73     uint8_t raw_test2[] = { 10u, 20u, 30u, 45u };
74     ArrayRef<const uint8_t> test2(raw_test2);
75     array2 = deduplicator.Add(self, test2);
76     ASSERT_EQ(array2, array1);
77     ASSERT_TRUE(std::equal(test2.begin(), test2.end(), array2->begin()));
78   }
79 
80   const std::vector<uint8_t>* array3;
81   {
82     uint8_t raw_test3[] = { 10u, 22u, 30u, 47u };
83     ArrayRef<const uint8_t> test3(raw_test3);
84     array3 = deduplicator.Add(self, test3);
85     ASSERT_NE(array3, nullptr);
86     ASSERT_NE(array3, array1);
87     ASSERT_TRUE(std::equal(test3.begin(), test3.end(), array3->begin()));
88   }
89 }
90 
91 }  // namespace art
92