diff --git a/app/src/test/java/org/mozilla/fenix/ext/ImageButtonTest.kt b/app/src/test/java/org/mozilla/fenix/ext/ImageButtonTest.kt new file mode 100644 index 000000000..64704d29b --- /dev/null +++ b/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) + } +}