Emily Kager 5 years ago committed by Sawyer Blatz
parent 6dd829fdb4
commit 043f8829c9

@ -4,18 +4,16 @@
package org.mozilla.fenix.helpers package org.mozilla.fenix.helpers
import android.content.Context
import java.io.IOException
import android.net.Uri import android.net.Uri
import android.os.Handler import android.os.Handler
import android.os.Looper import android.os.Looper
import androidx.test.core.app.ApplicationProvider import androidx.test.platform.app.InstrumentationRegistry
import okhttp3.mockwebserver.Dispatcher import okhttp3.mockwebserver.Dispatcher
import okhttp3.mockwebserver.MockResponse import okhttp3.mockwebserver.MockResponse
import okhttp3.mockwebserver.MockWebServer import okhttp3.mockwebserver.MockWebServer
import okhttp3.mockwebserver.RecordedRequest import okhttp3.mockwebserver.RecordedRequest
import org.mozilla.fenix.helpers.ext.toUri import org.mozilla.fenix.helpers.ext.toUri
import java.io.IOException
object MockWebServerHelper { object MockWebServerHelper {
@ -48,7 +46,7 @@ class AndroidAssetDispatcher : Dispatcher() {
private val mainThreadHandler = Handler(Looper.getMainLooper()) private val mainThreadHandler = Handler(Looper.getMainLooper())
override fun dispatch(request: RecordedRequest): MockResponse { override fun dispatch(request: RecordedRequest): MockResponse {
val assetManager = ApplicationProvider.getApplicationContext<Context>().assets val assetManager = InstrumentationRegistry.getInstrumentation().context.assets
val assetContents = try { val assetContents = try {
val pathNoLeadingSlash = request.path.drop(1) val pathNoLeadingSlash = request.path.drop(1)
assetManager.open(pathNoLeadingSlash).use { inputStream -> assetManager.open(pathNoLeadingSlash).use { inputStream ->

Loading…
Cancel
Save