From 3d4819a5663ed1ce167f817ee6f9fbd4070bba7e Mon Sep 17 00:00:00 2001
From: Michelle Wong <mwong@mozilla.com>
Date: Fri, 12 Jul 2019 17:06:57 -0700
Subject: [PATCH] Issue #2543: wrote tests for languageAndMaybeCountryMatch

---
 .../mozilla/tv/firefox/ext/LocaleKtTest.kt    | 46 +++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/app/src/test/java/org/mozilla/tv/firefox/ext/LocaleKtTest.kt b/app/src/test/java/org/mozilla/tv/firefox/ext/LocaleKtTest.kt
index db367556f9..183e7afde2 100644
--- a/app/src/test/java/org/mozilla/tv/firefox/ext/LocaleKtTest.kt
+++ b/app/src/test/java/org/mozilla/tv/firefox/ext/LocaleKtTest.kt
@@ -1,7 +1,12 @@
 package org.mozilla.tv.firefox.ext
 
+import io.mockk.every
+import io.mockk.mockk
+import io.mockk.mockkStatic
+import io.mockk.verify
 import org.junit.Assert.*
 import org.junit.Test
+import org.mozilla.tv.firefox.architecture.KillswitchLocales
 import java.util.*
 
 class LocaleKtTest {
@@ -62,4 +67,45 @@ class LocaleKtTest {
 
         assertTrue(Locale.US.languageAndMaybeCountryMatch(arrayOf(Locale.CANADA, allowed)))
     }
+
+    @Test
+    fun `GIVEN all killswitch locale is allowed WHEN any locale is passed THEN should return true`() {
+        val allowed = KillswitchLocales.All
+
+        assertTrue(Locale.US.languageAndMaybeCountryMatch(allowed))
+        assertTrue(Locale.CANADA.languageAndMaybeCountryMatch(allowed))
+        assertTrue(Locale.UK.languageAndMaybeCountryMatch(allowed))
+        assertTrue(Locale.GERMANY.languageAndMaybeCountryMatch(allowed))
+        assertTrue(Locale.CHINA.languageAndMaybeCountryMatch(allowed))
+
+        assertTrue(Locale.ENGLISH.languageAndMaybeCountryMatch(allowed))
+        assertTrue(Locale.GERMAN.languageAndMaybeCountryMatch(allowed))
+        assertTrue(Locale.CHINESE.languageAndMaybeCountryMatch(allowed))
+    }
+
+    @Test
+    fun `WHEN KillswitchLocale ActiveIn is passed THEN overload should be called`() {
+        // Mock extension functions in this class
+        // If this test fails, check if this class has been moved, causing this string to be incorrect
+        mockkStatic("org.mozilla.tv.firefox.ext.LocaleKt")
+
+        val allowed = KillswitchLocales.ActiveIn(Locale.US)
+
+        val locales = listOf(
+                Locale.US,
+                Locale.CANADA,
+                Locale.UK,
+                Locale.GERMANY,
+                Locale.CHINA,
+                Locale.ENGLISH,
+                Locale.GERMAN,
+                Locale.CHINESE
+        )
+
+        locales.forEach { it.languageAndMaybeCountryMatch(allowed) }
+
+        // Verify that languageAndMaybeCountryMatch is called with an array of locales (i.e., the overload)
+        locales.forEach { verify(exactly = 1) { it.languageAndMaybeCountryMatch(any<Array<Locale>>()) } }
+    }
+
 }