add getAccessToken implementation

Author: cmonfortep

subscriptions/subscriptions-impl/src/main/java/com/duckduckgo/subscriptions/impl/messaging/SubscriptionsJSHelper.kt

@@ -19,6 +19,7 @@ package com.duckduckgo.subscriptions.impl.messaging
 import com.duckduckgo.di.scopes.AppScope
 import com.duckduckgo.js.messaging.api.JsCallbackData
 import com.duckduckgo.subscriptions.api.SubscriptionsJSHelper
+import com.duckduckgo.subscriptions.impl.AccessTokenResult
 import com.duckduckgo.subscriptions.impl.SubscriptionsManager
 import com.squareup.anvil.annotations.ContributesBinding
 import javax.inject.Inject
@@ -52,6 +53,10 @@ class RealSubscriptionsJSHelper @Inject constructor(
             getSubscriptionDetailsData(featureName, method, it)
         }
 
+        METHOD_GET_AUTH_ACCESS_TOKEN -> id?.let {
+            getAuthAccessTokenData(featureName, method, it)
+        }
+
         else -> null
     }
 
@@ -72,6 +77,17 @@ class RealSubscriptionsJSHelper @Inject constructor(
         return JsCallbackData(jsonPayload, featureName, method, id)
     }
 
+    private suspend fun getAuthAccessTokenData(featureName: String, method: String, id: String): JsCallbackData {
+        val jsonPayload = when (val result = subscriptionsManager.getAccessToken()) {
+            is AccessTokenResult.Success -> JSONObject().apply {
+                put(ACCESS_TOKEN, result.accessToken)
+            }
+            is AccessTokenResult.Failure -> JSONObject()
+        }
+
+        return JsCallbackData(jsonPayload, featureName, method, id)
+    }
+
     companion object {
         private const val METHOD_HANDSHAKE = "handshake"
         private const val METHOD_SUBSCRIPTION_DETAILS = "subscriptionDetails"
@@ -89,5 +105,6 @@ class RealSubscriptionsJSHelper @Inject constructor(
         private const val EXPIRES_OR_RENEWS_AT = "expiresOrRenewsAt"
         private const val PAYMENT_PLATFORM = "paymentPlatform"
         private const val STATUS = "status"
+        private const val ACCESS_TOKEN = "accessToken"
     }
 }

subscriptions/subscriptions-impl/src/test/java/com/duckduckgo/subscriptions/impl/messaging/RealSubscriptionsJSHelperTest.kt

@@ -5,6 +5,7 @@ import com.duckduckgo.common.test.CoroutineTestRule
 import com.duckduckgo.js.messaging.api.JsCallbackData
 import com.duckduckgo.subscriptions.api.SubscriptionStatus.AUTO_RENEWABLE
 import com.duckduckgo.subscriptions.api.SubscriptionStatus.EXPIRED
+import com.duckduckgo.subscriptions.impl.AccessTokenResult
 import com.duckduckgo.subscriptions.impl.SubscriptionsConstants
 import com.duckduckgo.subscriptions.impl.SubscriptionsConstants.MONTHLY
 import com.duckduckgo.subscriptions.impl.SubscriptionsConstants.YEARLY
@@ -20,6 +21,7 @@ import org.junit.Test
 import org.junit.runner.RunWith
 import org.mockito.kotlin.mock
 import org.mockito.kotlin.whenever
+import org.robolectric.RobolectricTestRunner
 
 @RunWith(AndroidJUnit4::class)
 class RealSubscriptionsJSHelperTest {
@@ -168,4 +170,45 @@ class RealSubscriptionsJSHelperTest {
         assertEquals(expected.featureName, result.featureName)
         assertEquals(expected.params.toString(), result.params.toString())
     }
+
+    @Test
+    fun whenGetAuthAccessTokenRequestWithSuccessfulTokenThenReturnJsCallbackDataWithToken() = runTest {
+        val method = "getAuthAccessToken"
+        val id = "123"
+        val expectedToken = "test-access-token"
+
+        whenever(mockSubscriptionsManager.getAccessToken()).thenReturn(AccessTokenResult.Success(expectedToken))
+
+        val result = testee.processJsCallbackMessage(featureName, method, id, null)
+
+        val jsonPayload = JSONObject().apply {
+            put("accessToken", expectedToken)
+        }
+
+        val expected = JsCallbackData(jsonPayload, featureName, method, id)
+
+        assertEquals(expected.id, result?.id)
+        assertEquals(expected.featureName, result?.featureName)
+        assertEquals(expected.method, result?.method)
+        assertEquals(expected.params.toString(), result?.params.toString())
+    }
+
+    @Test
+    fun whenGetAuthAccessTokenRequestWithFailedTokenThenReturnJsCallbackDataWithEmptyObject() = runTest {
+        val method = "getAuthAccessToken"
+        val id = "123"
+
+        whenever(mockSubscriptionsManager.getAccessToken()).thenReturn(AccessTokenResult.Failure("Token not found"))
+
+        val result = testee.processJsCallbackMessage(featureName, method, id, null)
+
+        val jsonPayload = JSONObject()
+
+        val expected = JsCallbackData(jsonPayload, featureName, method, id)
+
+        assertEquals(expected.id, result?.id)
+        assertEquals(expected.featureName, result?.featureName)
+        assertEquals(expected.method, result?.method)
+        assertEquals(expected.params.toString(), result?.params.toString())
+    }
 }