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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
package kotlinx.serialization
import kotlinx.serialization.descriptors.*
import kotlinx.serialization.encoding.CompositeDecoder
import kotlinx.serialization.internal.ElementMarker
import kotlin.test.Test
import kotlin.test.assertEquals
class ElementMarkerTest {
@Test
fun testNothingWasRead() {
val size = 5
val descriptor = createClassDescriptor(size)
val reader = ElementMarker(descriptor) { _, _ -> true }
for (i in 0 until size) {
assertEquals(i, reader.nextUnmarkedIndex())
}
assertEquals(CompositeDecoder.DECODE_DONE, reader.nextUnmarkedIndex())
}
@Test
fun testAllWasRead() {
val size = 5
val descriptor = createClassDescriptor(size)
val reader = ElementMarker(descriptor) { _, _ -> true }
for (i in 0 until size) {
reader.mark(i)
}
assertEquals(CompositeDecoder.DECODE_DONE, reader.nextUnmarkedIndex())
}
@Test
fun testFilteredRead() {
val size = 10
val readIndex = 4
val predicate: (Any?, Int) -> Boolean = { _, i -> i % 2 == 0 }
val descriptor = createClassDescriptor(size)
val reader = ElementMarker(descriptor, predicate)
reader.mark(readIndex)
for (i in 0 until size) {
if (predicate(descriptor, i) && i != readIndex) {
//`readIndex` already read and only filtered elements must be read
assertEquals(i, reader.nextUnmarkedIndex())
}
}
assertEquals(CompositeDecoder.DECODE_DONE, reader.nextUnmarkedIndex())
}
@Test
fun testSmallPartiallyRead() {
testPartiallyRead(Long.SIZE_BITS / 3)
}
@Test
fun test64PartiallyRead() {
testPartiallyRead(Long.SIZE_BITS)
}
@Test
fun test128PartiallyRead() {
testPartiallyRead(Long.SIZE_BITS * 2)
}
@Test
fun testLargePartiallyRead() {
testPartiallyRead(Long.SIZE_BITS * 2 + Long.SIZE_BITS / 3)
}
private fun testPartiallyRead(size: Int) {
val descriptor = createClassDescriptor(size)
val reader = ElementMarker(descriptor) { _, _ -> true }
for (i in 0 until size) {
if (i % 2 == 0) {
reader.mark(i)
}
}
for (i in 0 until size) {
if (i % 2 != 0) {
assertEquals(i, reader.nextUnmarkedIndex())
}
}
assertEquals(CompositeDecoder.DECODE_DONE, reader.nextUnmarkedIndex())
}
private fun createClassDescriptor(size: Int): SerialDescriptor {
return buildClassSerialDescriptor("descriptor") {
for (i in 0 until size) {
element("element$i", buildSerialDescriptor("int", PrimitiveKind.INT))
}
}
}
}
|