/*
 * Copyright 2014-2019 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
 */

package io.ktor.client.plugins.compression

import io.ktor.client.call.*
import io.ktor.client.request.*
import io.ktor.client.tests.utils.*
import io.ktor.http.*
import kotlin.test.*

private const val TEST_URL = "$TEST_SERVER/compression"

class ContentEncodingTest : ClientLoader() {
    @Test
    fun testIdentity() = clientTests {
        config {
            ContentEncoding {
                identity()
            }
        }

        test { client ->
            val response = client.get("$TEST_URL/identity")
            // JS browser engines don't have Content-Encoding header for identity encoding
            if (response.appliedDecoders.isNotEmpty()) {
                assertEquals(listOf("identity"), response.appliedDecoders)
            }
            assertEquals("Compressed response!", response.body<String>())
        }
    }

    @Test
    fun testDeflate() = clientTests(listOf("native:CIO")) {
        config {
            ContentEncoding {
                deflate()
            }
        }

        test { client ->
            val response = client.get("$TEST_URL/deflate")
            assertNull(response.headers[HttpHeaders.ContentEncoding])
            assertEquals(listOf("deflate"), response.appliedDecoders)
            assertEquals("Compressed response!", response.body<String>())
        }
    }

    @Test
    fun testGZip() = clientTests(listOf("native:CIO")) {
        config {
            ContentEncoding {
                gzip()
            }
        }

        test { client ->
            val response = client.get("$TEST_URL/gzip")
            assertNull(response.headers[HttpHeaders.ContentEncoding])
            assertEquals(listOf("gzip"), response.appliedDecoders)
            assertEquals("Compressed response!", response.body<String>())
        }
    }

    @Test
    fun testGZipEmpty() = clientTests {
        config {
            ContentEncoding {
                gzip()
            }
        }

        test { client ->
            val response = client.get("$TEST_URL/gzip-empty")
            assertEquals("gzip", response.headers[HttpHeaders.ContentEncoding])
            assertEquals("", response.body())
        }
    }

    @Test
    fun testGzipByteArray() = clientTests {
        config {
            ContentEncoding {
                gzip()
            }
        }

        test { client ->
            val response = client.get("$TEST_URL/gzip-precompressed")
            assertEquals(HttpStatusCode.OK, response.status)
            assertEquals(listOf("gzip"), response.appliedDecoders)
            val body = response.body<ByteArray>()
            assertContentEquals(ByteArray(500) { it.toByte() }, body)
        }
    }

    @Test
    fun testDisableDecompression() = clientTests(listOf("OkHttp")) {
        config {
            ContentEncoding(mode = ContentEncodingConfig.Mode.CompressRequest) {
                gzip()
            }
        }

        test { client ->
            val response = client.get("$TEST_URL/gzip-precompressed")
            assertEquals(HttpStatusCode.OK, response.status)
            assertEquals("gzip", response.headers[HttpHeaders.ContentEncoding])
        }
    }
}
