1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
/* -*- 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 <iostream>
// nrappkit includes
extern "C" {
#include "nr_api.h"
#include "async_timer.h"
}
#include "runnable_utils.h"
#define GTEST_HAS_RTTI 0
#include "gtest/gtest.h"
#include "gtest_utils.h"
using namespace mozilla;
namespace {
class TimerTest : public MtransportTest {
public:
TimerTest() : MtransportTest(), handle_(nullptr), fired_(false) {}
virtual ~TimerTest() = default;
int ArmTimer(int timeout) {
int ret;
test_utils_->SyncDispatchToSTS(
WrapRunnableRet(&ret, this, &TimerTest::ArmTimer_w, timeout));
return ret;
}
int ArmCancelTimer(int timeout) {
int ret;
test_utils_->SyncDispatchToSTS(
WrapRunnableRet(&ret, this, &TimerTest::ArmCancelTimer_w, timeout));
return ret;
}
int ArmTimer_w(int timeout) {
return NR_ASYNC_TIMER_SET(timeout, cb, this, &handle_);
}
int ArmCancelTimer_w(int timeout) {
int r;
r = ArmTimer_w(timeout);
if (r) return r;
return CancelTimer_w();
}
int CancelTimer() {
int ret;
test_utils_->SyncDispatchToSTS(
WrapRunnableRet(&ret, this, &TimerTest::CancelTimer_w));
return ret;
}
int CancelTimer_w() { return NR_async_timer_cancel(handle_); }
int Schedule() {
int ret;
test_utils_->SyncDispatchToSTS(
WrapRunnableRet(&ret, this, &TimerTest::Schedule_w));
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<TimerTest*>(arg);
t->fired_ = true;
}
protected:
void* handle_;
bool fired_;
};
} // namespace
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, CancelTimer0) {
ArmCancelTimer(0);
PR_Sleep(100);
ASSERT_FALSE(fired_);
}
TEST_F(TimerTest, ScheduleTest) {
Schedule();
ASSERT_TRUE_WAIT(fired_, 1000);
}
|
