1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/ntp_snippets/remote/request_throttler.h"
#include <memory>
#include "base/strings/stringprintf.h"
#include "components/ntp_snippets/pref_names.h"
#include "components/prefs/pref_registry_simple.h"
#include "components/prefs/testing_pref_service.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
const int kCounterQuota = 2;
} // namespace
namespace ntp_snippets {
class RequestThrottlerTest : public testing::Test {
public:
RequestThrottlerTest() {
RequestThrottler::RegisterProfilePrefs(test_prefs_.registry());
// Use any arbitrary RequestType for this unittest.
throttler_.reset(new RequestThrottler(
&test_prefs_, RequestThrottler::RequestType::
CONTENT_SUGGESTION_FETCHER_ACTIVE_NTP_USER));
throttler_->quota_ = kCounterQuota;
}
protected:
TestingPrefServiceSimple test_prefs_;
std::unique_ptr<RequestThrottler> throttler_;
private:
DISALLOW_COPY_AND_ASSIGN(RequestThrottlerTest);
};
TEST_F(RequestThrottlerTest, QuotaExceeded) {
EXPECT_TRUE(throttler_->DemandQuotaForRequest(false));
EXPECT_TRUE(throttler_->DemandQuotaForRequest(false));
EXPECT_FALSE(throttler_->DemandQuotaForRequest(false));
}
TEST_F(RequestThrottlerTest, ForcedDoesNotCountInQuota) {
EXPECT_TRUE(throttler_->DemandQuotaForRequest(false));
EXPECT_TRUE(throttler_->DemandQuotaForRequest(true));
EXPECT_TRUE(throttler_->DemandQuotaForRequest(false));
}
TEST_F(RequestThrottlerTest, ForcedWorksRegardlessOfQuota) {
EXPECT_TRUE(throttler_->DemandQuotaForRequest(false));
EXPECT_TRUE(throttler_->DemandQuotaForRequest(false));
EXPECT_TRUE(throttler_->DemandQuotaForRequest(true));
}
TEST_F(RequestThrottlerTest, QuotaIsPerDay) {
EXPECT_TRUE(throttler_->DemandQuotaForRequest(false));
EXPECT_TRUE(throttler_->DemandQuotaForRequest(false));
// Now fake the day pref so that the counter believes the count comes from
// yesterday.
int now_day = (base::Time::Now() - base::Time::UnixEpoch()).InDays();
test_prefs_.SetInteger(ntp_snippets::prefs::kSnippetFetcherRequestsDay,
now_day - 1);
// The quota should get reset as the day has changed.
EXPECT_TRUE(throttler_->DemandQuotaForRequest(false));
}
} // namespace ntp_snippets
|
