From 0d4ceb186c1e9abb843fa509051eda343aba4a65 Mon Sep 17 00:00:00 2001
From: Tiger Oakes <toakes@mozilla.com>
Date: Fri, 12 Jun 2020 12:01:52 -0700
Subject: [PATCH] Add test for SearchStringValidator

---
 .../search/SearchStringValidatorTest.kt       | 104 ++++++++++++++++++
 1 file changed, 104 insertions(+)
 create mode 100644 app/src/test/java/org/mozilla/fenix/settings/search/SearchStringValidatorTest.kt

diff --git a/app/src/test/java/org/mozilla/fenix/settings/search/SearchStringValidatorTest.kt b/app/src/test/java/org/mozilla/fenix/settings/search/SearchStringValidatorTest.kt
new file mode 100644
index 000000000..6e8190411
--- /dev/null
+++ b/app/src/test/java/org/mozilla/fenix/settings/search/SearchStringValidatorTest.kt
@@ -0,0 +1,104 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+package org.mozilla.fenix.settings.search
+
+import io.mockk.every
+import io.mockk.mockk
+import mozilla.components.concept.fetch.Client
+import mozilla.components.concept.fetch.MutableHeaders
+import mozilla.components.concept.fetch.Request
+import mozilla.components.concept.fetch.Response
+import org.junit.Assert.assertEquals
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.mozilla.fenix.helpers.FenixRobolectricTestRunner
+import java.io.ByteArrayInputStream
+import java.io.IOException
+
+@RunWith(FenixRobolectricTestRunner::class)
+class SearchStringValidatorTest {
+
+    private val client: Client = mockk()
+
+    @Test
+    fun `test MDN url`() {
+        val request = Request(
+            url = "https://developer.mozilla.org/en-US/search?q=1"
+        )
+        every { client.fetch(request) } returns Response(
+            url = "",
+            status = 200,
+            headers = MutableHeaders(),
+            body = Response.Body(ByteArrayInputStream("".toByteArray()))
+        )
+
+        val result = SearchStringValidator.isSearchStringValid(client, "https://developer.mozilla.org/en-US/search?q=%s")
+
+        assertEquals(SearchStringValidator.Result.Success, result)
+    }
+
+    @Test
+    fun `normalize search url`() {
+        val request = Request(
+            url = "http://firefox.com/search?q=1"
+        )
+        every { client.fetch(request) } returns Response(
+            url = "",
+            status = 200,
+            headers = MutableHeaders(),
+            body = Response.Body(ByteArrayInputStream("".toByteArray()))
+        )
+
+        val result = SearchStringValidator.isSearchStringValid(client, "firefox.com/search?q=%s")
+
+        assertEquals(SearchStringValidator.Result.Success, result)
+    }
+
+    @Test
+    fun `fail if IOException is thrown`() {
+        every { client.fetch(any()) } throws IOException()
+
+        val result = SearchStringValidator.isSearchStringValid(client, "https://developer.mozilla.org/en-US/search?q=%s")
+
+        assertEquals(SearchStringValidator.Result.CannotReach, result)
+    }
+
+    @Test
+    fun `fail if IllegalArgumentException is thrown`() {
+        every { client.fetch(any()) } throws IllegalArgumentException()
+
+        val result = SearchStringValidator.isSearchStringValid(client, "https://developer.mozilla.org/en-US/search?q=%s")
+
+        assertEquals(SearchStringValidator.Result.CannotReach, result)
+    }
+
+    @Test
+    fun `pass if status code is not in success range`() {
+        every { client.fetch(any()) } returns Response(
+            url = "",
+            status = 400,
+            headers = MutableHeaders(),
+            body = Response.Body(ByteArrayInputStream("".toByteArray()))
+        )
+
+        val result = SearchStringValidator.isSearchStringValid(client, "https://developer.mozilla.org/en-US/search?q=%s")
+
+        assertEquals(SearchStringValidator.Result.CannotReach, result)
+    }
+
+    @Test
+    fun `pass even if 404 status is returned`() {
+        every { client.fetch(any()) } returns Response(
+            url = "",
+            status = 404,
+            headers = MutableHeaders(),
+            body = Response.Body(ByteArrayInputStream("".toByteArray()))
+        )
+
+        val result = SearchStringValidator.isSearchStringValid(client, "https://developer.mozilla.org/en-US/search?q=%s")
+
+        assertEquals(SearchStringValidator.Result.Success, result)
+    }
+}