RTCEngineMockE2ETest.kt
5.2 KB
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
148
149
150
151
152
153
154
/*
* Copyright 2023-2024 LiveKit, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.livekit.android.room
import io.livekit.android.test.MockE2ETest
import io.livekit.android.test.mock.TestData
import io.livekit.android.test.util.toOkioByteString
import io.livekit.android.test.util.toPBByteString
import io.livekit.android.util.toOkioByteString
import kotlinx.coroutines.ExperimentalCoroutinesApi
import livekit.LivekitModels
import livekit.LivekitRtc
import livekit.org.webrtc.PeerConnection
import org.junit.Assert
import org.junit.Assert.assertEquals
import org.junit.Before
import org.junit.Test
import org.junit.runner.RunWith
import org.robolectric.RobolectricTestRunner
@ExperimentalCoroutinesApi
@RunWith(RobolectricTestRunner::class)
class RTCEngineMockE2ETest : MockE2ETest() {
lateinit var rtcEngine: RTCEngine
@Before
fun setupRTCEngine() {
rtcEngine = component.rtcEngine()
}
@Test
fun iceServersSetOnJoin() = runTest {
connect()
val sentIceServers = TestData.JOIN.join.iceServersList
.map { it.toWebrtc() }
val subPeerConnection = getSubscriberPeerConnection()
assertEquals(sentIceServers, subPeerConnection.rtcConfig.iceServers)
}
@Test
fun iceSubscriberConnect() = runTest {
connect()
assertEquals(
TestData.OFFER.offer.sdp,
getSubscriberPeerConnection().remoteDescription?.description,
)
val ws = wsFactory.ws
val sentRequest = LivekitRtc.SignalRequest.newBuilder()
.mergeFrom(ws.sentRequests[0].toPBByteString())
.build()
val subPeerConnection = getSubscriberPeerConnection()
val localAnswer = subPeerConnection.localDescription ?: throw IllegalStateException("no answer was created.")
Assert.assertTrue(sentRequest.hasAnswer())
assertEquals(localAnswer.description, sentRequest.answer.sdp)
assertEquals(localAnswer.type.canonicalForm(), sentRequest.answer.type)
assertEquals(ConnectionState.CONNECTED, rtcEngine.connectionState)
}
@Test
fun reconnectOnFailure() = runTest {
connect()
val oldWs = wsFactory.ws
wsFactory.listener.onFailure(oldWs, Exception(), null)
testScheduler.advanceTimeBy(1000)
val newWs = wsFactory.ws
Assert.assertNotEquals(oldWs, newWs)
}
@Test
fun reconnectOnSubscriberFailure() = runTest {
connect()
val oldWs = wsFactory.ws
val subPeerConnection = getSubscriberPeerConnection()
subPeerConnection.moveToIceConnectionState(PeerConnection.IceConnectionState.FAILED)
testScheduler.advanceTimeBy(1000)
val newWs = wsFactory.ws
Assert.assertNotEquals(oldWs, newWs)
}
@Test
fun reconnectOnPublisherFailure() = runTest {
connect()
val oldWs = wsFactory.ws
val pubPeerConnection = getPublisherPeerConnection()
pubPeerConnection.moveToIceConnectionState(PeerConnection.IceConnectionState.FAILED)
testScheduler.advanceTimeBy(1000)
val newWs = wsFactory.ws
Assert.assertNotEquals(oldWs, newWs)
}
@Test
fun refreshToken() = runTest {
connect()
val oldToken = wsFactory.request.url.queryParameter(SignalClient.CONNECT_QUERY_TOKEN)
wsFactory.listener.onMessage(wsFactory.ws, TestData.REFRESH_TOKEN.toOkioByteString())
wsFactory.listener.onFailure(wsFactory.ws, Exception(), null)
testScheduler.advanceUntilIdle()
val newToken = wsFactory.request.url.queryParameter(SignalClient.CONNECT_QUERY_TOKEN)
Assert.assertNotEquals(oldToken, newToken)
assertEquals(TestData.REFRESH_TOKEN.refreshToken, newToken)
}
@Test
fun relayConfiguration() = runTest {
connect(
with(TestData.JOIN.toBuilder()) {
join = with(join.toBuilder()) {
clientConfiguration = with(LivekitModels.ClientConfiguration.newBuilder()) {
forceRelay = LivekitModels.ClientConfigSetting.ENABLED
build()
}
build()
}
build()
},
)
val subPeerConnection = getSubscriberPeerConnection()
assertEquals(PeerConnection.IceTransportsType.RELAY, subPeerConnection.rtcConfig.iceTransportsType)
}
fun participantIdOnReconnect() = runTest {
connect()
wsFactory.listener.onFailure(wsFactory.ws, Exception(), null)
testScheduler.advanceUntilIdle()
val sid = wsFactory.request.url.queryParameter(SignalClient.CONNECT_QUERY_PARTICIPANT_SID)
assertEquals(TestData.JOIN.join.participant.sid, sid)
}
}