AquaStream/app/src/test/java/com/lagradost/cloudstream3/ProviderTests.kt

package com.lagradost.cloudstream3

import com.lagradost.cloudstream3.movieproviders.ThenosProvider
import com.lagradost.cloudstream3.mvvm.logError
import com.lagradost.cloudstream3.utils.Qualities
import com.lagradost.cloudstream3.utils.SubtitleHelper
import org.junit.Assert
import org.junit.Test

class ProviderTests {
    private fun getAllProviders(): List<MainAPI> {
        val allApis = APIHolder.apis
        allApis.addAll(APIHolder.restrictedApis)
        return allApis
    }

    private fun loadLinks(api: MainAPI, url: String?): Boolean {
        Assert.assertNotNull("Api ${api.name} has invalid url on episode", url)
        if (url == null) return true
        var linksLoaded = 0
        try {
            val success = api.loadLinks(url, false, {}) { link ->
                Assert.assertTrue(
                    "Api ${api.name} returns link with invalid Quality",
                    Qualities.values().map { it.value }.contains(link.quality)
                )
                Assert.assertTrue("Api ${api.name} returns link with invalid url", link.url.length > 4)
                linksLoaded++
            }
            if (success) {
                return linksLoaded > 0
            }
            Assert.assertTrue("Api ${api.name} has returns false on .loadLinks", success)
        } catch (e: Exception) {
            if (e.cause is NotImplementedError) {
                Assert.fail("Provider has not implemented .loadLinks")
            }
            logError(e)
        }
        return true
    }

    private fun testSingleProviderApi(api: MainAPI): Boolean {
        val searchQueries = listOf("over", "iron", "guy")
        var correctResponses = 0
        var searchResult: List<SearchResponse>? = null
        for (query in searchQueries) {
            val response = try {
                api.search(query)
            } catch (e: Exception) {
                if (e.cause is NotImplementedError) {
                    Assert.fail("Provider has not implemented .search")
                }
                logError(e)
                null
            }
            if (!response.isNullOrEmpty()) {
                correctResponses++
                if (searchResult == null) {
                    searchResult = response
                }
            }
        }

        if (correctResponses == 0 || searchResult == null) {
            System.err.println("Api ${api.name} did not return any valid search responses")
            return false
        }

        try {
            var validResults = false
            for (result in searchResult) {
                Assert.assertEquals("Invalid apiName on response on ${api.name}", result.apiName, api.name)
                val load = api.load(result.url) ?: continue
                Assert.assertEquals("Invalid apiName on load on ${api.name}", load.apiName, result.apiName)
                Assert.assertTrue(
                    "Api ${api.name} on load does not contain any of the supportedTypes",
                    api.supportedTypes.contains(load.type)
                )
                when (load) {
                    is AnimeLoadResponse -> {
                        val gotNoEpisodes =
                            load.dubEpisodes.isNullOrEmpty() && load.subEpisodes.isNullOrEmpty()
                        if (gotNoEpisodes) {
                            println("Api ${api.name} got no episodes on ${load.url}")
                            continue
                        }

                        val url = (load.dubEpisodes ?: load.subEpisodes)?.first()?.url
                        validResults = loadLinks(api, url)
                        if (!validResults) continue
                    }
                    is MovieLoadResponse -> {
                        val gotNoEpisodes = load.dataUrl.isBlank()
                        if (gotNoEpisodes) {
                            println("Api ${api.name} got no movie on ${load.url}")
                            continue
                        }

                        validResults = loadLinks(api, load.dataUrl)
                        if (!validResults) continue
                    }
                    is TvSeriesLoadResponse -> {
                        val gotNoEpisodes = load.episodes.isEmpty()
                        if (gotNoEpisodes) {
                            println("Api ${api.name} got no episodes on ${load.url}")
                            continue
                        }

                        validResults = loadLinks(api, load.episodes.first().data)
                        if (!validResults) continue
                    }
                }
                break
            }

            Assert.assertTrue("Api ${api.name} did not load on any}", validResults)
        } catch (e: Exception) {
            if (e.cause is NotImplementedError) {
                Assert.fail("Provider has not implemented .load")
            }
            logError(e)
            return false
        }
        return true
    }

    @Test
    fun providersExist() {
        Assert.assertTrue(getAllProviders().isNotEmpty())
    }

    @Test
    fun providerCorrectData() {
        val isoNames = SubtitleHelper.languages.map { it.ISO_639_1 }
        Assert.assertFalse("ISO does not contain any languages", isoNames.isNullOrEmpty())
        for (api in getAllProviders()) {
            Assert.assertTrue("Api does not contain a mainUrl", api.mainUrl != "NONE")
            Assert.assertTrue("Api does not contain a name", api.name != "NONE")
            Assert.assertTrue("Api ${api.name} does not contain a valid language code", isoNames.contains(api.lang))
            Assert.assertTrue("Api ${api.name} does not contain any supported types", api.supportedTypes.isNotEmpty())
        }
    }

    @Test
    fun providerCorrectHomepage() {
        getAllProviders().pmap { api ->
            if (api.hasMainPage) {
                try {
                    val homepage = api.getMainPage()
                    when {
                        homepage == null -> {
                            Assert.fail("Homepage provider ${api.name} did not correctly load homepage!")
                        }
                        homepage.items.isEmpty() -> {
                            Assert.fail("Homepage provider ${api.name} does not contain any items!")
                        }
                        homepage.items.any { it.list.isEmpty() } -> {
                            Assert.fail("Homepage provider ${api.name} does not have any items on result!")
                        }
                    }
                } catch (e: Exception) {
                    if (e.cause is NotImplementedError) {
                        Assert.fail("Provider marked as hasMainPage, while in reality is has not been implemented")
                    }
                    logError(e)
                }
            }
        }
    }

    @Test
    fun testSingleProvider() {
        testSingleProviderApi(ThenosProvider())
    }

    @Test
    fun providerCorrect() {
        val providers = getAllProviders()
        providers.pmap { api ->
            try {
                println("Trying $api")
                if (testSingleProviderApi(api)) {
                    println("Success $api")
                } else {
                    System.err.println("Error $api")
                }
            } catch (e: Exception) {
                logError(e)
            }
        }
    }
}