1 /*
2 * Copyright (C) 2024 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 "chre_api/chre.h"
18 #include "gtest/gtest.h"
19
20 #include "chre/platform/linux/system_time.h"
21 #include "chre/util/duplicate_message_detector.h"
22
23 using chre::platform_linux::SystemTimeOverride;
24
25 namespace chre {
26
27 constexpr Nanoseconds kTimeout = Nanoseconds(100);
28 constexpr uint32_t kNumMessages = 100;
29
TEST(DuplicateMessageDetectorTest,AddMessageCanBeFound)30 TEST(DuplicateMessageDetectorTest, AddMessageCanBeFound) {
31 DuplicateMessageDetector duplicateMessageDetector(kTimeout);
32 uint32_t messageSequenceNumber = 1;
33 uint16_t hostEndpoint = 2;
34
35 EXPECT_FALSE(duplicateMessageDetector.findOrAdd(messageSequenceNumber,
36 hostEndpoint).has_value());
37 }
38
TEST(DuplicateMessageDetectorTest,AddMultipleCanBeFound)39 TEST(DuplicateMessageDetectorTest, AddMultipleCanBeFound) {
40 DuplicateMessageDetector duplicateMessageDetector(kTimeout);
41 for (size_t i = 0; i < kNumMessages; ++i) {
42 EXPECT_FALSE(duplicateMessageDetector.findOrAdd(i, i).has_value());
43 }
44 }
45
TEST(DuplicateMessageDetectorTest,RemoveOldEntries)46 TEST(DuplicateMessageDetectorTest, RemoveOldEntries) {
47 DuplicateMessageDetector duplicateMessageDetector(kTimeout);
48
49 for (size_t i = 0; i < kNumMessages; ++i) {
50 SystemTimeOverride override(i);
51 EXPECT_FALSE(duplicateMessageDetector.findOrAdd(i, kNumMessages - i)
52 .has_value());
53 }
54
55 SystemTimeOverride override(kTimeout * 10);
56 duplicateMessageDetector.removeOldEntries();
57
58 for (size_t i = 0; i < kNumMessages; ++i) {
59 EXPECT_FALSE(duplicateMessageDetector.findAndSetError(i,
60 kNumMessages - i,
61 CHRE_ERROR_NONE));
62 }
63 }
64
TEST(DuplicateMessageDetectorTest,RemoveOldEntriesDoesNotRemoveRecentEntries)65 TEST(DuplicateMessageDetectorTest, RemoveOldEntriesDoesNotRemoveRecentEntries) {
66 DuplicateMessageDetector duplicateMessageDetector(kTimeout);
67
68 for (size_t i = 0; i < kNumMessages; ++i) {
69 SystemTimeOverride override(i);
70 EXPECT_FALSE(duplicateMessageDetector.findOrAdd(i, i).has_value());
71 }
72
73 {
74 constexpr uint32_t kNumMessagesToRemove = kNumMessages / 2;
75 SystemTimeOverride override(kNumMessagesToRemove +
76 kTimeout.toRawNanoseconds());
77 duplicateMessageDetector.removeOldEntries();
78
79 for (size_t i = 0; i <= kNumMessagesToRemove; ++i) {
80 EXPECT_FALSE(duplicateMessageDetector.findAndSetError(i, i,
81 CHRE_ERROR_NONE));
82 }
83 for (size_t i = kNumMessagesToRemove + 1; i < kNumMessages; ++i) {
84 bool isDuplicate = false;
85 EXPECT_FALSE(
86 duplicateMessageDetector.findOrAdd(i, i, &isDuplicate).has_value());
87 EXPECT_TRUE(isDuplicate);
88 EXPECT_TRUE(duplicateMessageDetector.findAndSetError(i, i,
89 CHRE_ERROR_NONE));
90
91 isDuplicate = false;
92 Optional<chreError> error =
93 duplicateMessageDetector.findOrAdd(i, i, &isDuplicate);
94 EXPECT_TRUE(error.has_value());
95 EXPECT_EQ(error.value(), CHRE_ERROR_NONE);
96 EXPECT_TRUE(isDuplicate);
97 }
98 }
99 }
100
101 } // namespace chre
102