summaryrefslogtreecommitdiff
path: root/formats/json/commonTest/src/kotlinx/serialization/json/MapLikeSerializerTest.kt
blob: 37abd954bb74ce30c6f41ccee53b3a645ea37f4a (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
57
58
59
60
61
62
63
64
65
66
67
/*
 * Copyright 2017-2020 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
 */

package kotlinx.serialization.json

import kotlinx.serialization.*
import kotlinx.serialization.builtins.*
import kotlinx.serialization.descriptors.*
import kotlinx.serialization.encoding.*
import kotlin.test.*

class MapLikeSerializerTest : JsonTestBase() {

    @Serializable
    data class StringPair(val a: String, val b: String)

    @Serializer(forClass = StringPair::class)
    object StringPairSerializer : KSerializer<StringPair> {

        override val descriptor: SerialDescriptor = buildSerialDescriptor("package.StringPair", StructureKind.MAP) {
            element<String>("a")
            element<String>("b")
        }

        override fun serialize(encoder: Encoder, value: StringPair) {
            val structuredEncoder = encoder.beginStructure(descriptor)
            structuredEncoder.encodeSerializableElement(descriptor, 0, String.serializer(), value.a)
            structuredEncoder.encodeSerializableElement(descriptor, 1, String.serializer(), value.b)
            structuredEncoder.endStructure(descriptor)
        }

        override fun deserialize(decoder: Decoder): StringPair {
            val composite = decoder.beginStructure(descriptor)
            if (composite.decodeSequentially()) {
                val key = composite.decodeSerializableElement(descriptor, 0, String.serializer())
                val value = composite.decodeSerializableElement(descriptor, 1, String.serializer())
                return StringPair(key, value)
            }

            var key: String? = null
            var value: String? = null
            mainLoop@ while (true) {
                when (val idx = composite.decodeElementIndex(descriptor)) {
                    CompositeDecoder.DECODE_DONE -> {
                        break@mainLoop
                    }
                    0 -> {
                        key = composite.decodeSerializableElement(descriptor, 0, String.serializer())
                    }
                    1 -> {
                        value = composite.decodeSerializableElement(descriptor, 1, String.serializer())
                    }
                    else -> throw SerializationException("Invalid index: $idx")
                }
            }
            composite.endStructure(descriptor)
            if (key == null) throw SerializationException("Element 'a' is missing")
            if (value == null) throw SerializationException("Element 'b' is missing")
            @Suppress("UNCHECKED_CAST")
            return StringPair(key, value)
        }
    }

    @Test
    fun testStringPair() = assertJsonFormAndRestored(StringPairSerializer, StringPair("a", "b"), """{"a":"b"}""")
}