github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-01-08 09:10:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add tests

Browse Source
This commit is contained in:
Him188 2020-03-06 15:16:33 +08:00
parent 19cc131247
commit e52a394060
1 changed files with 50 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
mirai-core-qqandroid/src/commonTest/kotlin/net.mamoe.mirai.qqandroid.io.serialization/JceInputTest.kt
View File

@ -2,12 +2,12 @@
package net.mamoe.mirai.qqandroid.io.serialization
import io.ktor.utils.io.core.Output
import io.ktor.utils.io.core.buildPacket
import io.ktor.utils.io.core.writeInt
import io.ktor.utils.io.core.*
import net.mamoe.mirai.qqandroid.io.serialization.jce.JceInput
import net.mamoe.mirai.qqandroid.io.serialization.jce.writeJceHead
import kotlin.test.Test
import kotlin.test.assertEquals
import kotlin.test.assertFailsWith
internal const val BYTE: Byte = 0
internal const val DOUBLE: Byte = 5
@ -37,31 +37,59 @@ internal const val ZERO_TYPE: Byte = 12
internal class JceInputTest {
@Test
fun testHeadStack() {
fun testHeadSkip() {
val input = JceInput(buildPacket {
writeJceHead(BYTE, 0)
writeByte(66)
writeJceHead(SHORT, 1)
writeShort(123)
writeJceHead(INT, 3)
writeInt(123456)
writeJceHead(FLOAT, 8)
writeFloat(123f)
writeJceHead(LONG, 15)
writeLong(123456789123456789L)
writeJceHead(DOUBLE, 16)
writeDouble(123456.0)
writeJceHead(BYTE, 17)
writeByte(1) // boolean
}, JceCharset.UTF8)
assertEquals(123456, input.skipToHeadAndUseIfPossibleOrFail(3) { input.readJceIntValue(it) })
assertEquals(true, input.skipToHeadAndUseIfPossibleOrFail(17) { input.readJceBooleanValue(it) })
assertFailsWith<EOFException> {
input.skipToHeadAndUseIfPossibleOrFail(18) {
error("test failed")
}
}
}
@Test
fun testReadInt() {
fun testReadPrimitive() {
val input = JceInput(buildPacket {
writeHead(INT, 0)
writeJceHead(BYTE, 0)
writeByte(66)
writeJceHead(SHORT, 1)
writeShort(123)
writeJceHead(INT, 3)
writeInt(123456)
writeJceHead(FLOAT, 8)
writeFloat(123f)
writeJceHead(LONG, 15)
writeLong(123456789123456789L)
writeJceHead(DOUBLE, 16)
writeDouble(123456.0)
writeJceHead(BYTE, 17)
writeByte(1) // boolean
}, JceCharset.UTF8)
assertEquals(123456, input.readJceIntValue(input.nextHead()))
}
@PublishedApi
internal fun Output.writeHead(type: Byte, tag: Int) {
if (tag < 15) {
writeByte(((tag shl 4) or type.toInt()).toByte())
return
}
if (tag < 256) {
writeByte((type.toInt() or 0xF0).toByte())
writeByte(tag.toByte())
return
}
error("tag is too large: $tag")
assertEquals(66, input.useHead { input.readJceByteValue(it) })
assertEquals(123, input.useHead { input.readJceShortValue(it) })
assertEquals(123456, input.useHead { input.readJceIntValue(it) })
assertEquals(123f, input.useHead { input.readJceFloatValue(it) })
assertEquals(123456789123456789, input.useHead { input.readJceLongValue(it) })
assertEquals(123456.0, input.useHead { input.readJceDoubleValue(it) })
assertEquals(true, input.useHead { input.readJceBooleanValue(it) })
}
}