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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
// Copyright 2008 Google Inc.
//
// Author: Ronghua Wu (ronghuawu@google.com)
#include <string>
#include "talk/media/base/testutils.h"
#include "talk/media/webrtc/webrtcpassthroughrender.h"
#include "webrtc/base/gunit.h"
class WebRtcPassthroughRenderTest : public testing::Test {
public:
class ExternalRenderer : public webrtc::VideoRenderCallback {
public:
ExternalRenderer() : frame_num_(0) {
}
virtual ~ExternalRenderer() {
}
virtual int32_t RenderFrame(const uint32_t stream_id,
webrtc::I420VideoFrame& videoFrame) {
++frame_num_;
LOG(INFO) << "RenderFrame stream_id: " << stream_id
<< " frame_num: " << frame_num_;
return 0;
}
int frame_num() const {
return frame_num_;
}
private:
int frame_num_;
};
WebRtcPassthroughRenderTest()
: renderer_(new cricket::WebRtcPassthroughRender()) {
}
~WebRtcPassthroughRenderTest() {
}
webrtc::VideoRenderCallback* AddIncomingRenderStream(int stream_id) {
return renderer_->AddIncomingRenderStream(stream_id, 0, 0, 0, 0, 0);
}
bool HasIncomingRenderStream(int stream_id) {
return renderer_->HasIncomingRenderStream(stream_id);
}
bool DeleteIncomingRenderStream(int stream_id) {
return (renderer_->DeleteIncomingRenderStream(stream_id) == 0);
}
bool AddExternalRenderCallback(int stream_id,
webrtc::VideoRenderCallback* renderer) {
return (renderer_->AddExternalRenderCallback(stream_id, renderer) == 0);
}
bool StartRender(int stream_id) {
return (renderer_->StartRender(stream_id) == 0);
}
bool StopRender(int stream_id) {
return (renderer_->StopRender(stream_id) == 0);
}
private:
rtc::scoped_ptr<cricket::WebRtcPassthroughRender> renderer_;
};
TEST_F(WebRtcPassthroughRenderTest, Streams) {
const int stream_id1 = 1234;
const int stream_id2 = 5678;
const int stream_id3 = 9012; // A stream that doesn't exist.
webrtc::VideoRenderCallback* stream = NULL;
// Add a new stream
stream = AddIncomingRenderStream(stream_id1);
EXPECT_TRUE(stream != NULL);
EXPECT_TRUE(HasIncomingRenderStream(stream_id1));
// Tried to add a already existed stream should return null
stream =AddIncomingRenderStream(stream_id1);
EXPECT_TRUE(stream == NULL);
stream = AddIncomingRenderStream(stream_id2);
EXPECT_TRUE(stream != NULL);
EXPECT_TRUE(HasIncomingRenderStream(stream_id2));
// Remove the stream
EXPECT_FALSE(DeleteIncomingRenderStream(stream_id3));
EXPECT_TRUE(DeleteIncomingRenderStream(stream_id2));
EXPECT_TRUE(!HasIncomingRenderStream(stream_id2));
// Add back the removed stream
stream = AddIncomingRenderStream(stream_id2);
EXPECT_TRUE(stream != NULL);
EXPECT_TRUE(HasIncomingRenderStream(stream_id2));
}
TEST_F(WebRtcPassthroughRenderTest, Renderer) {
webrtc::I420VideoFrame frame;
const int stream_id1 = 1234;
const int stream_id2 = 5678;
const int stream_id3 = 9012; // A stream that doesn't exist.
webrtc::VideoRenderCallback* stream1 = NULL;
webrtc::VideoRenderCallback* stream2 = NULL;
// Add two new stream
stream1 = AddIncomingRenderStream(stream_id1);
EXPECT_TRUE(stream1 != NULL);
EXPECT_TRUE(HasIncomingRenderStream(stream_id1));
stream2 = AddIncomingRenderStream(stream_id2);
EXPECT_TRUE(stream2 != NULL);
EXPECT_TRUE(HasIncomingRenderStream(stream_id2));
// Register the external renderer
WebRtcPassthroughRenderTest::ExternalRenderer renderer1;
WebRtcPassthroughRenderTest::ExternalRenderer renderer2;
EXPECT_FALSE(AddExternalRenderCallback(stream_id3, &renderer1));
EXPECT_TRUE(AddExternalRenderCallback(stream_id1, &renderer1));
EXPECT_TRUE(AddExternalRenderCallback(stream_id2, &renderer2));
int test_frame_num = 10;
// RenderFrame without starting the render
for (int i = 0; i < test_frame_num; ++i) {
stream1->RenderFrame(stream_id1, frame);
}
EXPECT_EQ(0, renderer1.frame_num());
// Start the render and test again.
EXPECT_FALSE(StartRender(stream_id3));
EXPECT_TRUE(StartRender(stream_id1));
for (int i = 0; i < test_frame_num; ++i) {
stream1->RenderFrame(stream_id1, frame);
}
EXPECT_EQ(test_frame_num, renderer1.frame_num());
// Stop the render and test again.
EXPECT_FALSE(StopRender(stream_id3));
EXPECT_TRUE(StopRender(stream_id1));
for (int i = 0; i < test_frame_num; ++i) {
stream1->RenderFrame(stream_id1, frame);
}
// The frame number should not have changed.
EXPECT_EQ(test_frame_num, renderer1.frame_num());
// Test on stream2 with a differnt number.
EXPECT_TRUE(StartRender(stream_id2));
test_frame_num = 30;
for (int i = 0; i < test_frame_num; ++i) {
stream2->RenderFrame(stream_id2, frame);
}
EXPECT_EQ(test_frame_num, renderer2.frame_num());
}
|
