From c773c55dbb2249828340537eebf4a87a6aa23444 Mon Sep 17 00:00:00 2001
From: mcarare <48995920+mcarare@users.noreply.github.com>
Date: Thu, 6 Apr 2023 21:04:58 +0300
Subject: [PATCH] Bug 1825071 - Refactor ImageButtonTest to remove
 FenixRobolectricTestRunner.

---
 .../org/mozilla/fenix/ext/ImageButtonTest.kt  | 36 +++++++++++--------
 1 file changed, 22 insertions(+), 14 deletions(-)

diff --git a/app/src/test/java/org/mozilla/fenix/ext/ImageButtonTest.kt b/app/src/test/java/org/mozilla/fenix/ext/ImageButtonTest.kt
index d29f27f9d5..f532951963 100644
--- a/app/src/test/java/org/mozilla/fenix/ext/ImageButtonTest.kt
+++ b/app/src/test/java/org/mozilla/fenix/ext/ImageButtonTest.kt
@@ -6,36 +6,44 @@ package org.mozilla.fenix.ext
 
 import android.view.View
 import android.widget.ImageButton
-import mozilla.components.support.test.robolectric.testContext
-import org.junit.Assert.assertEquals
-import org.junit.Assert.assertFalse
-import org.junit.Assert.assertTrue
+import io.mockk.Runs
+import io.mockk.every
+import io.mockk.just
+import io.mockk.mockk
+import io.mockk.verify
+import org.junit.Before
 import org.junit.Test
-import org.junit.runner.RunWith
-import org.mozilla.fenix.helpers.FenixRobolectricTestRunner
 
-@RunWith(FenixRobolectricTestRunner::class)
 class ImageButtonTest {
-    private val imageButton = ImageButton(testContext)
+    private val imageButton: ImageButton = mockk()
+
+    @Before
+    fun setup() {
+        every { imageButton.visibility = any() } just Runs
+        every { imageButton.isEnabled = any() } just Runs
+    }
 
     @Test
     fun `Hide and disable`() {
         imageButton.hideAndDisable()
-        assertFalse(imageButton.isEnabled)
-        assertEquals(View.INVISIBLE, imageButton.visibility)
+
+        verify { imageButton.isEnabled = false }
+        verify { imageButton.visibility = View.INVISIBLE }
     }
 
     @Test
     fun `Show and enable`() {
         imageButton.showAndEnable()
-        assertTrue(imageButton.isEnabled)
-        assertEquals(View.VISIBLE, imageButton.visibility)
+
+        verify { imageButton.isEnabled = true }
+        verify { imageButton.visibility = View.VISIBLE }
     }
 
     @Test
     fun `Remove and disable`() {
         imageButton.removeAndDisable()
-        assertFalse(imageButton.isEnabled)
-        assertEquals(View.GONE, imageButton.visibility)
+
+        verify { imageButton.isEnabled = false }
+        verify { imageButton.visibility = View.GONE }
     }
 }