-
Notifications
You must be signed in to change notification settings - Fork 1k
/
WebSocketDeflateTest.kt
75 lines (60 loc) · 2.25 KB
/
WebSocketDeflateTest.kt
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
/*
* Copyright 2014-2021 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
*/
package io.ktor.tests.websocket
import io.ktor.websocket.*
import io.ktor.websocket.internals.*
import java.util.zip.*
import kotlin.random.*
import kotlin.test.*
class WebSocketDeflateTest {
private val deflater = Deflater(Deflater.DEFAULT_COMPRESSION, true)
private val inflater = Inflater(true)
private val config = WebSocketDeflateExtension.Config()
private val extension = WebSocketDeflateExtension(config)
@Test
fun testDeflateInflateEmpty() {
val data = byteArrayOf()
val deflated = deflater.deflateFully(data)
val inflated = inflater.inflateFully(deflated)
assertTrue { data.contentEquals(inflated) }
}
@Test
fun testDeflateInflateForRandomData() {
repeat(1000) {
val data = Random.nextBytes(it * 10)
val deflated = deflater.deflateFully(data)
val inflated = inflater.inflateFully(deflated)
assertTrue {
data.contentEquals(inflated)
}
}
}
@Test
fun testClientAcceptsServerNoContextTakeover() {
val negotiatedProtocols = listOf(
WebSocketExtensionHeader("permessage-deflate", listOf("server_no_context_takeover"))
)
extension.clientNegotiation(negotiatedProtocols)
assertEquals(extension.incomingNoContextTakeover, true)
assertEquals(extension.outgoingNoContextTakeover, false)
}
@Test
fun testClientAcceptsClientNoContextTakeover() {
val negotiatedProtocols = listOf(
WebSocketExtensionHeader("permessage-deflate", listOf("client_no_context_takeover"))
)
extension.clientNegotiation(negotiatedProtocols)
assertEquals(extension.incomingNoContextTakeover, false)
assertEquals(extension.outgoingNoContextTakeover, true)
}
@Test
fun testManualConfig() {
val config = WebSocketDeflateExtension.Config()
config.manualConfig(mutableListOf())
config.configureProtocols {
it.add(WebSocketExtensionHeader("permessage-deflate", listOf("client_no_context_takeover")))
}
config.manualConfig(mutableListOf())
}
}