Util.h 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. /*
  2. * Copyright 2014-present 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. #pragma once
  17. #include <folly/io/async/test/TimeUtil.h>
  18. #include <folly/portability/GTest.h>
  19. #include <folly/test/TestUtils.h>
  20. /**
  21. * Check how long a timeout took to fire.
  22. *
  23. * This method verifies:
  24. * - that the timeout did not fire too early (never less than expectedMS)
  25. * - that the timeout fired within a reasonable period of the expected
  26. * duration. It must fire within the specified tolerance, excluding time
  27. * that this process spent waiting to be scheduled.
  28. *
  29. * @param start A TimePoint object set just before the timeout
  30. * was scheduled.
  31. * @param end A TimePoint object set when the timeout fired.
  32. * @param expectedMS The timeout duration, in milliseconds
  33. * @param tolerance The tolerance, in milliseconds.
  34. */
  35. #define T_CHECK_TIMEOUT(start, end, expectedMS, ...) \
  36. if (!::folly::checkTimeout( \
  37. (start), (end), (expectedMS), false, ##__VA_ARGS__)) { \
  38. SKIP() << "T_CHECK_TIMEOUT lapsed"; \
  39. }
  40. /**
  41. * Verify that an event took less than a specified amount of time.
  42. *
  43. * This is similar to T_CHECK_TIMEOUT, but does not fail if the event took less
  44. * than the allowed time.
  45. */
  46. #define T_CHECK_TIME_LT(start, end, expectedMS, ...) \
  47. if (!::folly::checkTimeout( \
  48. (start), (end), (expectedMS), true, ##__VA_ARGS__)) { \
  49. SKIP() << "T_CHECK_TIMEOUT_LT lapsed"; \
  50. }