/* 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.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 org.junit.Test
import org.junit.runner.RunWith
import org.mozilla.fenix.helpers.FenixRobolectricTestRunner

@RunWith(FenixRobolectricTestRunner::class)
class ImageButtonTest {
    private val imageButton = ImageButton(testContext)

    @Test
    fun `Hide and disable`() {
        imageButton.hideAndDisable()
        assertFalse(imageButton.isEnabled)
        assertEquals(View.INVISIBLE, imageButton.visibility)
    }

    @Test
    fun `Show and enable`() {
        imageButton.showAndEnable()
        assertTrue(imageButton.isEnabled)
        assertEquals(View.VISIBLE, imageButton.visibility)
    }

    @Test
    fun `Remove and disable`() {
        imageButton.removeAndDisable()
        assertFalse(imageButton.isEnabled)
        assertEquals(View.GONE, imageButton.visibility)
    }
}