Fix flaky test in TLVBlockFileReader

DataDog · Jul 24, 2024 · 89d5a79 · 89d5a79
1 parent 54f5f12
commit 89d5a79
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/.../kotlin/com/datadog/android/core/internal/persistence/tlvformat/TLVBlockFileReaderTest.kt b/.../kotlin/com/datadog/android/core/internal/persistence/tlvformat/TLVBlockFileReaderTest.kt
@@ -96,6 +96,11 @@ internal class TLVBlockFileReaderTest {
         testedReader.read(file = mockFile)
 
         // Then
+        val expectedMessage = if (fakeDataString.length >= 2) {
+            "TLV header corrupt. Invalid type"
+        } else {
+            "Failed to deserialize TLV data length"
+        }
         val captor = argumentCaptor<() -> String>()
         verify(mockInternalLogger).log(
             level = eq(InternalLogger.Level.WARN),
@@ -106,7 +111,7 @@ internal class TLVBlockFileReaderTest {
             anyOrNull()
         )
         assertThat(captor.firstValue.invoke())
-            .startsWith("TLV header corrupt. Invalid type")
+            .startsWith(expectedMessage)
     }
 
     @Test