1 files changed, 57 insertions, 0 deletions
diff --git a/security/nss/gtests/ssl_gtest/gtest_utils.h b/security/nss/gtests/ssl_gtest/gtest_utils.h
new file mode 100644
index 0000000000..2344c3cea9
--- /dev/null
+++ b/security/nss/gtests/ssl_gtest/gtest_utils.h
@@ -0,0 +1,57 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=2 et sw=2 tw=80: */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this file,
+ * You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#ifndef gtest_utils_h__
+#define gtest_utils_h__
+
+#define GTEST_HAS_RTTI 0
+#include "gtest/gtest.h"
+#include "test_io.h"
+
+namespace nss_test {
+
+// Gtest utilities
+class Timeout : public PollTarget {
+ public:
+  Timeout(int32_t timer_ms) : handle_(nullptr) {
+    Poller::Instance()->SetTimer(timer_ms, this, &Timeout::ExpiredCallback,
+                                 &handle_);
+  }
+  ~Timeout() {
+    if (handle_) {
+      handle_->Cancel();
+    }
+  }
+
+  static void ExpiredCallback(PollTarget* target, Event event) {
+    Timeout* timeout = static_cast<Timeout*>(target);
+    timeout->handle_ = nullptr;
+  }
+
+  bool timed_out() const { return !handle_; }
+
+ private:
+  std::shared_ptr<Poller::Timer> handle_;
+};
+
+}  // namespace nss_test
+
+#define WAIT_(expression, timeout) \
+  do {                             \
+    Timeout tm(timeout);           \
+    while (!(expression)) {        \
+      Poller::Instance()->Poll();  \
+      if (tm.timed_out()) break;   \
+    }                              \
+  } while (0)
+
+#define ASSERT_TRUE_WAIT(expression, timeout) \
+  do {                                        \
+    WAIT_(expression, timeout);               \
+    ASSERT_TRUE(expression);                  \
+  } while (0)
+
+#endif