Merge pull request #3107 from wordpress-mobile/issue/12775-connect-info-error-handling

JorgeMucientes · web-flow · commit 5dfbbc993d24 · 2024-10-16T17:20:18.000+02:00
Allow identifying black-flagged websites when fetching WPCom site info
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/activity/ActivityLogRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/activity/ActivityLogRestClientTest.kt
@@ -591,7 +591,8 @@ class ActivityLogRestClientTest {
                 eq(false),
                 any(),
                 eq(false),
-                customGsonBuilder = anyOrNull()
+                customGsonBuilder = anyOrNull(),
+                authenticatedRequest = any()
         )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
@@ -612,7 +613,8 @@ class ActivityLogRestClientTest {
                 eq(false),
                 any(),
                 eq(false),
-                customGsonBuilder = anyOrNull()
+                customGsonBuilder = anyOrNull(),
+                authenticatedRequest = any()
         )).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
         return response
@@ -698,7 +700,8 @@ class ActivityLogRestClientTest {
                 eq(false),
                 any(),
                 eq(false),
-                customGsonBuilder = anyOrNull()
+                customGsonBuilder = anyOrNull(),
+                authenticatedRequest = any()
         )).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
         return response
@@ -717,7 +720,8 @@ class ActivityLogRestClientTest {
                 eq(false),
                 any(),
                 eq(false),
-                customGsonBuilder = anyOrNull()
+                customGsonBuilder = anyOrNull(),
+                authenticatedRequest = any()
         )
         ).thenReturn(response)
         return response
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/bloggingprompts/BloggingPromptsRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/bloggingprompts/BloggingPromptsRestClientTest.kt
@@ -212,7 +212,8 @@ class BloggingPromptsRestClientTest {
                 eq(BloggingPromptsListResponseTypeToken.type),
                 eq(false),
                 any(),
-                eq(false)
+                eq(false),
+                authenticatedRequest = any()
             )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/comments/CommentsRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/comments/CommentsRestClientTest.kt
