1 /*
2 * Copyright (C) 2018 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 "src/perfetto_cmd/rate_limiter.h"
18
19 #include "test/gtest_and_gmock.h"
20
21 using testing::_;
22 using testing::Contains;
23 using testing::Invoke;
24 using testing::NiceMock;
25 using testing::Return;
26 using testing::StrictMock;
27
28 namespace perfetto {
29 namespace {
30
TEST(RateLimiterTest,CantTraceOnUser)31 TEST(RateLimiterTest, CantTraceOnUser) {
32 RateLimiter limiter;
33 RateLimiter::Args args;
34
35 args.is_user_build = true;
36 args.allow_user_build_tracing = false;
37 args.is_uploading = true;
38
39 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kNotAllowedOnUserBuild);
40 }
41
TEST(RateLimiterTest,CanTraceOnUser)42 TEST(RateLimiterTest, CanTraceOnUser) {
43 RateLimiter limiter;
44 RateLimiter::Args args;
45
46 args.is_user_build = false;
47 args.allow_user_build_tracing = false;
48 args.is_uploading = true;
49
50 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kOkToTrace);
51 }
52
53 } // namespace
54
55 } // namespace perfetto
56