2 files changed, 63 insertions, 1 deletions
diff --git a/app/src/test/kotlin/com/pitchedapps/frost/MiscTest.kt b/app/src/test/kotlin/com/pitchedapps/frost/MiscTest.kt
index c9d27a1c..91e2149c 100644
--- a/app/src/test/kotlin/com/pitchedapps/frost/MiscTest.kt
+++ b/app/src/test/kotlin/com/pitchedapps/frost/MiscTest.kt
@@ -2,6 +2,7 @@ package com.pitchedapps.frost
 
 import com.pitchedapps.frost.injectors.CssHider
 import org.junit.Test
+import kotlin.test.assertEquals
 
 /**
  * Created by Allan Wang on 2017-06-14.
@@ -9,7 +10,12 @@ import org.junit.Test
 class MiscTest {
 
     @Test
-    fun asdf() {
+    fun headerFunction() {
         print(CssHider.HEADER.injector.function)
     }
+
+    @Test
+    fun nullPair() {
+        assertEquals(Pair<String?, Int>(null, 2), Pair<String?, Int>(null, 2))
+    }
 }
 \ No newline at end of file
diff --git a/app/src/test/kotlin/com/pitchedapps/frost/utils/JsoupCleanerTest.kt b/app/src/test/kotlin/com/pitchedapps/frost/utils/JsoupCleanerTest.kt
new file mode 100644
index 00000000..4ec09ea6
--- /dev/null
+++ b/app/src/test/kotlin/com/pitchedapps/frost/utils/JsoupCleanerTest.kt
@@ -0,0 +1,56 @@
+package com.pitchedapps.frost.utils
+
+import org.junit.Test
+import kotlin.test.assertEquals
+
+/**
+ * Created by Allan Wang on 2017-08-10.
+ */
+class JsoupCleanerTest {
+
+    val whitespaceRegex = Regex("\\s+")
+
+    fun String.cleanWhitespace() = replace("\n", "").replace(whitespaceRegex, " ").replace("> <", "><")
+
+    private fun String.assertCleanHtml(expected: String) {
+        assertEquals(expected.cleanWhitespace(), cleanHtml().cleanWhitespace())
+    }
+
+    private fun String.assertCleanJsoup(expected: String) {
+        assertEquals(expected.cleanWhitespace(), cleanJsoup().cleanWhitespace())
+    }
+
+    private fun String.assertCleanText(expected: String) {
+        assertEquals(expected.cleanWhitespace(), cleanText().cleanWhitespace())
+    }
+
+    @Test
+    fun noChange() {
+        "<a><aa> HI </aa></a>".assertCleanJsoup("<a><aa> HI </aa></a>")
+    }
+
+    @Test
+    fun basicText() {
+        """<div class="test">Hello world</div>""".assertCleanHtml("""<div class="test"></div>""")
+    }
+
+    @Test
+    fun multiLineText() {
+        """<div class="test">Hello
+        world</div>""".assertCleanHtml("""<div class="test"></div>""")
+    }
+
+    @Test
+    fun textRemoval() {
+        """<div>Hello<a>World</a></div>""".assertCleanText("<div><a></a></div>")
+    }
+
+    @Test
+    fun kau() {
+        val html = """<div class="col s12 m6"> <div id="kau" class="card medium sticky-action"> <div class="card-image waves-effect waves-block waves-light"> <img class="activator" src="images/kau.jpg"> <span class="card-title activator background-gradient">KAU</span> </div><div class="card-content"><p>An extensive collection of Kotlin Android Utils</p></div><div class="card-action"> <a href="https://github.com/AllanWang/KAU" target="_blank" class="inline-block">Github</a> <a href="https://allanwang.github.io/KAU/" target="_blank" class="inline-block">Page</a> </div><div class="card-reveal"> <span class="card-title grey-text text-darken-4">KAU<i class="material-icons right">close</i></span> <ul class="browser-default"> <li>Huge package of one line extension functions</li><li>Custom UI views</li><li>Adapter items and animators</li><li>SearchView</li><li>Custom delegates</li></ul> </div></div></div>"""
+        val expected = """<div class="col s12 m6"><div id="kau" class="card medium sticky-action"><div class="card-image waves-effect waves-block waves-light"><img class="activator" src="images/kau.jpg"></div><div class="card-action"><a href="-" target="_blank" class="inline-block"></a><a href="-" target="_blank" class="inline-block"></a></div></div></div>"""
+        html.assertCleanHtml(expected)
+    }
+
+}
+