@@ -572,7 +572,8 @@ class CommentsRestClientTest {
                         any(),
                         any(),
                         any(),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         return response
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/dashboard/CardsRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/dashboard/CardsRestClientTest.kt
@@ -400,7 +400,8 @@ class CardsRestClientTest {
                         eq(false),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/experiments/ExperimentRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/experiments/ExperimentRestClientTest.kt
@@ -176,7 +176,8 @@ class ExperimentRestClientTest {
                         eq(false),
                         any(),
                         eq(true),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
     }
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/mobile/FeatureFlagsRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/mobile/FeatureFlagsRestClientTest.kt
@@ -212,7 +212,8 @@ class FeatureFlagsRestClientTest {
                 eq(false),
                 any(),
                 eq(false),
-                customGsonBuilder = anyOrNull()
+                customGsonBuilder = anyOrNull(),
+                authenticatedRequest = any()
             )
         ).thenReturn(response)
     }
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/mobile/RemoteConfigRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/mobile/RemoteConfigRestClientTest.kt
@@ -138,7 +138,8 @@ class RemoteConfigRestClientTest {
                 eq(false),
                 any(),
                 eq(false),
-                customGsonBuilder = anyOrNull()
+                customGsonBuilder = anyOrNull(),
+                authenticatedRequest = any()
             )
         ).thenReturn(response)
     }
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/planoffers/PlanOffersRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/planoffers/PlanOffersRestClientTest.kt
@@ -87,7 +87,8 @@ class PlanOffersRestClientTest {
                         eq(false),
                         any(),
                         eq(true),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
     }
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/products/ProductsRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/products/ProductsRestClientTest.kt
@@ -92,7 +92,8 @@ class ProductsRestClientTest {
                         eq(false),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
     }
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/scan/ScanRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/scan/ScanRestClientTest.kt
@@ -423,7 +423,8 @@ class ScanRestClientTest {
                         eq(false),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
@@ -508,7 +509,8 @@ class ScanRestClientTest {
                         eq(false),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         return response
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/site/SiteRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/site/SiteRestClientTest.kt
@@ -9,6 +9,7 @@ import org.junit.Before
 import org.junit.Test
 import org.junit.runner.RunWith
 import org.mockito.Mock
+import org.mockito.Mockito.mockStatic
 import org.mockito.junit.MockitoJUnitRunner
 import org.mockito.kotlin.KArgumentCaptor
 import org.mockito.kotlin.any
@@ -34,12 +35,15 @@ import org.wordpress.android.fluxc.network.rest.wpcom.site.SiteWPComRestResponse
 import org.wordpress.android.fluxc.network.rest.wpcom.site.StatusType.ERROR
 import org.wordpress.android.fluxc.network.rest.wpcom.site.StatusType.SUCCESS
 import org.wordpress.android.fluxc.store.SiteStore.PostFormatsErrorType
+import org.wordpress.android.fluxc.store.SiteStore.SiteErrorType
 import org.wordpress.android.fluxc.store.SiteStore.SiteFilter.WPCOM
 import org.wordpress.android.fluxc.store.SiteStore.SiteVisibility
 import org.wordpress.android.fluxc.store.SiteStore.SiteVisibility.COMING_SOON
 import org.wordpress.android.fluxc.store.SiteStore.SiteVisibility.PUBLIC
 import org.wordpress.android.fluxc.test
+import org.wordpress.android.fluxc.tools.initCoroutineEngine
 import org.wordpress.android.util.DateTimeUtils
+import org.wordpress.android.util.UrlUtils
 import kotlin.test.assertNotNull
 
 @RunWith(MockitoJUnitRunner::class)
@@ -63,13 +67,14 @@ class SiteRestClientTest {
         paramsCaptor = argumentCaptor()
         bodyCaptor = argumentCaptor()
         restClient = SiteRestClient(
-                null,
-                dispatcher,
-                requestQueue,
-                appSecrets,
-                wpComGsonRequestBuilder,
-                accessToken,
-                userAgent
+            appContext = null,
+            dispatcher = dispatcher,
+            requestQueue = requestQueue,
+            appSecrets = appSecrets,
+            wpComGsonRequestBuilder = wpComGsonRequestBuilder,
+            coroutineEngine = initCoroutineEngine(),
+            accessToken = accessToken,
+            userAgent = userAgent
         )
         whenever(site.siteId).thenReturn(siteId)
     }
@@ -610,6 +615,22 @@ class SiteRestClientTest {
         }
     }
 
+    @Test
+    fun `given a suspended WPCom website, when fetching site info, then return correct error`() = test {
+        val urlUtilsMock = mockStatic(UrlUtils::class.java)
+        whenever(UrlUtils.addUrlSchemeIfNeeded(any(), any())).thenAnswer { it.arguments[0] as String }
+        val error = WPComGsonNetworkError(BaseNetworkError(GenericErrorType.INVALID_RESPONSE, "")).apply {
+            apiError = "connection_disabled"
+        }
+        initGetResponse(ConnectSiteInfoResponse::class.java, null, error)
+
+        val result = restClient.fetchConnectSiteInfoSync("test.com")
+
+        assertThat(result.error).isNotNull
+        assertThat(result.error!!.type).isEqualTo(SiteErrorType.WPCOM_SITE_SUSPENDED)
+        urlUtilsMock.close()
+    }
+
     private suspend fun initSiteResponse(
         data: SiteWPComRestResponse? = null,
         error: WPComGsonNetworkError? = null
@@ -654,10 +675,13 @@ class SiteRestClientTest {
 
     private suspend fun <T> initGetResponse(
         clazz: Class<T>,
-        data: T,
+        data: T?,
         error: WPComGsonNetworkError? = null
     ): Response<T> {
-        val response = if (error != null) Response.Error(error) else Success(data)
+        if (error == null && data == null) {
+            error("Either data or error must be provided")
+        }
+        val response = if (error != null) Response.Error(error) else Success<T>(data!!)
         whenever(
                 wpComGsonRequestBuilder.syncGetRequest(
                         eq(restClient),
@@ -667,8 +691,8 @@ class SiteRestClientTest {
                         any(),
                         any(),
                         any(),
-                        customGsonBuilder = anyOrNull()
-
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         return response
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/InsightsRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/InsightsRestClientTest.kt
@@ -512,7 +512,8 @@ class InsightsRestClientTest {
                         eq(cachingEnabled),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/insights/MostPopularRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/insights/MostPopularRestClientTest.kt
@@ -103,7 +103,8 @@ class MostPopularRestClientTest {
                         eq(true),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/insights/PostingActivityRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/insights/PostingActivityRestClientTest.kt
@@ -120,7 +120,8 @@ class PostingActivityRestClientTest {
                         eq(true),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/insights/SummaryRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/insights/SummaryRestClientTest.kt
@@ -100,7 +100,8 @@ class SummaryRestClientTest {
                 eq(false),
                 any(),
                 eq(false),
-                customGsonBuilder = anyOrNull()
+                customGsonBuilder = anyOrNull(),
+                authenticatedRequest = any()
             )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/subscribers/SubscribersRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/subscribers/SubscribersRestClientTest.kt
@@ -146,7 +146,8 @@ class SubscribersRestClientTest {
                 eq(cachingEnabled),
                 any(),
                 eq(false),
-                customGsonBuilder = anyOrNull()
+                customGsonBuilder = anyOrNull(),
+                authenticatedRequest = any()
             )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/AuthorsRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/AuthorsRestClientTest.kt
@@ -174,7 +174,8 @@ class AuthorsRestClientTest {
                         eq(cachingEnabled),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/ClicksRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/ClicksRestClientTest.kt
@@ -220,7 +220,8 @@ class ClicksRestClientTest {
                         eq(cachingEnabled),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/CountryViewsRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/CountryViewsRestClientTest.kt
@@ -170,7 +170,8 @@ class CountryViewsRestClientTest {
                         eq(cachingEnabled),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/FileDownloadsRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/FileDownloadsRestClientTest.kt
@@ -174,7 +174,8 @@ class FileDownloadsRestClientTest {
                         eq(cachingEnabled),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/PostAndPageViewsRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/PostAndPageViewsRestClientTest.kt
@@ -170,7 +170,8 @@ class PostAndPageViewsRestClientTest {
                         eq(cachingEnabled),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/ReferrersRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/ReferrersRestClientTest.kt
@@ -313,7 +313,8 @@ class ReferrersRestClientTest {
                         eq(cachingEnabled),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/SearchTermsRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/SearchTermsRestClientTest.kt
@@ -170,7 +170,8 @@ class SearchTermsRestClientTest {
                         eq(cachingEnabled),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/VideoPlaysRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/VideoPlaysRestClientTest.kt
@@ -174,7 +174,8 @@ class VideoPlaysRestClientTest {
                         eq(cachingEnabled),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/VisitAndViewsRestClientTest.kt b/example/src/test/java/org/wordpress/android/fluxc/network/rest/wpcom/stats/time/VisitAndViewsRestClientTest.kt
@@ -164,7 +164,8 @@ class VisitAndViewsRestClientTest {
                         eq(cachingEnabled),
                         any(),
                         eq(false),
-                        customGsonBuilder = anyOrNull()
+                        customGsonBuilder = anyOrNull(),
+                        authenticatedRequest = any()
                 )
         ).thenReturn(response)
         whenever(site.siteId).thenReturn(siteId)
diff --git a/fluxc/src/main/java/org/wordpress/android/fluxc/network/rest/wpcom/WPComGsonRequestBuilder.kt b/fluxc/src/main/java/org/wordpress/android/fluxc/network/rest/wpcom/WPComGsonRequestBuilder.kt
diff --git a/fluxc/src/main/java/org/wordpress/android/fluxc/network/rest/wpcom/site/SiteRestClient.kt b/fluxc/src/main/java/org/wordpress/android/fluxc/network/rest/wpcom/site/SiteRestClient.kt
diff --git a/fluxc/src/main/java/org/wordpress/android/fluxc/store/SiteStore.kt b/fluxc/src/main/java/org/wordpress/android/fluxc/store/SiteStore.kt

Original file line number	Diff line number	Diff line change
`@@ -176,7 +176,8 @@ class ExperimentRestClientTest {`
`176`	`176`	`eq(false),`
`177`	`177`	`any(),`
`178`	`178`	`eq(true),`
`179`		`- customGsonBuilder = anyOrNull()`
	`179`	`+ customGsonBuilder = anyOrNull(),`
	`180`	`+ authenticatedRequest = any()`
`180`	`181`	`)`
`181`	`182`	`).thenReturn(response)`
`182`	`183`	`}`
Original file line number	Diff line number	Diff line change
`@@ -212,7 +212,8 @@ class FeatureFlagsRestClientTest {`
`212`	`212`	`eq(false),`
`213`	`213`	`any(),`
`214`	`214`	`eq(false),`
`215`		`- customGsonBuilder = anyOrNull()`
	`215`	`+ customGsonBuilder = anyOrNull(),`
	`216`	`+ authenticatedRequest = any()`
`216`	`217`	`)`
`217`	`218`	`).thenReturn(response)`
`218`	`219`	`}`
Original file line number	Diff line number	Diff line change
`@@ -138,7 +138,8 @@ class RemoteConfigRestClientTest {`
`138`	`138`	`eq(false),`
`139`	`139`	`any(),`
`140`	`140`	`eq(false),`
`141`		`- customGsonBuilder = anyOrNull()`
	`141`	`+ customGsonBuilder = anyOrNull(),`
	`142`	`+ authenticatedRequest = any()`
`142`	`143`	`)`
`143`	`144`	`).thenReturn(response)`
`144`	`145`	`}`
Original file line number	Diff line number	Diff line change
`@@ -87,7 +87,8 @@ class PlanOffersRestClientTest {`
`87`	`87`	`eq(false),`
`88`	`88`	`any(),`
`89`	`89`	`eq(true),`
`90`		`- customGsonBuilder = anyOrNull()`
	`90`	`+ customGsonBuilder = anyOrNull(),`
	`91`	`+ authenticatedRequest = any()`
`91`	`92`	`)`
`92`	`93`	`).thenReturn(response)`
`93`	`94`	`}`
Original file line number	Diff line number	Diff line change
`@@ -92,7 +92,8 @@ class ProductsRestClientTest {`
`92`	`92`	`eq(false),`
`93`	`93`	`any(),`
`94`	`94`	`eq(false),`
`95`		`- customGsonBuilder = anyOrNull()`
	`95`	`+ customGsonBuilder = anyOrNull(),`
	`96`	`+ authenticatedRequest = any()`
`96`	`97`	`)`
`97`	`98`	`).thenReturn(response)`
`98`	`99`	`}`