Ext Test For ImageButton

app/src/test/java/org/mozilla/fenix/ext/ImageButtonTest.kt

@@ -0,0 +1,34 @@
+package org.mozilla.fenix.ext
+
+import kotlinx.coroutines.ObsoleteCoroutinesApi
+import mozilla.components.support.test.robolectric.testContext
+import org.mozilla.fenix.TestApplication
+import org.junit.Test
+import org.junit.Assert.assertFalse
+import org.junit.Assert.assertTrue
+import org.junit.Assert.assertEquals
+import org.junit.runner.RunWith
+import org.robolectric.RobolectricTestRunner
+import org.robolectric.annotation.Config
+import android.widget.ImageButton
+
+@ObsoleteCoroutinesApi
+@RunWith(RobolectricTestRunner::class)
+@Config(application = TestApplication::class)
+class ImageButtonTest {
+    val imageButton = ImageButton(testContext)
+
+    @Test
+    fun `Hide and disable`() {
+        imageButton.hideAndDisable()
+        assertFalse(imageButton.isEnabled)
+        assertEquals(4, imageButton.visibility)
+    }
+
+    @Test
+    fun `Show and enable`() {
+        imageButton.showAndEnable()
+        assertTrue(imageButton.isEnabled)
+        assertEquals(0, imageButton.visibility)
+    }
+}