fix unit tests

DataDog · Oct 28, 2024 · 7aea39c · 7aea39c
1 parent a803ecb
commit 7aea39c
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/...lin/com/datadog/android/sessionreplay/internal/recorder/resources/ResourceResolverTest.kt b/...lin/com/datadog/android/sessionreplay/internal/recorder/resources/ResourceResolverTest.kt
@@ -11,6 +11,7 @@ import android.content.res.Resources
 import android.graphics.Bitmap
 import android.graphics.drawable.BitmapDrawable
 import android.graphics.drawable.Drawable
+import android.graphics.drawable.Drawable.ConstantState
 import android.graphics.drawable.LayerDrawable
 import android.graphics.drawable.StateListDrawable
 import android.util.DisplayMetrics
@@ -103,6 +104,12 @@ internal class ResourceResolverTest {
     @Mock
     lateinit var mockResources: Resources
 
+    @Mock
+    lateinit var mockConstantState: ConstantState
+
+    @Mock
+    lateinit var mockBitmapConstantState: ConstantState
+
     private var fakeBitmapWidth: Int = 1
 
     private var fakeBitmapHeight: Int = 1
@@ -117,6 +124,11 @@ internal class ResourceResolverTest {
 
     @BeforeEach
     fun setup(forge: Forge) {
+        whenever(mockBitmapDrawable.constantState).thenReturn(mockBitmapConstantState)
+        whenever(mockBitmapConstantState.newDrawable()).thenReturn(mockBitmapDrawable)
+
+        whenever(mockDrawable.constantState).thenReturn(mockConstantState)
+        whenever(mockConstantState.newDrawable()).thenReturn(mockDrawable)
         fakeImageCompressionByteArray = forge.aString().toByteArray()
 
         fakeBitmapWidth = forge.anInt(min = 1)
@@ -619,6 +631,8 @@ internal class ResourceResolverTest {
     fun `M cache bitmap W resolveResourceId() { not a BitmapDrawable }`() {
         // Given
         val mockLayerDrawable = mock<LayerDrawable>()
+        whenever(mockLayerDrawable.constantState).thenReturn(mockConstantState)
+        whenever(mockConstantState.newDrawable()).thenReturn(mockLayerDrawable)
 
         // When
         testedResourceResolver.resolveResourceId(