folly/test/TestUtils.h

   1 /*
   2  * Copyright 2017 Facebook, Inc.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #pragma once
  18
  19 #include <chrono>
  20
  21 #include <folly/portability/GTest.h>
  22
  23 // We use this to indicate that tests have failed because of timing
  24 // or dependencies that may be flakey. Internally this is used by
  25 // our test runner to retry the test. To gtest this will look like
  26 // a normal test failure; there is only an effect if the test framework
  27 // interprets the message.
  28 #define SKIP() GTEST_FATAL_FAILURE_("Test skipped by client")
  29
  30 namespace folly {
  31 namespace test {
  32
  33 template <typename T1, typename T2>
  34 ::testing::AssertionResult
  35 AreWithinSecs(T1 val1, T2 val2, std::chrono::seconds acceptableDeltaSecs) {
  36   auto deltaSecs =
  37       std::chrono::duration_cast<std::chrono::seconds>(val1 - val2);
  38   if (deltaSecs <= acceptableDeltaSecs &&
  39       deltaSecs >= -1 * acceptableDeltaSecs) {
  40     return ::testing::AssertionSuccess();
  41   } else {
  42     return ::testing::AssertionFailure()
  43         << val1.count() << " and " << val2.count() << " are not within "
  44         << acceptableDeltaSecs.count() << " secs of each other";
  45   }
  46 }
  47 }
  48 }