mirror of https://github.com/grpc/grpc.git
rmstar
6 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