update common test

pedroSG94 · Sep 30, 2024 · 897f069 · 897f069
1 parent b99341c
commit 897f069
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/common/src/test/java/com/pedro/common/ExtensionTest.kt b/common/src/test/java/com/pedro/common/ExtensionTest.kt
@@ -16,7 +16,7 @@
 
 package com.pedro.common
 
-import android.media.MediaCodec
+import com.pedro.common.frame.MediaFrame
 import org.junit.Assert.assertEquals
 import org.junit.Test
 import java.nio.ByteBuffer
@@ -29,14 +29,9 @@ class ExtensionTest {
   @Test
   fun `remove info`() {
     val buffer = ByteBuffer.wrap(ByteArray(256) { 0x00 }.mapIndexed { index, byte -> index.toByte()  }.toByteArray())
-    val info = MediaCodec.BufferInfo()
     val offset = 4
     val minusLimit = 2
-    info.presentationTimeUs = 0
-    info.offset = offset
-    info.size = buffer.remaining() - minusLimit
-    info.flags = 0
-
+    val info = MediaFrame.Info(4, buffer.remaining() - minusLimit, 0, 0)
     val result = buffer.removeInfo(info)
     assertEquals(buffer.capacity() - offset - minusLimit, result.remaining())
     assertEquals(offset.toByte(), result.get(0))