/*
 * Copyright 2018 Allan Wang
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.pitchedapps.frost.activities

import android.content.Intent
import androidx.test.ext.junit.runners.AndroidJUnit4
import androidx.test.rule.ActivityTestRule
import com.pitchedapps.frost.helper.getResource
import com.pitchedapps.frost.utils.ARG_COOKIE
import com.pitchedapps.frost.utils.ARG_IMAGE_URL
import com.pitchedapps.frost.utils.ARG_TEXT
import com.pitchedapps.frost.utils.isIndirectImageUrl
import okhttp3.mockwebserver.Dispatcher
import okhttp3.mockwebserver.MockResponse
import okhttp3.mockwebserver.MockWebServer
import okhttp3.mockwebserver.RecordedRequest
import okio.Buffer
import okio.source
import org.junit.Rule
import org.junit.Test
import org.junit.rules.Timeout
import org.junit.runner.RunWith
import kotlin.test.assertEquals
import kotlin.test.assertFalse
import kotlin.test.assertNull
import kotlin.test.assertTrue

@RunWith(AndroidJUnit4::class)
class ImageActivityTest {

    @get:Rule
    val activity: ActivityTestRule<ImageActivity> =
        ActivityTestRule(ImageActivity::class.java, true, false)

    @get:Rule
    val globalTimeout: Timeout = Timeout.seconds(15)

    private fun launchActivity(imageUrl: String, text: String? = null, cookie: String? = null) {
        assertFalse(
            imageUrl.isIndirectImageUrl,
            "For simplicity, urls that are direct will be used without modifications in the production code."
        )
        val intent = Intent().apply {
            putExtra(ARG_IMAGE_URL, imageUrl)
            putExtra(ARG_TEXT, text)
            putExtra(ARG_COOKIE, cookie)
        }
        activity.launchActivity(intent)
    }

    private val mockServer: MockWebServer by lazy {
        val img = Buffer()
        img.writeAll(getResource("bayer-pattern.jpg").source())
        MockWebServer().apply {
            dispatcher = object : Dispatcher() {
                override fun dispatch(request: RecordedRequest): MockResponse =
                    when {
                        request.path?.contains("text") == true -> MockResponse().setResponseCode(200).setBody(
                            "Valid mock text response"
                        )
                        request.path?.contains("image") == true -> MockResponse().setResponseCode(
                            200
                        ).setBody(
                            img
                        )
                        else -> MockResponse().setResponseCode(404).setBody("Error mock response")
                    }
            }
            start()
        }
    }

    @Test
    fun validImageTest() {
        launchActivity(mockServer.url("image").toString())
        mockServer.takeRequest()
        with(activity.activity) {
            assertEquals(1, mockServer.requestCount, "One http request expected")
            assertEquals(
                FabStates.DOWNLOAD,
                fabAction,
                "Image should be successful, image should be downloaded"
            )
            assertTrue(tempFile.exists(), "Image should be located at temp file")
            assertTrue(
                System.currentTimeMillis() - tempFile.lastModified() < 2000L,
                "Image should have been modified within the last few seconds"
            )
            assertNull(errorRef, "No error should exist")
            tempFile.delete()
        }
    }

    @Test
    fun invalidImageTest() {
        launchActivity(mockServer.url("text").toString())
        mockServer.takeRequest()
        activity.activity.isFinishing
        with(activity.activity) {
            assertEquals(1, mockServer.requestCount, "One http request expected")
            assertEquals(
                FabStates.ERROR,
                fabAction,
                "Text should not be a valid image format, error state expected"
            )
            assertEquals("Image format not supported", errorRef?.message, "Error message mismatch")
            assertFalse(tempFile.exists(), "Temp file should have been removed")
        }
    }

    @Test
    fun errorTest() {
        launchActivity(mockServer.url("error").toString())
        mockServer.takeRequest()
        with(activity.activity) {
            assertEquals(1, mockServer.requestCount, "One http request expected")
            assertEquals(FabStates.ERROR, fabAction, "Error response code, error state expected")
            assertEquals(
                "Unsuccessful response for image: Error mock response",
                errorRef?.message,
                "Error message mismatch"
            )
            assertFalse(tempFile.exists(), "Temp file should have been removed")
        }
    }
}