aboutsummaryrefslogtreecommitdiff
path: root/app/src/test/kotlin/com/pitchedapps/frost/utils/JsoupCleanerTest.kt
blob: 4ec09ea69387d5b97923df06cfcf7b687c1789b3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
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)
    }

}