From c6f68bf0a00fdbe38525718490c19127e8878a08 Mon Sep 17 00:00:00 2001
From: Wojtek Zieba <wojtek.zieba@automattic.com>
Date: Tue, 14 Nov 2023 16:43:14 +0100
Subject: [PATCH] tests: add unit tests for `AndroidDeviceInfoRepository`

---
 .../AndroidDeviceInfoRepositoryTest.kt        | 78 +++++++++++++++++++
 1 file changed, 78 insertions(+)
 create mode 100644 parsely/src/test/java/com/parsely/parselyandroid/AndroidDeviceInfoRepositoryTest.kt

diff --git a/parsely/src/test/java/com/parsely/parselyandroid/AndroidDeviceInfoRepositoryTest.kt b/parsely/src/test/java/com/parsely/parselyandroid/AndroidDeviceInfoRepositoryTest.kt
new file mode 100644
index 00000000..0f41f7b6
--- /dev/null
+++ b/parsely/src/test/java/com/parsely/parselyandroid/AndroidDeviceInfoRepositoryTest.kt
@@ -0,0 +1,78 @@
+package com.parsely.parselyandroid
+
+import org.assertj.core.api.Assertions.assertThat
+import org.junit.Before
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.robolectric.RobolectricTestRunner
+import org.robolectric.annotation.Config
+import org.robolectric.shadows.ShadowBuild
+
+private const val SDK_VERSION = 33
+private const val MANUFACTURER = "test manufacturer"
+
+@RunWith(RobolectricTestRunner::class)
+@Config(sdk = [SDK_VERSION])
+internal class AndroidDeviceInfoRepositoryTest {
+
+    private lateinit var sut: AndroidDeviceInfoRepository
+
+    @Before
+    fun setUp() {
+        ShadowBuild.setManufacturer(MANUFACTURER)
+    }
+
+    @Test
+    fun `given the advertisement id exists, when collecting device info, then parsely site uuid is advertisement id`() {
+        // given
+        val advertisementId = "ad id"
+        sut = AndroidDeviceInfoRepository(
+            advertisementIdProvider = { advertisementId },
+            androidIdProvider = { "android id" })
+
+        // when
+        val result = sut.collectDeviceInfo()
+
+        // then
+        assertThat(result).isEqualTo(expectedConstantDeviceInfo + ("parsely_site_uuid" to advertisementId))
+    }
+
+    @Test
+    fun `given the advertisement is null and android id is not, when collecting device info, then parsely id is android id`() {
+        // given
+        val androidId = "android id"
+        sut = AndroidDeviceInfoRepository(
+            advertisementIdProvider = { null },
+            androidIdProvider = { androidId }
+        )
+
+        // when
+        val result = sut.collectDeviceInfo()
+
+        // then
+        assertThat(result).isEqualTo(expectedConstantDeviceInfo + ("parsely_site_uuid" to androidId))
+    }
+
+    @Test
+    fun `given both advertisement id and android id are null, when collecting device info, then parsely id is empty`() {
+        // given
+        sut = AndroidDeviceInfoRepository(
+            advertisementIdProvider = { null },
+            androidIdProvider = { null }
+        )
+
+        // when
+        val result = sut.collectDeviceInfo()
+
+        // then
+        assertThat(result).isEqualTo(expectedConstantDeviceInfo + ("parsely_site_uuid" to ""))
+    }
+
+    private companion object {
+        val expectedConstantDeviceInfo = mapOf(
+            "manufacturer" to MANUFACTURER,
+            "os" to "android",
+            "os_version" to "$SDK_VERSION"
+        )
+    }
+}