mirror of https://github.com/grpc/grpc.git
parent
7564664f07
commit
d2b5fd75f0
3 changed files with 145 additions and 9 deletions
@ -0,0 +1,128 @@ |
||||
/*
|
||||
* |
||||
* 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 <spawn.h> |
||||
#include <sstream> |
||||
#include <string> |
||||
#include <thread> |
||||
#include <vector> |
||||
|
||||
#include <grpc/grpc.h> |
||||
#include <grpc/support/log.h> |
||||
#include <gtest/gtest.h> |
||||
|
||||
#include "src/core/lib/gprpp/sync.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" |
||||
|
||||
extern char** environ; |
||||
|
||||
void run_cmd(const char* cmd) { |
||||
pid_t pid; |
||||
const char* argv[] = {const_cast<const char*>("sh"), |
||||
const_cast<const char*>("-c"), cmd, nullptr}; |
||||
int status; |
||||
|
||||
status = posix_spawn(&pid, const_cast<const char*>("/bin/sh"), nullptr, |
||||
nullptr, const_cast<char**>(argv), environ); |
||||
if (status == 0) { |
||||
if (waitpid(pid, &status, 0) == -1) { |
||||
perror("waitpid"); |
||||
} |
||||
} |
||||
} |
||||
|
||||
class TimeJumpTest : public ::testing::TestWithParam<std::string> { |
||||
protected: |
||||
void SetUp() override { grpc_init(); } |
||||
void TearDown() override { |
||||
run_cmd("sudo sntp -sS pool.ntp.org"); |
||||
grpc_shutdown_blocking(); |
||||
} |
||||
|
||||
const int kWaitTimeMs = 1500; |
||||
}; |
||||
|
||||
std::vector<std::string> CreateTestScenarios() { |
||||
return {"-1M", "+1M", "-1H", "+1H", "-1d", "+1d", "-1y", "+1y"}; |
||||
} |
||||
INSTANTIATE_TEST_CASE_P(TimeJump, TimeJumpTest, |
||||
::testing::ValuesIn(CreateTestScenarios())); |
||||
|
||||
TEST_P(TimeJumpTest, TimerRunning) { |
||||
grpc_core::ExecCtx exec_ctx; |
||||
grpc_timer timer; |
||||
grpc_timer_init(&timer, grpc_core::ExecCtx::Get()->Now() + 3000, |
||||
GRPC_CLOSURE_CREATE( |
||||
[](void*, grpc_error* error) { |
||||
GPR_ASSERT(error == GRPC_ERROR_CANCELLED); |
||||
}, |
||||
nullptr, grpc_schedule_on_exec_ctx)); |
||||
gpr_sleep_until(grpc_timeout_milliseconds_to_deadline(100)); |
||||
std::ostringstream cmd; |
||||
cmd << "sudo date `date -v" << GetParam() << " \"+%m%d%H%M%y\"`"; |
||||
run_cmd(cmd.str().c_str()); |
||||
gpr_sleep_until(grpc_timeout_milliseconds_to_deadline(kWaitTimeMs)); |
||||
// We expect 1 wakeup/sec when there are not timer expiries
|
||||
int64_t wakeups = grpc_timer_manager_get_wakeups_testonly(); |
||||
gpr_log(GPR_DEBUG, "wakeups: %" PRId64 "", wakeups); |
||||
GPR_ASSERT(wakeups <= 3); |
||||
grpc_timer_cancel(&timer); |
||||
} |
||||
|
||||
TEST_P(TimeJumpTest, TimedWait) { |
||||
grpc_core::CondVar cond; |
||||
grpc_core::Mutex mu; |
||||
{ |
||||
grpc_core::MutexLock lock(&mu); |
||||
std::thread thd = std::thread([]() { |
||||
gpr_sleep_until(grpc_timeout_milliseconds_to_deadline(100)); |
||||
std::ostringstream cmd; |
||||
cmd << "sudo date `date -v" << GetParam() << " \"+%m%d%H%M%y\"`"; |
||||
run_cmd(cmd.str().c_str()); |
||||
}); |
||||
gpr_timespec before = gpr_now(GPR_CLOCK_MONOTONIC); |
||||
int timedout = cond.Wait( |
||||
&mu, grpc_millis_to_timespec(kWaitTimeMs, GPR_CLOCK_REALTIME)); |
||||
gpr_timespec after = gpr_now(GPR_CLOCK_MONOTONIC); |
||||
int32_t elapsed_ms = gpr_time_to_millis(gpr_time_sub(after, before)); |
||||
gpr_log(GPR_DEBUG, "After wait, timedout = %d elapsed_ms = %d", timedout, |
||||
elapsed_ms); |
||||
GPR_ASSERT(1 == timedout); |
||||
GPR_ASSERT(1 == |
||||
gpr_time_similar(gpr_time_sub(after, before), |
||||
gpr_time_from_millis(kWaitTimeMs, GPR_TIMESPAN), |
||||
gpr_time_from_millis(10, GPR_TIMESPAN))); |
||||
|
||||
thd.join(); |
||||
} |
||||
// We expect 1 wakeup/sec when there are not timer expiries
|
||||
int64_t wakeups = grpc_timer_manager_get_wakeups_testonly(); |
||||
gpr_log(GPR_DEBUG, "wakeups: %" PRId64 "", wakeups); |
||||
GPR_ASSERT(wakeups <= 3); |
||||
} |
||||
|
||||
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