kotlin-fork/libraries/stdlib/test/collections/CollectionJVMTest.kt

package test.collections

import kotlin.test.*

import java.util.*

import org.junit.Test as test

class CollectionJVMTest {

    test fun flatMap() {
        val data = arrayListOf("", "foo", "bar", "x", "")
        val characters = data.flatMap { it.toCharList() }
        println("Got list of characters ${characters}")
        assertEquals(7, characters.size())
        val text = characters.joinToString("")
        assertEquals("foobarx", text)
    }


    test fun filterIntolinkedListOf() {
        val data = arrayListOf("foo", "bar")
        val foo = data.filterTo(linkedListOf<String>()) { it.startsWith("f") }

        assertTrue {
            foo.all { it.startsWith("f") }
        }
        assertEquals(1, foo.size)
        assertEquals(linkedListOf("foo"), foo)

        assertTrue {
            foo is LinkedList<String>
        }
    }

    test fun filterNotIntolinkedListOf() {
        val data = arrayListOf("foo", "bar")
        val foo = data.filterNotTo(linkedListOf<String>()) { it.startsWith("f") }

        assertTrue {
            foo.all { !it.startsWith("f") }
        }
        assertEquals(1, foo.size)
        assertEquals(linkedListOf("bar"), foo)

        assertTrue {
            foo is LinkedList<String>
        }
    }

    test fun filterNotNullIntoLinkedListOf() {
        val data = arrayListOf(null, "foo", null, "bar")
        val foo = data.filterNotNullTo(linkedListOf<String>())

        assertEquals(2, foo.size)
        assertEquals(linkedListOf("foo", "bar"), foo)

        assertTrue {
            foo is LinkedList<String>
        }
    }

    test fun filterIntoSortedSet() {
        val data = arrayListOf("foo", "bar")
        val sorted = data.filterTo(sortedSetOf<String>()) { it.length == 3 }
        assertEquals(2, sorted.size)
        assertEquals(sortedSetOf("bar", "foo"), sorted)
        assertTrue {
            sorted is TreeSet<String>
        }
    }

    test fun last() {
        val data = arrayListOf("foo", "bar")
        assertEquals("bar", data.last())
        assertEquals(25, arrayListOf(15, 19, 20, 25).last())
        assertEquals('a', linkedListOf('a').last())
    }

    test fun lastException() {
        fails { linkedListOf<String>().last() }
    }

    test fun contains() {
        assertTrue(linkedListOf(15, 19, 20).contains(15))
    }

    test fun sortBy() {
        expect(arrayListOf("two" to 2, "three" to 3)) {
            arrayListOf("three" to 3, "two" to 2).sortBy { it.second }
        }
        expect(arrayListOf("three" to 3, "two" to 2)) {
            arrayListOf("three" to 3, "two" to 2).sortBy { it.first }
        }
        expect(arrayListOf("two" to 2, "three" to 3)) {
            arrayListOf("three" to 3, "two" to 2).sortBy { it.first.length }
        }
    }


    test fun sortFunctionShouldReturnSortedCopyForList() {
        val list: List<Int> = arrayListOf(2, 3, 1)
        expect(arrayListOf(1, 2, 3)) { list.sort() }
        expect(arrayListOf(2, 3, 1)) { list }
    }

    test fun sortFunctionShouldReturnSortedCopyForIterable() {
        val list: Iterable<Int> = arrayListOf(2, 3, 1)
        expect(arrayListOf(1, 2, 3)) { list.sort() }
        expect(arrayListOf(2, 3, 1)) { list }
    }
}