diff -r 000000000000 -r 6474c204b198 media/mtransport/test/nrappkit_unittest.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/media/mtransport/test/nrappkit_unittest.cpp Wed Dec 31 06:09:35 2014 +0100 @@ -0,0 +1,122 @@ + +/* -*- 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/. */ + +// Original author: ekr@rtfm.com +#include + +#include "nsThreadUtils.h" +#include "nsXPCOM.h" + +// nrappkit includes +extern "C" { +#include "nr_api.h" +#include "async_timer.h" +} + +#include "mtransport_test_utils.h" +#include "runnable_utils.h" + +#define GTEST_HAS_RTTI 0 +#include "gtest/gtest.h" +#include "gtest_utils.h" + +using namespace mozilla; + +MtransportTestUtils *test_utils; + +namespace { + +class TimerTest : public ::testing::Test { + public: + TimerTest() : handle_(nullptr), fired_(false) {} + + int ArmTimer(int timeout) { + int ret; + + test_utils->sts_target()->Dispatch( + WrapRunnableRet(this, &TimerTest::ArmTimer_w, timeout, &ret), + NS_DISPATCH_SYNC); + + return ret; + } + + int ArmTimer_w(int timeout) { + return NR_ASYNC_TIMER_SET(timeout, cb, this, &handle_); + } + + int CancelTimer() { + int ret; + + test_utils->sts_target()->Dispatch( + WrapRunnableRet(this, &TimerTest::CancelTimer_w, &ret), + NS_DISPATCH_SYNC); + + return ret; + } + + int CancelTimer_w() { + return NR_async_timer_cancel(handle_); + } + + int Schedule() { + int ret; + + test_utils->sts_target()->Dispatch( + WrapRunnableRet(this, &TimerTest::Schedule_w, &ret), + NS_DISPATCH_SYNC); + + return ret; + } + + int Schedule_w() { + NR_ASYNC_SCHEDULE(cb, this); + + return 0; + } + + + static void cb(NR_SOCKET r, int how, void *arg) { + std::cerr << "Timer fired " << std::endl; + + TimerTest *t = static_cast(arg); + + t->fired_ = true; + } + + protected: + void *handle_; + bool fired_; +}; +} + +TEST_F(TimerTest, SimpleTimer) { + ArmTimer(100); + ASSERT_TRUE_WAIT(fired_, 1000); +} + +TEST_F(TimerTest, CancelTimer) { + ArmTimer(1000); + CancelTimer(); + PR_Sleep(2000); + ASSERT_FALSE(fired_); +} + +TEST_F(TimerTest, ScheduleTest) { + Schedule(); + ASSERT_TRUE_WAIT(fired_, 1000); +} + +int main(int argc, char **argv) { + test_utils = new MtransportTestUtils(); + + // Start the tests + ::testing::InitGoogleTest(&argc, argv); + + int rv = RUN_ALL_TESTS(); + delete test_utils; + return rv; +}