Fix checkable mapper privacy issue

DataDog · Oct 24, 2024 · 1224430 · 1224430
1 parent caa2a5d
commit 1224430
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 6 deletions.
diff --git a/...lin/com/datadog/android/sessionreplay/internal/recorder/mapper/CheckableTextViewMapper.kt b/...lin/com/datadog/android/sessionreplay/internal/recorder/mapper/CheckableTextViewMapper.kt
@@ -111,7 +111,7 @@ internal abstract class CheckableTextViewMapper<T>(
             )
             mappingContext.imageWireframeHelper.createImageWireframe(
                 view = view,
-                imagePrivacy = mappingContext.imagePrivacy,
+                imagePrivacy = mapInputPrivacyToImagePrivacy(mappingContext.textAndInputPrivacy),
                 currentWireframeIndex = 0,
                 x = checkBoxBounds.x,
                 y = checkBoxBounds.y,

diff --git a/...in/com/datadog/android/sessionreplay/internal/recorder/mapper/CheckableWireframeMapper.kt b/...in/com/datadog/android/sessionreplay/internal/recorder/mapper/CheckableWireframeMapper.kt
@@ -10,6 +10,7 @@ import android.view.View
 import android.widget.Checkable
 import androidx.annotation.UiThread
 import com.datadog.android.api.InternalLogger
+import com.datadog.android.sessionreplay.ImagePrivacy
 import com.datadog.android.sessionreplay.TextAndInputPrivacy
 import com.datadog.android.sessionreplay.model.MobileSegment
 import com.datadog.android.sessionreplay.recorder.MappingContext
@@ -52,6 +53,14 @@ internal abstract class CheckableWireframeMapper<T>(
         return mainWireframes
     }
 
+    protected fun mapInputPrivacyToImagePrivacy(inputPrivacy: TextAndInputPrivacy): ImagePrivacy {
+        return when (inputPrivacy) {
+            TextAndInputPrivacy.MASK_SENSITIVE_INPUTS -> ImagePrivacy.MASK_NONE
+            TextAndInputPrivacy.MASK_ALL_INPUTS,
+            TextAndInputPrivacy.MASK_ALL -> ImagePrivacy.MASK_ALL
+        }
+    }
+
     @UiThread
     abstract fun resolveMainWireframes(
         view: T,

diff --git a/...n/kotlin/com/datadog/android/sessionreplay/internal/recorder/mapper/SwitchCompatMapper.kt b/...n/kotlin/com/datadog/android/sessionreplay/internal/recorder/mapper/SwitchCompatMapper.kt
@@ -80,7 +80,7 @@ internal open class SwitchCompatMapper(
             }?.let { drawable ->
                 mappingContext.imageWireframeHelper.createImageWireframe(
                     view = view,
-                    imagePrivacy = mappingContext.imagePrivacy,
+                    imagePrivacy = mapInputPrivacyToImagePrivacy(mappingContext.textAndInputPrivacy),
                     currentWireframeIndex = prevIndex + 1,
                     x = trackBounds.x.densityNormalized(mappingContext.systemInformation.screenDensity).toLong(),
                     y = trackBounds.y.densityNormalized(mappingContext.systemInformation.screenDensity).toLong(),
@@ -111,7 +111,7 @@ internal open class SwitchCompatMapper(
             thumbBounds?.let { thumbBounds ->
                 mappingContext.imageWireframeHelper.createImageWireframe(
                     view = view,
-                    imagePrivacy = mappingContext.imagePrivacy,
+                    imagePrivacy = mapInputPrivacyToImagePrivacy(mappingContext.textAndInputPrivacy),
                     currentWireframeIndex = prevIndex + 1,
                     x = thumbBounds.x.densityNormalized(mappingContext.systemInformation.screenDensity).toLong(),
                     y = thumbBounds.y.densityNormalized(mappingContext.systemInformation.screenDensity).toLong(),

diff --git a/...datadog/android/sessionreplay/internal/recorder/mapper/BaseCheckableTextViewMapperTest.kt b/...datadog/android/sessionreplay/internal/recorder/mapper/BaseCheckableTextViewMapperTest.kt
@@ -218,7 +218,7 @@ internal abstract class BaseCheckableTextViewMapperTest<T> :
         // Then
         verify(fakeMappingContext.imageWireframeHelper).createImageWireframe(
             view = eq(mockCheckableTextView),
-            imagePrivacy = eq(ImagePrivacy.MASK_LARGE_ONLY),
+            imagePrivacy = eq(ImagePrivacy.MASK_NONE),
             currentWireframeIndex = anyInt(),
             x = eq(expectedX),
             y = eq(expectedY),
@@ -261,7 +261,7 @@ internal abstract class BaseCheckableTextViewMapperTest<T> :
         // Then
         verify(fakeMappingContext.imageWireframeHelper).createImageWireframe(
             view = eq(mockCheckableTextView),
-            imagePrivacy = eq(ImagePrivacy.MASK_LARGE_ONLY),
+            imagePrivacy = eq(ImagePrivacy.MASK_NONE),
             currentWireframeIndex = anyInt(),
             x = eq(expectedX),
             y = eq(expectedY),

diff --git a/...tlin/com/datadog/android/sessionreplay/internal/recorder/mapper/SwitchCompatMapperTest.kt b/...tlin/com/datadog/android/sessionreplay/internal/recorder/mapper/SwitchCompatMapperTest.kt
@@ -89,7 +89,7 @@ internal class SwitchCompatMapperTest : BaseSwitchCompatMapperTest() {
 
             verify(fakeMappingContext.imageWireframeHelper, times(2)).createImageWireframe(
                 view = eq(mockSwitch),
-                imagePrivacy = eq(ImagePrivacy.MASK_LARGE_ONLY),
+                imagePrivacy = eq(ImagePrivacy.MASK_NONE),
                 currentWireframeIndex = ArgumentMatchers.anyInt(),
                 x = xCaptor.capture(),
                 y = yCaptor.capture(),