More remote participant tests

@@ -2,23 +2,35 @@ package io.livekit.android.room.participant
 
 
 import io.livekit.android.room.RTCClient
 import io.livekit.android.room.RTCClient
 import livekit.LivekitModels
 import livekit.LivekitModels
-import org.junit.Assert.assertEquals
-import org.junit.Assert.assertNotNull
+import org.junit.Assert.*
 import org.junit.Before
 import org.junit.Before
 import org.junit.Test
 import org.junit.Test
 import org.mockito.Mockito
 import org.mockito.Mockito
 
 
 class RemoteParticipantTest {
 class RemoteParticipantTest {
 
 
+    lateinit var rtcClient: RTCClient
     lateinit var participant: RemoteParticipant
     lateinit var participant: RemoteParticipant
 
 
     @Before
     @Before
     fun setup() {
     fun setup() {
-        val rtcClient = Mockito.mock(RTCClient::class.java)
+        rtcClient = Mockito.mock(RTCClient::class.java)
         participant = RemoteParticipant(rtcClient, "sid")
         participant = RemoteParticipant(rtcClient, "sid")
     }
     }
 
 
     @Test
     @Test
+    fun constructorAddsTrack() {
+        val info = LivekitModels.ParticipantInfo.newBuilder(INFO)
+            .addTracks(TRACK_INFO)
+            .build()
+
+        participant = RemoteParticipant(rtcClient, info)
+
+        assertEquals(1, participant.tracks.values.size)
+        assertNotNull(participant.getTrackPublication(TRACK_INFO.sid))
+    }
+
+    @Test
     fun updateFromInfoAddsTrack() {
     fun updateFromInfoAddsTrack() {
         val newTrackInfo = LivekitModels.ParticipantInfo.newBuilder(INFO)
         val newTrackInfo = LivekitModels.ParticipantInfo.newBuilder(INFO)
             .addTracks(TRACK_INFO)
             .addTracks(TRACK_INFO)
@@ -30,6 +42,33 @@ class RemoteParticipantTest {
         assertNotNull(participant.getTrackPublication(TRACK_INFO.sid))
         assertNotNull(participant.getTrackPublication(TRACK_INFO.sid))
     }
     }
 
 
+    @Test
+    fun updateFromInfoRemovesTrack() {
+        val newTrackInfo = LivekitModels.ParticipantInfo.newBuilder(INFO)
+            .addTracks(TRACK_INFO)
+            .build()
+
+        participant.updateFromInfo(newTrackInfo)
+        participant.updateFromInfo(INFO)
+
+        assertEquals(0, participant.tracks.values.size)
+        assertNull(participant.getTrackPublication(TRACK_INFO.sid))
+    }
+
+
+    @Test
+    fun unpublishTrackRemoves() {
+        val newTrackInfo = LivekitModels.ParticipantInfo.newBuilder(INFO)
+            .addTracks(TRACK_INFO)
+            .build()
+
+        participant.updateFromInfo(newTrackInfo)
+        participant.unpublishTrack(TRACK_INFO.sid)
+
+        assertEquals(0, participant.tracks.values.size)
+        assertNull(participant.getTrackPublication(TRACK_INFO.sid))
+    }
+
 
 
     companion object {
     companion object {
         val INFO = LivekitModels.ParticipantInfo.newBuilder()
         val INFO = LivekitModels.ParticipantInfo.newBuilder()