Merge pull request #20105 from rmstar/timejumptest

Added test for time jumps

test/cpp/common/BUILD

@@ -42,6 +42,19 @@ grpc_cc_test(
     ],
 )
 
+grpc_cc_test(
+    name = "time_jump_test",
+    srcs = ["time_jump_test.cc"],
+    external_deps = [
+        "gtest",
+    ],
+    tags = ["manual"],
+    deps = [
+        "//:grpc++",
+        "//test/core/util:grpc_test_util",
+    ],
+)
+
 grpc_cc_test(
     name = "auth_property_iterator_test",
     srcs = ["auth_property_iterator_test.cc"],

test/cpp/common/time_jump_test.cc

@@ -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();
+}

tools/internal_ci/macos/grpc_run_bazel_tests.sh

@@ -18,18 +18,13 @@ set -ex
 # change to grpc repo root
 cd $(dirname $0)/../../..
 
-# Download bazel
-temp_dir="$(mktemp -d)"
-wget -q https://github.com/bazelbuild/bazel/releases/download/0.26.0/bazel-0.26.0-darwin-x86_64 -O "${temp_dir}/bazel"
-chmod 755 "${temp_dir}/bazel"
-export PATH="${temp_dir}:${PATH}"
-# This should show ${temp_dir}/bazel
-which bazel
-
 ./tools/run_tests/start_port_server.py
 
 # run cfstream_test separately because it messes with the network
-bazel test $RUN_TESTS_FLAGS --spawn_strategy=standalone --genrule_strategy=standalone --test_output=all //test/cpp/end2end:cfstream_test
+tools/bazel test $RUN_TESTS_FLAGS --spawn_strategy=standalone --genrule_strategy=standalone --test_output=all //test/cpp/end2end:cfstream_test
+
+# run time_jump_test separately because it changes system time
+tools/bazel test $RUN_TESTS_FLAGS --spawn_strategy=standalone --genrule_strategy=standalone --test_output=all //test/cpp/common:time_jump_test
 
 # kill port_server.py to prevent the build from hanging
 ps aux | grep port_server\\.py | awk '{print $2}' | xargs kill -9