Lines Matching refs:limiter
80 NiceMock<MockRateLimiter> limiter; in TEST() local
86 ASSERT_TRUE(limiter.SaveState(input)); in TEST()
87 ASSERT_TRUE(limiter.LoadState(&output)); in TEST()
93 NiceMock<MockRateLimiter> limiter; in TEST() local
109 ASSERT_TRUE(limiter.SaveState(input)); in TEST()
110 ASSERT_TRUE(limiter.LoadState(&output)); in TEST()
128 NiceMock<MockRateLimiter> limiter; in TEST() local
136 ASSERT_TRUE(limiter.SaveState(input)); in TEST()
137 ASSERT_TRUE(limiter.LoadState(&output)); in TEST()
142 NiceMock<MockRateLimiter> limiter; in TEST() local
144 ASSERT_FALSE(limiter.LoadState(&output)); in TEST()
149 NiceMock<MockRateLimiter> limiter; in TEST() local
151 WriteGarbageToFile(limiter.GetStateFilePath().c_str()); in TEST()
154 ASSERT_FALSE(limiter.LoadState(&output)); in TEST()
159 StrictMock<MockRateLimiter> limiter; in TEST() local
161 ASSERT_EQ(limiter.ShouldTrace({}), RateLimiter::kOkToTrace); in TEST()
162 ASSERT_TRUE(limiter.OnTraceDone({}, true, 10000)); in TEST()
163 ASSERT_FALSE(limiter.StateFileExists()); in TEST()
167 StrictMock<MockRateLimiter> limiter; in TEST() local
169 ASSERT_FALSE(limiter.OnTraceDone({}, false, 0)); in TEST()
170 ASSERT_FALSE(limiter.StateFileExists()); in TEST()
174 StrictMock<MockRateLimiter> limiter; in TEST() local
182 EXPECT_CALL(limiter, SaveState(_)); in TEST()
183 EXPECT_CALL(limiter, LoadState(_)); in TEST()
184 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kOkToTrace); in TEST()
186 EXPECT_CALL(limiter, SaveState(_)); in TEST()
187 ASSERT_TRUE(limiter.OnTraceDone(args, true, 42u)); in TEST()
190 ASSERT_TRUE(limiter.LoadStateConcrete(&output)); in TEST()
197 StrictMock<MockRateLimiter> limiter; in TEST() local
204 EXPECT_CALL(limiter, SaveState(_)); in TEST()
205 EXPECT_CALL(limiter, LoadState(_)); in TEST()
206 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kOkToTrace); in TEST()
208 EXPECT_CALL(limiter, SaveState(_)); in TEST()
209 ASSERT_TRUE(limiter.OnTraceDone(args, true, 1024 * 1024)); in TEST()
212 ASSERT_TRUE(limiter.LoadStateConcrete(&output)); in TEST()
219 StrictMock<MockRateLimiter> limiter; in TEST() local
226 ASSERT_TRUE(limiter.SaveStateConcrete(input)); in TEST()
232 EXPECT_CALL(limiter, LoadState(_)); in TEST()
233 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kOkToTrace); in TEST()
235 EXPECT_CALL(limiter, SaveState(_)); in TEST()
236 ASSERT_TRUE(limiter.OnTraceDone(args, true, 1024 * 1024)); in TEST()
239 ASSERT_TRUE(limiter.LoadStateConcrete(&output)); in TEST()
247 StrictMock<MockRateLimiter> limiter; in TEST() local
254 ASSERT_TRUE(limiter.SaveStateConcrete(input)); in TEST()
261 EXPECT_CALL(limiter, LoadState(_)); in TEST()
262 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kOkToTrace); in TEST()
264 EXPECT_CALL(limiter, SaveState(_)); in TEST()
265 ASSERT_TRUE(limiter.OnTraceDone(args, true, 1024 * 1024)); in TEST()
268 ASSERT_TRUE(limiter.LoadStateConcrete(&output)); in TEST()
286 StrictMock<MockRateLimiter> limiter; in TEST() local
292 WriteGarbageToFile(limiter.GetStateFilePath().c_str()); in TEST()
294 EXPECT_CALL(limiter, LoadState(_)); in TEST()
295 EXPECT_CALL(limiter, SaveState(_)); in TEST()
296 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kInvalidState); in TEST()
299 ASSERT_TRUE(limiter.LoadStateConcrete(&output)); in TEST()
306 StrictMock<MockRateLimiter> limiter; in TEST() local
312 ASSERT_TRUE(limiter.SaveStateConcrete(input)); in TEST()
318 EXPECT_CALL(limiter, LoadState(_)); in TEST()
319 EXPECT_CALL(limiter, SaveState(_)); in TEST()
320 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kInvalidState); in TEST()
323 ASSERT_TRUE(limiter.LoadStateConcrete(&output)); in TEST()
330 StrictMock<MockRateLimiter> limiter; in TEST() local
338 ASSERT_TRUE(limiter.SaveStateConcrete(input)); in TEST()
346 EXPECT_CALL(limiter, LoadState(_)); in TEST()
347 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kOkToTrace); in TEST()
351 StrictMock<MockRateLimiter> limiter; in TEST() local
359 ASSERT_TRUE(limiter.SaveStateConcrete(input)); in TEST()
367 EXPECT_CALL(limiter, LoadState(_)); in TEST()
368 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kHitUploadLimit); in TEST()
372 StrictMock<MockRateLimiter> limiter; in TEST() local
377 ASSERT_TRUE(limiter.SaveStateConcrete(input)); in TEST()
384 EXPECT_CALL(limiter, LoadState(_)); in TEST()
385 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kHitUploadLimit); in TEST()
389 StrictMock<MockRateLimiter> limiter; in TEST() local
396 ASSERT_TRUE(limiter.SaveStateConcrete(input)); in TEST()
404 EXPECT_CALL(limiter, LoadState(_)); in TEST()
405 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kOkToTrace); in TEST()
410 StrictMock<MockRateLimiter> limiter; in TEST() local
415 ASSERT_TRUE(limiter.SaveStateConcrete(input)); in TEST()
422 EXPECT_CALL(limiter, LoadState(_)); in TEST()
423 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kHitUploadLimit); in TEST()
427 StrictMock<MockRateLimiter> limiter; in TEST() local
434 ASSERT_TRUE(limiter.SaveStateConcrete(input)); in TEST()
439 EXPECT_CALL(limiter, LoadState(_)); in TEST()
440 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kOkToTrace); in TEST()
442 EXPECT_CALL(limiter, SaveState(_)); in TEST()
443 ASSERT_TRUE(limiter.OnTraceDone(args, true, 1024 * 1024)); in TEST()
446 ASSERT_TRUE(limiter.LoadStateConcrete(&output)); in TEST()
455 StrictMock<MockRateLimiter> limiter; in TEST() local
461 EXPECT_CALL(limiter, SaveState(_)); in TEST()
462 EXPECT_CALL(limiter, LoadState(_)); in TEST()
463 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kOkToTrace); in TEST()
464 ASSERT_FALSE(limiter.OnTraceDone(args, false, 1024 * 1024)); in TEST()
468 StrictMock<MockRateLimiter> limiter; in TEST() local
474 EXPECT_CALL(limiter, SaveState(_)); in TEST()
475 EXPECT_CALL(limiter, LoadState(_)); in TEST()
476 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kOkToTrace); in TEST()
478 EXPECT_CALL(limiter, SaveState(_)).WillOnce(Return(false)); in TEST()
479 ASSERT_FALSE(limiter.OnTraceDone(args, true, 1024 * 1024)); in TEST()
483 StrictMock<MockRateLimiter> limiter; in TEST() local
491 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kNotAllowedOnUserBuild); in TEST()
495 StrictMock<MockRateLimiter> limiter; in TEST() local
503 EXPECT_CALL(limiter, SaveState(_)); in TEST()
504 EXPECT_CALL(limiter, LoadState(_)); in TEST()
505 ASSERT_EQ(limiter.ShouldTrace(args), RateLimiter::kOkToTrace); in TEST()