1 /* 2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. 3 * 4 * Use of this source code is governed by a BSD-style license 5 * that can be found in the LICENSE file in the root of the source 6 * tree. An additional intellectual property rights grant can be found 7 * in the file PATENTS. All contributing project authors may 8 * be found in the AUTHORS file in the root of the source tree. 9 */ 10 11 #ifndef WEBRTC_BASE_GUNIT_H_ 12 #define WEBRTC_BASE_GUNIT_H_ 13 14 #include "webrtc/base/logging.h" 15 #include "webrtc/base/thread.h" 16 #if defined(GTEST_RELATIVE_PATH) 17 #include "testing/gtest/include/gtest/gtest.h" 18 #else 19 #include "testing/base/public/gunit.h" 20 #endif 21 22 // Wait until "ex" is true, or "timeout" expires. 23 #define WAIT(ex, timeout) \ 24 for (uint32_t start = rtc::Time(); !(ex) && rtc::Time() < start + timeout;) \ 25 rtc::Thread::Current()->ProcessMessages(1); 26 27 // This returns the result of the test in res, so that we don't re-evaluate 28 // the expression in the XXXX_WAIT macros below, since that causes problems 29 // when the expression is only true the first time you check it. 30 #define WAIT_(ex, timeout, res) \ 31 do { \ 32 uint32_t start = rtc::Time(); \ 33 res = (ex); \ 34 while (!res && rtc::Time() < start + timeout) { \ 35 rtc::Thread::Current()->ProcessMessages(1); \ 36 res = (ex); \ 37 } \ 38 } while (0) 39 40 // The typical EXPECT_XXXX and ASSERT_XXXXs, but done until true or a timeout. 41 #define EXPECT_TRUE_WAIT(ex, timeout) \ 42 do { \ 43 bool res; \ 44 WAIT_(ex, timeout, res); \ 45 if (!res) EXPECT_TRUE(ex); \ 46 } while (0) 47 48 #define EXPECT_EQ_WAIT(v1, v2, timeout) \ 49 do { \ 50 bool res; \ 51 WAIT_(v1 == v2, timeout, res); \ 52 if (!res) EXPECT_EQ(v1, v2); \ 53 } while (0) 54 55 #define ASSERT_TRUE_WAIT(ex, timeout) \ 56 do { \ 57 bool res; \ 58 WAIT_(ex, timeout, res); \ 59 if (!res) ASSERT_TRUE(ex); \ 60 } while (0) 61 62 #define ASSERT_EQ_WAIT(v1, v2, timeout) \ 63 do { \ 64 bool res; \ 65 WAIT_(v1 == v2, timeout, res); \ 66 if (!res) ASSERT_EQ(v1, v2); \ 67 } while (0) 68 69 // Version with a "soft" timeout and a margin. This logs if the timeout is 70 // exceeded, but it only fails if the expression still isn't true after the 71 // margin time passes. 72 #define EXPECT_TRUE_WAIT_MARGIN(ex, timeout, margin) \ 73 do { \ 74 bool res; \ 75 WAIT_(ex, timeout, res); \ 76 if (res) { \ 77 break; \ 78 } \ 79 LOG(LS_WARNING) << "Expression " << #ex << " still not true after " << \ 80 timeout << "ms; waiting an additional " << margin << "ms"; \ 81 WAIT_(ex, margin, res); \ 82 if (!res) { \ 83 EXPECT_TRUE(ex); \ 84 } \ 85 } while (0) 86 87 #endif // WEBRTC_BASE_GUNIT_H_ 88