From 1b03de724d446d24b92a28b77433994be9d5962e Mon Sep 17 00:00:00 2001
From: Guillaume Bernos <guillaume@bernos.dev>
Date: Thu, 9 Jan 2025 15:02:13 +0100
Subject: [PATCH] test without crash

---
 .../test-app/src/main/AndroidManifest.xml     | 11 ++-
 .../testing/crashlytics/FirstFragment.kt      | 14 ++++
 .../crashlytics/PreFirebaseProvider.kt        | 78 +++++++++++++++++++
 3 files changed, 101 insertions(+), 2 deletions(-)
 create mode 100644 firebase-crashlytics/test-app/src/main/kotlin/com/google/firebase/testing/crashlytics/PreFirebaseProvider.kt

diff --git a/firebase-crashlytics/test-app/src/main/AndroidManifest.xml b/firebase-crashlytics/test-app/src/main/AndroidManifest.xml
index 82ff1703e99..ba3a22fd363 100644
--- a/firebase-crashlytics/test-app/src/main/AndroidManifest.xml
+++ b/firebase-crashlytics/test-app/src/main/AndroidManifest.xml
@@ -29,7 +29,6 @@
       android:exported="true"
       android:label="@string/app_name"
       android:name="com.google.firebase.testing.crashlytics.MainActivity"
-      android:process=":main"
       android:theme="@style/Theme.Widget_test_app.NoActionBar">
       <intent-filter>
         <action android:name="android.intent.action.MAIN" />
@@ -52,6 +51,15 @@
       android:name="firebase_crashlytics_crashlytics_restart_timeout"
       android:value="5" />
 
+    <provider
+        android:authorities="${applicationId}.preFirebase"
+        android:directBootAware="true"
+        android:exported="false"
+        android:initOrder="101"
+        android:name="com.google.firebase.testing.crashlytics.PreFirebaseProvider" />
+
+    <receiver android:name=".CrashBroadcastReceiver" />
+
     <receiver
       android:exported="false"
       android:name="com.google.firebase.testing.crashlytics.CrashWidgetProvider"
@@ -64,7 +72,6 @@
         android:resource="@xml/homescreen_widget" />
     </receiver>
 
-    <receiver android:name="com.google.firebase.testing.crashlytics.CrashBroadcastReceiver" />
 
     <service
       android:enabled="true"
