Wrap InstallReferrerCallback to use a latch

Author: jamesnrokt

android-core/src/androidTest/kotlin/com/mparticle/InstallReferrerHelperTest.kt

@@ -63,21 +63,24 @@ class InstallReferrerHelperTest : BaseCleanInstallEachTest() {
     fun testFetchInstallReferrerFails() {
         assertNull(InstallReferrerHelper.getInstallReferrer(mContext))
         val latch = CountDownLatch(1)
-        val callback = object : InstallReferrerHelper.InstallReferrerCallback {
-            override fun onReceived(installReferrer: String?) {
-                latch.countDown()
-            }
-
-            override fun onFailed() {
-                latch.countDown()
-            }
-        }
-        val spiedCallback = Mockito.spy(callback)
-        InstallReferrerHelper.fetchInstallReferrer(mContext, spiedCallback)
+        val callback = Mockito.spy(SpyableInstallReferrerCallback(latch))
 
+        InstallReferrerHelper.fetchInstallReferrer(mContext, callback)
         latch.await(5, TimeUnit.SECONDS)
 
-        verify(spiedCallback).onFailed()
-        verify(spiedCallback, Mockito.never()).onReceived(Mockito.anyString())
+        verify(callback).onFailed()
+        verify(callback, Mockito.never()).onReceived(Mockito.anyString())
+    }
+}
+
+open class SpyableInstallReferrerCallback(
+    val latch: CountDownLatch
+) : InstallReferrerHelper.InstallReferrerCallback {
+    override fun onReceived(installReferrer: String?) {
+        latch.countDown()
+    }
+
+    override fun onFailed() {
+        latch.countDown()
     }
 }