Fixed image not loading

app/src/androidTest/java/xyz/quaver/pupil/ExampleInstrumentedTest.kt

@@ -21,15 +21,14 @@
 package xyz.quaver.pupil
 
 import android.util.Log
-import android.webkit.WebView
+import android.webkit.*
 import androidx.test.ext.junit.runners.AndroidJUnit4
 import androidx.test.platform.app.InstrumentationRegistry
-import kotlinx.coroutines.MainScope
-import kotlinx.coroutines.launch
-import kotlinx.coroutines.runBlocking
-import kotlinx.coroutines.withContext
+import kotlinx.coroutines.*
+import org.junit.Before
 import org.junit.Test
 import org.junit.runner.RunWith
+import xyz.quaver.pupil.hitomi.*
 
 /**
  * Instrumented test, which will execute on an Android device.
@@ -38,22 +37,95 @@ import org.junit.runner.RunWith
  */
 @RunWith(AndroidJUnit4::class)
 class ExampleInstrumentedTest {
-
-    @Test
-    fun useAppContext() {
-        // Context of the app under test.
+    @Before
+    fun init() {
         val appContext = InstrumentationRegistry.getInstrumentation().targetContext
 
         runBlocking {
-            MainScope().launch {
-                val webView = WebView(appContext).apply {
+            withContext(Dispatchers.Main) {
+                webView = WebView(appContext).apply {
                     settings.javaScriptEnabled = true
+
+                    addJavascriptInterface(object {
+                        @JavascriptInterface
+                        fun onResult(uid: String, result: String) {
+                            _webViewFlow.tryEmit(uid to result)
+                        }
+                    }, "Callback")
+
+                    loadDataWithBaseURL(
+                        "https://hitomi.la/",
+                        """
+                            <script src="https://ltn.hitomi.la/jquery.min.js"></script>
+                            <script src="https://ltn.hitomi.la/common.js"></script>
+                            <script src="https://ltn.hitomi.la/search.js"></script>
+                            <script src="https://ltn.hitomi.la/searchlib.js"></script>
+                            <script src="https://ltn.hitomi.la/results.js></script>
+                        """.trimIndent(),
+                        "text/html",
+                        null,
+                        null
+                    )
                 }
-                webView.evaluateJavascript("3") {
-                    Log.d("PUPILD", it)
-                }
-                Log.d("PUPILD", "SYNC?")
-            }.join()
+            }
+        }
+    }
+
+    @Test
+    fun test_getGalleryIDsFromNozomi() {
+        runBlocking {
+            val result = getGalleryIDsFromNozomi(null, "index", "all")
+
+            Log.d("PUPILD", "getGalleryIDsFromNozomi: ${result.size}")
+        }
+    }
+
+    @Test
+    fun test_getGalleryIDsForQuery() {
+        runBlocking {
+            val result = getGalleryIDsForQuery("female:crotch tattoo")
+
+            Log.d("PUPILD", "getGalleryIDsForQuery: ${result.size}")
+        }
+    }
+
+    @Test
+    fun test_getSuggestionsForQuery() {
+        runBlocking {
+            val result = getSuggestionsForQuery("fem")
+
+            Log.d("PUPILD", "getSuggestionsForQuery: ${result.size}")
+        }
+    }
+
+    @Test
+    fun test_urlFromUrlFromHash() {
+        runBlocking {
+            val galleryInfo = getGalleryInfo(2102416)
+
+            val result = galleryInfo.files.map {
+                imageUrlFromImage(2102416, it, false)
+            }
+
+            Log.d("PUPILD", result.toString())
+        }
+    }
+
+    @Test
+    fun test_getGalleryInfo() {
+        runBlocking {
+            val galleryInfo = getGalleryInfo(2102416)
+
+            Log.d("PUPILD", galleryInfo.toString())
+        }
+    }
+
+    @Test
+    fun test_getGalleryBlock() {
+        runBlocking {
+            val block = getGalleryBlock(2102731)
+
+            Log.d("PUPILD", block.toString())
         }
     }
 }