diff --git a/firebase-crashlytics/test-app/src/main/kotlin/com/google/firebase/testing/crashlytics/FirstFragment.kt b/firebase-crashlytics/test-app/src/main/kotlin/com/google/firebase/testing/crashlytics/FirstFragment.kt
index 10011a0ca20..a9f46bdef03 100644
--- a/firebase-crashlytics/test-app/src/main/kotlin/com/google/firebase/testing/crashlytics/FirstFragment.kt
+++ b/firebase-crashlytics/test-app/src/main/kotlin/com/google/firebase/testing/crashlytics/FirstFragment.kt
@@ -63,6 +63,7 @@ class FirstFragment : Fragment() {
     }
 
     binding.buttonSharedGenerateCrash.setOnClickListener {
+      PreFirebaseProvider.expectedMessage = "Shared_Generate_Crash"
       val userId = "SharedGenerate_${System.currentTimeMillis()}"
       saveAndApplyUserId(userId)
 
@@ -70,6 +71,7 @@ class FirstFragment : Fragment() {
     }
 
     binding.buttonSharedVerifyCrash.setOnClickListener {
+      PreFirebaseProvider.expectedMessage = "Shared_Verify_Crash"
       val userId = "SharedVerifyCrash_${System.currentTimeMillis()}"
       saveAndApplyUserId(userId)
 
@@ -88,6 +90,7 @@ class FirstFragment : Fragment() {
     }
 
     binding.buttonFirebasecoreFatalError.setOnClickListener {
+      PreFirebaseProvider.expectedMessage = "FirebaseCore_Fatal_Error"
       val userId = "FirebaseCoreFatal_${System.currentTimeMillis()}"
       saveAndApplyUserId(userId)
 
@@ -95,6 +98,7 @@ class FirstFragment : Fragment() {
     }
 
     binding.buttonPublicApiLog.setOnClickListener {
+      PreFirebaseProvider.expectedMessage = "Public_API_Log"
       val userId = "PublicApiLog_${System.currentTimeMillis()}"
       saveAndApplyUserId(userId)
 
@@ -105,6 +109,7 @@ class FirstFragment : Fragment() {
     }
 
     binding.buttonPublicApiSetcustomvalue.setOnClickListener {
+      PreFirebaseProvider.expectedMessage = "Public_API_SetCustomValue"
       val userId = "PublicApiSetCustomValue_${System.currentTimeMillis()}"
       saveAndApplyUserId(userId)
 
@@ -114,6 +119,7 @@ class FirstFragment : Fragment() {
     }
 
     binding.buttonPublicApiSetuserid.setOnClickListener {
+      PreFirebaseProvider.expectedMessage = "Public_API_SetUserID"
       val userId = "PublicApiSetUserID_${System.currentTimeMillis()}"
       saveAndApplyUserId(userId)
 
@@ -121,6 +127,7 @@ class FirstFragment : Fragment() {
     }
 
     binding.buttonPublicApiDidcrashpreviously.setOnClickListener {
+      PreFirebaseProvider.expectedMessage = "Public_API_DidCrashPreviously"
       val userId = "DidCrashPreviously_${System.currentTimeMillis()}"
       saveAndApplyUserId(userId)
 
@@ -138,6 +145,7 @@ class FirstFragment : Fragment() {
     }
 
     binding.buttonDatacollectionDefault.setOnClickListener {
+      PreFirebaseProvider.expectedMessage = "DataCollection_Default"
       val userId = "DataCollectionDefault_${System.currentTimeMillis()}"
       saveAndApplyUserId(userId)
 
@@ -145,6 +153,7 @@ class FirstFragment : Fragment() {
     }
 
     binding.buttonDatacollectionFirebaseOff.setOnClickListener {
+      PreFirebaseProvider.expectedMessage = "DataCollection_Firebase_Off"
       val userId = "DataCollectionFirebaseOff_${System.currentTimeMillis()}"
       saveAndApplyUserId(userId)
 
@@ -152,6 +161,7 @@ class FirstFragment : Fragment() {
     }
 
     binding.buttonDatacollectionCrashlyticsOff.setOnClickListener {
+      PreFirebaseProvider.expectedMessage = "DataCollection_Crashlytics_Off"
       val userId = "DataCollectionCrashlyticsOff_${System.currentTimeMillis()}"
       saveAndApplyUserId(userId)
 
@@ -159,6 +169,7 @@ class FirstFragment : Fragment() {
     }
 
     binding.buttonDatacollectionOffThenOn.setOnClickListener {
+      PreFirebaseProvider.expectedMessage = "DataCollection_Crashlytics_Off_Then_On"
       val userId = "DataCollectionOffThenOn_${System.currentTimeMillis()}"
       saveAndApplyUserId(userId)
 
@@ -166,6 +177,7 @@ class FirstFragment : Fragment() {
     }
 
     binding.buttonDatacollectionOffThenSend.setOnClickListener {
+      PreFirebaseProvider.expectedMessage = "DataCollection_Crashlytics_Off_Then_Send"
       val userId = "DataCollectionOffThenSend_${System.currentTimeMillis()}"
       saveAndApplyUserId(userId)
 
@@ -173,6 +185,7 @@ class FirstFragment : Fragment() {
     }
 
     binding.buttonDatacollectionOffThenDelete.setOnClickListener {
+      PreFirebaseProvider.expectedMessage = "DataCollection_Crashlytics_Off_Then_Delete"
       val userId = "DataCollectionOffThenDelete_${System.currentTimeMillis()}"
       saveAndApplyUserId(userId)
 
@@ -180,6 +193,7 @@ class FirstFragment : Fragment() {
     }
 
     binding.buttonInteroperabilityIid.setOnClickListener {
+      PreFirebaseProvider.expectedMessage = "Interoperability_IID"
       val userId = "InteroperabilityIID_${System.currentTimeMillis()}"
       saveAndApplyUserId(userId)
 
diff --git a/firebase-crashlytics/test-app/src/main/kotlin/com/google/firebase/testing/crashlytics/PreFirebaseProvider.kt b/firebase-crashlytics/test-app/src/main/kotlin/com/google/firebase/testing/crashlytics/PreFirebaseProvider.kt
new file mode 100644
index 00000000000..b7c027e889b
--- /dev/null
+++ b/firebase-crashlytics/test-app/src/main/kotlin/com/google/firebase/testing/crashlytics/PreFirebaseProvider.kt
@@ -0,0 +1,78 @@
+/*
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.firebase.testing.crashlytics
+
+import android.content.ContentProvider
+import android.content.ContentValues
+import android.database.Cursor
+import android.net.Uri
+import java.lang.Thread.UncaughtExceptionHandler
+import kotlin.system.exitProcess
+
+class PreFirebaseProvider : ContentProvider(), UncaughtExceptionHandler {
+    private var defaultUncaughtExceptionHandler: UncaughtExceptionHandler? = null
+
+    companion object {
+        /**
+         * Set an expected exception message. If the uncaught exception contains the given string, then
+         * the app will exit cleanly. Otherwise, it will propagate up to the default exception handler.
+         */
+        var expectedMessage: String? = null
+    }
+
+    override fun onCreate(): Boolean {
+        defaultUncaughtExceptionHandler = Thread.getDefaultUncaughtExceptionHandler()
+        Thread.setDefaultUncaughtExceptionHandler(this)
+
+        return false
+    }
+
+    /* Returns if the given exception contains the expectedMessage in its message. */
+    private fun isExpected(throwable: Throwable): Boolean =
+        expectedMessage?.run { throwable.message?.contains(this) } ?: false
+
+    override fun uncaughtException(thread: Thread, throwable: Throwable) {
+        if (isExpected(throwable)) {
+            // Exit cleanly
+            exitProcess(0)
+        } else {
+            // Propagate up to the default exception handler
+            defaultUncaughtExceptionHandler?.uncaughtException(thread, throwable)
+        }
+    }
+
+    override fun query(
+        uri: Uri,
+        projection: Array<out String>?,
+        selection: String?,
+        selectionArgs: Array<out String>?,
+        sortOrder: String?
+    ): Cursor? = null
+
+    override fun getType(uri: Uri): String? = null
+
+    override fun insert(uri: Uri, values: ContentValues?): Uri? = null
+
+    override fun delete(uri: Uri, selection: String?, selectionArgs: Array<out String>?): Int = 0
+
+    override fun update(
+        uri: Uri,
+        values: ContentValues?,
+        selection: String?,
+        selectionArgs: Array<out String>?
+    ): Int = 0
+}
\ No newline at end of file