65 lines
2.4 KiB
Kotlin
65 lines
2.4 KiB
Kotlin
/* 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.helpers
|
|
|
|
import android.content.Context
|
|
import java.io.IOException
|
|
|
|
import android.net.Uri
|
|
import android.os.Handler
|
|
import android.os.Looper
|
|
import androidx.test.core.app.ApplicationProvider
|
|
import okhttp3.mockwebserver.Dispatcher
|
|
import okhttp3.mockwebserver.MockResponse
|
|
import okhttp3.mockwebserver.MockWebServer
|
|
import okhttp3.mockwebserver.RecordedRequest
|
|
import org.mozilla.fenix.helpers.ext.toUri
|
|
|
|
object MockWebServerHelper {
|
|
|
|
fun initMockWebServerAndReturnEndpoints(vararg messages: String): List<Uri> {
|
|
val mockServer = MockWebServer()
|
|
var uniquePath = 0
|
|
val uris = mutableListOf<Uri>()
|
|
messages.forEach { message ->
|
|
val response = MockResponse().setBody("<html><body>$message</body></html>")
|
|
mockServer.enqueue(response)
|
|
val endpoint = mockServer.url(uniquePath++.toString()).toString().toUri()!!
|
|
uris += endpoint
|
|
}
|
|
return uris
|
|
}
|
|
}
|
|
|
|
/**
|
|
* A [MockWebServer] [Dispatcher] that will return Android assets in the body of requests.
|
|
*
|
|
* If the dispatcher is unable to read a requested asset, it will fail the test by throwing an
|
|
* Exception on the main thread.
|
|
*
|
|
* @sample [org.mozilla.tv.firefox.ui.BasicNavigationTest.basicNavigationTest]
|
|
*/
|
|
const val HTTP_OK = 200
|
|
const val HTTP_NOT_FOUND = 404
|
|
|
|
class AndroidAssetDispatcher : Dispatcher() {
|
|
private val mainThreadHandler = Handler(Looper.getMainLooper())
|
|
|
|
override fun dispatch(request: RecordedRequest): MockResponse {
|
|
val assetManager = ApplicationProvider.getApplicationContext<Context>().assets
|
|
val assetContents = try {
|
|
val pathNoLeadingSlash = request.path.drop(1)
|
|
assetManager.open(pathNoLeadingSlash).use { inputStream ->
|
|
inputStream.bufferedReader().use { it.readText() }
|
|
}
|
|
} catch (e: IOException) { // e.g. file not found.
|
|
// We're on a background thread so we need to forward the exception to the main thread.
|
|
mainThreadHandler.postAtFrontOfQueue { throw e }
|
|
return MockResponse().setResponseCode(HTTP_NOT_FOUND)
|
|
}
|
|
return MockResponse().setResponseCode(HTTP_OK).setBody(assetContents)
|
|
}
|
|
}
|