mirror of https://github.com/grpc/grpc.git
Merge pull request #17842 from rmstar/timechange
Added test for wall clock time change on the clientpull/17949/head
rmstar
6 years ago
committed by
GitHub
commit
0d9e6d1e36
9 changed files with 610 additions and 3 deletions
@ -0,0 +1,422 @@ |
||||
/*
|
||||
* |
||||
* 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 <grpc/support/time.h> |
||||
#include <grpcpp/channel.h> |
||||
#include <grpcpp/client_context.h> |
||||
#include <grpcpp/create_channel.h> |
||||
#include <grpcpp/server.h> |
||||
#include <grpcpp/server_builder.h> |
||||
#include <grpcpp/server_context.h> |
||||
|
||||
#include "src/core/lib/iomgr/timer.h" |
||||
#include "src/proto/grpc/testing/echo.grpc.pb.h" |
||||
#include "test/core/util/port.h" |
||||
#include "test/core/util/test_config.h" |
||||
#include "test/cpp/end2end/test_service_impl.h" |
||||
#include "test/cpp/util/subprocess.h" |
||||
|
||||
#include <gtest/gtest.h> |
||||
#include <pthread.h> |
||||
#include <sys/time.h> |
||||
#include <thread> |
||||
|
||||
using grpc::testing::EchoRequest; |
||||
using grpc::testing::EchoResponse; |
||||
|
||||
static std::string g_root; |
||||
|
||||
static gpr_mu g_mu; |
||||
extern gpr_timespec (*gpr_now_impl)(gpr_clock_type clock_type); |
||||
gpr_timespec (*gpr_now_impl_orig)(gpr_clock_type clock_type) = gpr_now_impl; |
||||
static int g_time_shift_sec = 0; |
||||
static int g_time_shift_nsec = 0; |
||||
static gpr_timespec now_impl(gpr_clock_type clock) { |
||||
auto ts = gpr_now_impl_orig(clock); |
||||
// We only manipulate the realtime clock to simulate changes in wall-clock
|
||||
// time
|
||||
if (clock != GPR_CLOCK_REALTIME) { |
||||
return ts; |
||||
} |
||||
GPR_ASSERT(ts.tv_nsec >= 0); |
||||
GPR_ASSERT(ts.tv_nsec < GPR_NS_PER_SEC); |
||||
gpr_mu_lock(&g_mu); |
||||
ts.tv_sec += g_time_shift_sec; |
||||
ts.tv_nsec += g_time_shift_nsec; |
||||
gpr_mu_unlock(&g_mu); |
||||
if (ts.tv_nsec >= GPR_NS_PER_SEC) { |
||||
ts.tv_nsec -= GPR_NS_PER_SEC; |
||||
++ts.tv_sec; |
||||
} else if (ts.tv_nsec < 0) { |
||||
--ts.tv_sec; |
||||
ts.tv_nsec = GPR_NS_PER_SEC + ts.tv_nsec; |
||||
} |
||||
return ts; |
||||
} |
||||
|
||||
// offset the value returned by gpr_now(GPR_CLOCK_REALTIME) by msecs
|
||||
// milliseconds
|
||||
static void set_now_offset(int msecs) { |
||||
g_time_shift_sec = msecs / 1000; |
||||
g_time_shift_nsec = (msecs % 1000) * 1e6; |
||||
} |
||||
|
||||
// restore the original implementation of gpr_now()
|
||||
static void reset_now_offset() { |
||||
g_time_shift_sec = 0; |
||||
g_time_shift_nsec = 0; |
||||
} |
||||
|
||||
namespace grpc { |
||||
namespace testing { |
||||
|
||||
namespace { |
||||
|
||||
// gpr_now() is called with invalid clock_type
|
||||
TEST(TimespecTest, GprNowInvalidClockType) { |
||||
// initialize to some junk value
|
||||
gpr_clock_type invalid_clock_type = (gpr_clock_type)32641; |
||||
EXPECT_DEATH(gpr_now(invalid_clock_type), ".*"); |
||||
} |
||||
|
||||
// Add timespan with negative nanoseconds
|
||||
TEST(TimespecTest, GprTimeAddNegativeNs) { |
||||
gpr_timespec now = gpr_now(GPR_CLOCK_MONOTONIC); |
||||
gpr_timespec bad_ts = {1, -1000, GPR_TIMESPAN}; |
||||
EXPECT_DEATH(gpr_time_add(now, bad_ts), ".*"); |
||||
} |
||||
|
||||
// Subtract timespan with negative nanoseconds
|
||||
TEST(TimespecTest, GprTimeSubNegativeNs) { |
||||
// Nanoseconds must always be positive. Negative timestamps are represented by
|
||||
// (negative seconds, positive nanoseconds)
|
||||
gpr_timespec now = gpr_now(GPR_CLOCK_MONOTONIC); |
||||
gpr_timespec bad_ts = {1, -1000, GPR_TIMESPAN}; |
||||
EXPECT_DEATH(gpr_time_sub(now, bad_ts), ".*"); |
||||
} |
||||
|
||||
// Add negative milliseconds to gpr_timespec
|
||||
TEST(TimespecTest, GrpcNegativeMillisToTimespec) { |
||||
// -1500 milliseconds converts to timespec (-2 secs, 5 * 10^8 nsec)
|
||||
gpr_timespec ts = grpc_millis_to_timespec(-1500, GPR_CLOCK_MONOTONIC); |
||||
GPR_ASSERT(ts.tv_sec = -2); |
||||
GPR_ASSERT(ts.tv_nsec = 5e8); |
||||
GPR_ASSERT(ts.clock_type == GPR_CLOCK_MONOTONIC); |
||||
} |
||||
|
||||
class TimeChangeTest : public ::testing::Test { |
||||
protected: |
||||
TimeChangeTest() {} |
||||
|
||||
void SetUp() { |
||||
auto port = grpc_pick_unused_port_or_die(); |
||||
std::ostringstream addr_stream; |
||||
addr_stream << "localhost:" << port; |
||||
auto addr = addr_stream.str(); |
||||
server_.reset(new SubProcess({ |
||||
g_root + "/client_crash_test_server", |
||||
"--address=" + addr, |
||||
})); |
||||
GPR_ASSERT(server_); |
||||
channel_ = CreateChannel(addr, InsecureChannelCredentials()); |
||||
GPR_ASSERT(channel_); |
||||
stub_ = grpc::testing::EchoTestService::NewStub(channel_); |
||||
} |
||||
|
||||
void TearDown() { |
||||
server_.reset(); |
||||
reset_now_offset(); |
||||
} |
||||
|
||||
std::unique_ptr<grpc::testing::EchoTestService::Stub> CreateStub() { |
||||
return grpc::testing::EchoTestService::NewStub(channel_); |
||||
} |
||||
|
||||
std::shared_ptr<Channel> GetChannel() { return channel_; } |
||||
// time jump offsets in milliseconds
|
||||
const int TIME_OFFSET1 = 20123; |
||||
const int TIME_OFFSET2 = 5678; |
||||
|
||||
private: |
||||
std::unique_ptr<SubProcess> server_; |
||||
std::shared_ptr<Channel> channel_; |
||||
std::unique_ptr<grpc::testing::EchoTestService::Stub> stub_; |
||||
}; |
||||
|
||||
// Wall-clock time jumps forward on client before bidi stream is created
|
||||
TEST_F(TimeChangeTest, TimeJumpForwardBeforeStreamCreated) { |
||||
EchoRequest request; |
||||
EchoResponse response; |
||||
ClientContext context; |
||||
context.set_deadline(grpc_timeout_milliseconds_to_deadline(5000)); |
||||
context.AddMetadata(kServerResponseStreamsToSend, "1"); |
||||
|
||||
auto channel = GetChannel(); |
||||
GPR_ASSERT(channel); |
||||
EXPECT_TRUE( |
||||
channel->WaitForConnected(grpc_timeout_milliseconds_to_deadline(5000))); |
||||
auto stub = CreateStub(); |
||||
|
||||
// time jumps forward by TIME_OFFSET1 milliseconds
|
||||
set_now_offset(TIME_OFFSET1); |
||||
auto stream = stub->BidiStream(&context); |
||||
request.set_message("Hello"); |
||||
EXPECT_TRUE(stream->Write(request)); |
||||
|
||||
EXPECT_TRUE(stream->WritesDone()); |
||||
EXPECT_TRUE(stream->Read(&response)); |
||||
|
||||
auto status = stream->Finish(); |
||||
EXPECT_TRUE(status.ok()); |
||||
} |
||||
|
||||
// Wall-clock time jumps back on client before bidi stream is created
|
||||
TEST_F(TimeChangeTest, TimeJumpBackBeforeStreamCreated) { |
||||
EchoRequest request; |
||||
EchoResponse response; |
||||
ClientContext context; |
||||
context.set_deadline(grpc_timeout_milliseconds_to_deadline(5000)); |
||||
context.AddMetadata(kServerResponseStreamsToSend, "1"); |
||||
|
||||
auto channel = GetChannel(); |
||||
GPR_ASSERT(channel); |
||||
EXPECT_TRUE( |
||||
channel->WaitForConnected(grpc_timeout_milliseconds_to_deadline(5000))); |
||||
auto stub = CreateStub(); |
||||
|
||||
// time jumps back by TIME_OFFSET1 milliseconds
|
||||
set_now_offset(-TIME_OFFSET1); |
||||
auto stream = stub->BidiStream(&context); |
||||
request.set_message("Hello"); |
||||
EXPECT_TRUE(stream->Write(request)); |
||||
|
||||
EXPECT_TRUE(stream->WritesDone()); |
||||
EXPECT_TRUE(stream->Read(&response)); |
||||
EXPECT_EQ(request.message(), response.message()); |
||||
|
||||
auto status = stream->Finish(); |
||||
EXPECT_TRUE(status.ok()); |
||||
} |
||||
|
||||
// Wall-clock time jumps forward on client while call is in progress
|
||||
TEST_F(TimeChangeTest, TimeJumpForwardAfterStreamCreated) { |
||||
EchoRequest request; |
||||
EchoResponse response; |
||||
ClientContext context; |
||||
context.set_deadline(grpc_timeout_milliseconds_to_deadline(5000)); |
||||
context.AddMetadata(kServerResponseStreamsToSend, "2"); |
||||
|
||||
auto channel = GetChannel(); |
||||
GPR_ASSERT(channel); |
||||
EXPECT_TRUE( |
||||
channel->WaitForConnected(grpc_timeout_milliseconds_to_deadline(5000))); |
||||
auto stub = CreateStub(); |
||||
|
||||
auto stream = stub->BidiStream(&context); |
||||
|
||||
request.set_message("Hello"); |
||||
EXPECT_TRUE(stream->Write(request)); |
||||
EXPECT_TRUE(stream->Read(&response)); |
||||
|
||||
// time jumps forward by TIME_OFFSET1 milliseconds.
|
||||
set_now_offset(TIME_OFFSET1); |
||||
|
||||
request.set_message("World"); |
||||
EXPECT_TRUE(stream->Write(request)); |
||||
EXPECT_TRUE(stream->WritesDone()); |
||||
EXPECT_TRUE(stream->Read(&response)); |
||||
|
||||
auto status = stream->Finish(); |
||||
EXPECT_TRUE(status.ok()); |
||||
} |
||||
|
||||
// Wall-clock time jumps back on client while call is in progress
|
||||
TEST_F(TimeChangeTest, TimeJumpBackAfterStreamCreated) { |
||||
EchoRequest request; |
||||
EchoResponse response; |
||||
ClientContext context; |
||||
context.set_deadline(grpc_timeout_milliseconds_to_deadline(5000)); |
||||
context.AddMetadata(kServerResponseStreamsToSend, "2"); |
||||
|
||||
auto channel = GetChannel(); |
||||
GPR_ASSERT(channel); |
||||
EXPECT_TRUE( |
||||
channel->WaitForConnected(grpc_timeout_milliseconds_to_deadline(5000))); |
||||
auto stub = CreateStub(); |
||||
|
||||
auto stream = stub->BidiStream(&context); |
||||
|
||||
request.set_message("Hello"); |
||||
EXPECT_TRUE(stream->Write(request)); |
||||
EXPECT_TRUE(stream->Read(&response)); |
||||
|
||||
// time jumps back TIME_OFFSET1 milliseconds.
|
||||
set_now_offset(-TIME_OFFSET1); |
||||
|
||||
request.set_message("World"); |
||||
EXPECT_TRUE(stream->Write(request)); |
||||
EXPECT_TRUE(stream->WritesDone()); |
||||
EXPECT_TRUE(stream->Read(&response)); |
||||
|
||||
auto status = stream->Finish(); |
||||
EXPECT_TRUE(status.ok()); |
||||
} |
||||
|
||||
// Wall-clock time jumps forward on client before connection to server is up
|
||||
TEST_F(TimeChangeTest, TimeJumpForwardBeforeServerConnect) { |
||||
EchoRequest request; |
||||
EchoResponse response; |
||||
ClientContext context; |
||||
context.set_deadline(grpc_timeout_milliseconds_to_deadline(5000)); |
||||
context.AddMetadata(kServerResponseStreamsToSend, "2"); |
||||
|
||||
auto channel = GetChannel(); |
||||
GPR_ASSERT(channel); |
||||
|
||||
// time jumps forward by TIME_OFFSET2 milliseconds
|
||||
set_now_offset(TIME_OFFSET2); |
||||
|
||||
auto ret = |
||||
channel->WaitForConnected(grpc_timeout_milliseconds_to_deadline(5000)); |
||||
// We use monotonic clock for pthread_cond_timedwait() deadline on linux, and
|
||||
// realtime clock on other platforms - see gpr_cv_wait() in sync_posix.cc.
|
||||
// So changes in system clock affect deadlines on non-linux platforms
|
||||
#ifdef GPR_LINUX |
||||
EXPECT_TRUE(ret); |
||||
auto stub = CreateStub(); |
||||
auto stream = stub->BidiStream(&context); |
||||
|
||||
request.set_message("Hello"); |
||||
EXPECT_TRUE(stream->Write(request)); |
||||
EXPECT_TRUE(stream->Read(&response)); |
||||
request.set_message("World"); |
||||
EXPECT_TRUE(stream->Write(request)); |
||||
EXPECT_TRUE(stream->WritesDone()); |
||||
EXPECT_TRUE(stream->Read(&response)); |
||||
|
||||
auto status = stream->Finish(); |
||||
EXPECT_TRUE(status.ok()); |
||||
#else |
||||
EXPECT_FALSE(ret); |
||||
#endif |
||||
} |
||||
|
||||
// Wall-clock time jumps back on client before connection to server is up
|
||||
TEST_F(TimeChangeTest, TimeJumpBackBeforeServerConnect) { |
||||
EchoRequest request; |
||||
EchoResponse response; |
||||
ClientContext context; |
||||
context.set_deadline(grpc_timeout_milliseconds_to_deadline(5000)); |
||||
context.AddMetadata(kServerResponseStreamsToSend, "2"); |
||||
|
||||
auto channel = GetChannel(); |
||||
GPR_ASSERT(channel); |
||||
|
||||
// time jumps back by TIME_OFFSET2 milliseconds
|
||||
set_now_offset(-TIME_OFFSET2); |
||||
|
||||
EXPECT_TRUE( |
||||
channel->WaitForConnected(grpc_timeout_milliseconds_to_deadline(5000))); |
||||
auto stub = CreateStub(); |
||||
auto stream = stub->BidiStream(&context); |
||||
|
||||
request.set_message("Hello"); |
||||
EXPECT_TRUE(stream->Write(request)); |
||||
EXPECT_TRUE(stream->Read(&response)); |
||||
request.set_message("World"); |
||||
EXPECT_TRUE(stream->Write(request)); |
||||
EXPECT_TRUE(stream->WritesDone()); |
||||
EXPECT_TRUE(stream->Read(&response)); |
||||
|
||||
auto status = stream->Finish(); |
||||
EXPECT_TRUE(status.ok()); |
||||
} |
||||
|
||||
// Wall-clock time jumps forward and backwards during call
|
||||
TEST_F(TimeChangeTest, TimeJumpForwardAndBackDuringCall) { |
||||
EchoRequest request; |
||||
EchoResponse response; |
||||
ClientContext context; |
||||
context.set_deadline(grpc_timeout_milliseconds_to_deadline(5000)); |
||||
context.AddMetadata(kServerResponseStreamsToSend, "2"); |
||||
|
||||
auto channel = GetChannel(); |
||||
GPR_ASSERT(channel); |
||||
|
||||
EXPECT_TRUE( |
||||
channel->WaitForConnected(grpc_timeout_milliseconds_to_deadline(5000))); |
||||
auto stub = CreateStub(); |
||||
auto stream = stub->BidiStream(&context); |
||||
|
||||
request.set_message("Hello"); |
||||
EXPECT_TRUE(stream->Write(request)); |
||||
|
||||
// time jumps back by TIME_OFFSET2 milliseconds
|
||||
set_now_offset(-TIME_OFFSET2); |
||||
|
||||
EXPECT_TRUE(stream->Read(&response)); |
||||
request.set_message("World"); |
||||
|
||||
// time jumps forward by TIME_OFFSET milliseconds
|
||||
set_now_offset(TIME_OFFSET1); |
||||
|
||||
EXPECT_TRUE(stream->Write(request)); |
||||
|
||||
// time jumps back by TIME_OFFSET2 milliseconds
|
||||
set_now_offset(-TIME_OFFSET2); |
||||
|
||||
EXPECT_TRUE(stream->WritesDone()); |
||||
|
||||
// time jumps back by TIME_OFFSET2 milliseconds
|
||||
set_now_offset(-TIME_OFFSET2); |
||||
|
||||
EXPECT_TRUE(stream->Read(&response)); |
||||
|
||||
// time jumps back by TIME_OFFSET2 milliseconds
|
||||
set_now_offset(-TIME_OFFSET2); |
||||
|
||||
auto status = stream->Finish(); |
||||
EXPECT_TRUE(status.ok()); |
||||
} |
||||
|
||||
} // namespace
|
||||
|
||||
} // namespace testing
|
||||
} // namespace grpc
|
||||
|
||||
int main(int argc, char** argv) { |
||||
std::string me = argv[0]; |
||||
// get index of last slash in path to test binary
|
||||
auto lslash = me.rfind('/'); |
||||
// set g_root = path to directory containing test binary
|
||||
if (lslash != std::string::npos) { |
||||
g_root = me.substr(0, lslash); |
||||
} else { |
||||
g_root = "."; |
||||
} |
||||
|
||||
gpr_mu_init(&g_mu); |
||||
gpr_now_impl = now_impl; |
||||
|
||||
grpc::testing::TestEnvironment env(argc, argv); |
||||
::testing::InitGoogleTest(&argc, argv); |
||||
auto ret = RUN_ALL_TESTS(); |
||||
return ret; |
||||
} |
||||
Loading…
Reference in new issue