Files
kotlin-fork/libraries/stdlib/jvm/test/collections/ListBuilderTest.kt
T
2021-09-30 16:01:57 +00:00

76 lines
3.0 KiB
Kotlin

/*
* Copyright 2010-2021 JetBrains s.r.o. and Kotlin Programming Language contributors.
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
*/
package test.collections
import kotlin.collections.builders.*
import kotlin.test.*
@Suppress("INVISIBLE_MEMBER")
class ListBuilderTest {
@Test
@Suppress("PLATFORM_CLASS_MAPPED_TO_KOTLIN", "UNCHECKED_CAST")
fun toArray() {
val numberOfElements = 5
val expected = ArrayList<Int>().apply { addAll(0 until numberOfElements) }
val builder = ListBuilder<Int>().apply { addAll(0 until numberOfElements) }
fun testToArray(getter: List<Int>.() -> Array<Any?>) {
assertTrue(expected.getter() contentEquals builder.getter())
assertTrue(expected.subList(2, 4).getter() contentEquals builder.subList(2, 4).getter())
}
testToArray { (this as java.util.Collection<Int>).toArray() }
}
@Test
@Suppress("PLATFORM_CLASS_MAPPED_TO_KOTLIN", "UNCHECKED_CAST")
fun toArrayDestination() {
val numberOfElements = 5
val expected = ArrayList<Int>().apply { addAll(0 until numberOfElements) }
val builder = ListBuilder<Int>().apply { addAll(0 until numberOfElements) }
fun testToArray(destSize: Int, getter: List<Int>.(Array<Int>) -> Array<Int>) {
repeat(2) { index ->
val expectedDest = Array(destSize) { -it - 1 }
val builderDest = Array(destSize) { -it - 1 }
val takeSubList = index == 1
val expectedResult = (if (!takeSubList) expected else expected.subList(2, 4)).getter(expectedDest)
val builderResult = (if (!takeSubList) builder else builder.subList(2, 4)).getter(builderDest)
if (expectedResult.size <= expectedDest.size) {
assertSame(expectedDest, expectedResult)
assertSame(builderDest, builderResult)
}
assertTrue(expectedResult contentEquals builderResult)
}
}
testToArray(0) { (this as java.util.Collection<Int>).toArray(it) }
testToArray(numberOfElements - 1) { (this as java.util.Collection<Int>).toArray(it) }
testToArray(numberOfElements) { (this as java.util.Collection<Int>).toArray(it) }
testToArray(numberOfElements + 1) { (this as java.util.Collection<Int>).toArray(it) }
testToArray(numberOfElements + 2) { (this as java.util.Collection<Int>).toArray(it) }
}
@Test
fun capacityOverflow() {
val builderSize = 15
val giantListSize = Int.MAX_VALUE - builderSize + 1
val giantList = object : AbstractList<String>() {
override val size: Int get() = giantListSize
override fun get(index: Int): String = "element"
}
buildList {
repeat(builderSize) { add("element") }
assertFails { addAll(giantList) }
assertEquals(builderSize, size)
}
}
}