mirror of https://github.com/grpc/grpc.git
rmstar
5 years ago
committed by
GitHub
commit
1c89c363b3
8 changed files with 298 additions and 2 deletions
@ -0,0 +1,150 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#include <grpc/grpc.h> |
||||
#include <grpc/support/log.h> |
||||
#include <gtest/gtest.h> |
||||
|
||||
#include "src/core/lib/iomgr/closure.h" |
||||
#include "src/core/lib/iomgr/error.h" |
||||
#include "src/core/lib/iomgr/exec_ctx.h" |
||||
#include "src/core/lib/iomgr/timer.h" |
||||
#include "src/core/lib/iomgr/timer_manager.h" |
||||
#include "test/core/util/test_config.h" |
||||
|
||||
// MAYBE_SKIP_TEST is a macro to determine if this particular test configuration
|
||||
// should be skipped based on a decision made at SetUp time.
|
||||
#define MAYBE_SKIP_TEST \ |
||||
do { \
|
||||
if (do_not_test_) { \
|
||||
return; \
|
||||
} \
|
||||
} while (0) |
||||
|
||||
class TimerTest : public ::testing::Test { |
||||
protected: |
||||
void SetUp() override { |
||||
// Skip test if slowdown factor > 1.
|
||||
do_not_test_ = (grpc_test_slowdown_factor() != 1); |
||||
grpc_init(); |
||||
} |
||||
|
||||
void TearDown() override { grpc_shutdown_blocking(); } |
||||
|
||||
bool do_not_test_{false}; |
||||
}; |
||||
|
||||
TEST_F(TimerTest, NoTimers) { |
||||
MAYBE_SKIP_TEST; |
||||
grpc_core::ExecCtx exec_ctx; |
||||
gpr_sleep_until(grpc_timeout_milliseconds_to_deadline(1500)); |
||||
|
||||
// We expect to get 1 wakeup per second. Sometimes we also get a wakeup
|
||||
// during initialization, so in 1.5 seconds we expect to get 1 or 2 wakeups.
|
||||
int64_t wakeups = grpc_timer_manager_get_wakeups_testonly(); |
||||
GPR_ASSERT(wakeups == 1 || wakeups == 2); |
||||
} |
||||
|
||||
TEST_F(TimerTest, OneTimerExpires) { |
||||
MAYBE_SKIP_TEST; |
||||
grpc_core::ExecCtx exec_ctx; |
||||
grpc_timer timer; |
||||
int timer_fired = 0; |
||||
grpc_timer_init(&timer, 500, |
||||
GRPC_CLOSURE_CREATE( |
||||
[](void* arg, grpc_error*) { |
||||
int* timer_fired = static_cast<int*>(arg); |
||||
++*timer_fired; |
||||
}, |
||||
&timer_fired, grpc_schedule_on_exec_ctx)); |
||||
gpr_sleep_until(grpc_timeout_milliseconds_to_deadline(1500)); |
||||
GPR_ASSERT(1 == timer_fired); |
||||
|
||||
// We expect to get 1 wakeup/second + 1 wakeup for the expired timer + maybe 1
|
||||
// wakeup during initialization. i.e. in 1.5 seconds we expect 2 or 3 wakeups.
|
||||
// Actual number of wakeups is more due to bug
|
||||
// https://github.com/grpc/grpc/issues/19947
|
||||
int64_t wakeups = grpc_timer_manager_get_wakeups_testonly(); |
||||
gpr_log(GPR_DEBUG, "wakeups: %" PRId64 "", wakeups); |
||||
} |
||||
|
||||
TEST_F(TimerTest, MultipleTimersExpire) { |
||||
MAYBE_SKIP_TEST; |
||||
grpc_core::ExecCtx exec_ctx; |
||||
const int kNumTimers = 10; |
||||
grpc_timer timers[kNumTimers]; |
||||
int timer_fired = 0; |
||||
for (int i = 0; i < kNumTimers; ++i) { |
||||
grpc_timer_init(&timers[i], 500 + i, |
||||
GRPC_CLOSURE_CREATE( |
||||
[](void* arg, grpc_error*) { |
||||
int* timer_fired = static_cast<int*>(arg); |
||||
++*timer_fired; |
||||
}, |
||||
&timer_fired, grpc_schedule_on_exec_ctx)); |
||||
} |
||||
|
||||
gpr_sleep_until(grpc_timeout_milliseconds_to_deadline(1500)); |
||||
GPR_ASSERT(kNumTimers == timer_fired); |
||||
|
||||
// We expect to get 1 wakeup/second + 1 wakeup for per timer fired + maybe 1
|
||||
// wakeup during initialization. i.e. in 1.5 seconds we expect 11 or 12
|
||||
// wakeups. Actual number of wakeups is more due to bug
|
||||
// https://github.com/grpc/grpc/issues/19947
|
||||
int64_t wakeups = grpc_timer_manager_get_wakeups_testonly(); |
||||
gpr_log(GPR_DEBUG, "wakeups: %" PRId64 "", wakeups); |
||||
} |
||||
|
||||
TEST_F(TimerTest, CancelSomeTimers) { |
||||
MAYBE_SKIP_TEST; |
||||
grpc_core::ExecCtx exec_ctx; |
||||
const int kNumTimers = 10; |
||||
grpc_timer timers[kNumTimers]; |
||||
int timer_fired = 0; |
||||
for (int i = 0; i < kNumTimers; ++i) { |
||||
grpc_timer_init(&timers[i], 500 + i, |
||||
GRPC_CLOSURE_CREATE( |
||||
[](void* arg, grpc_error* error) { |
||||
if (error == GRPC_ERROR_CANCELLED) { |
||||
return; |
||||
} |
||||
int* timer_fired = static_cast<int*>(arg); |
||||
++*timer_fired; |
||||
}, |
||||
&timer_fired, grpc_schedule_on_exec_ctx)); |
||||
} |
||||
for (int i = 0; i < kNumTimers / 2; ++i) { |
||||
grpc_timer_cancel(&timers[i]); |
||||
} |
||||
|
||||
gpr_sleep_until(grpc_timeout_milliseconds_to_deadline(1500)); |
||||
GPR_ASSERT(kNumTimers / 2 == timer_fired); |
||||
|
||||
// We expect to get 1 wakeup/second + 1 wakeup per timer fired + maybe 1
|
||||
// wakeup during initialization. i.e. in 1.5 seconds we expect 6 or 7 wakeups.
|
||||
// Actual number of wakeups is more due to bug
|
||||
// https://github.com/grpc/grpc/issues/19947
|
||||
int64_t wakeups = grpc_timer_manager_get_wakeups_testonly(); |
||||
gpr_log(GPR_DEBUG, "wakeups: %" PRId64 "", wakeups); |
||||
} |
||||
|
||||
int main(int argc, char** argv) { |
||||
grpc::testing::TestEnvironment env(argc, argv); |
||||
::testing::InitGoogleTest(&argc, argv); |
||||
return RUN_ALL_TESTS(); |
||||
} |
||||
Loading…
Reference in new issue