mirror of
https://github.com/logos-storage/nim-websock.git
synced 2026-01-04 22:53:13 +00:00
6b76bd8261
* Update http to use chronos http. * Implement TLS in websocket. * Add webscoket TLS test. * Minor nit. * Add TLS test file. * Update http to use chronos http. (#6) * Update http to use chronos http. * Add stream.nim file. * Address comments. * Fix CI failure. * Minor change. * Address comments. * Fix windows CI failing test. * minor cleanup * spacess * more idiomatic connect * use stew/base10 Co-authored-by: Dmitriy Ryajov <dryajov@gmail.com> * Implement TLS in websocket. * Minor nit. * merge master * wip * Update http to use chronos http. (#6) * Update http to use chronos http. * Add stream.nim file. * Address comments. * Fix CI failure. * Minor change. * Address comments. * Fix windows CI failing test. * minor cleanup * spacess * more idiomatic connect * use stew/base10 Co-authored-by: Dmitriy Ryajov <dryajov@gmail.com> * Update http to use chronos http. * Implement TLS in websocket. * Minor nit. * Update http to use chronos http. (#6) * Update http to use chronos http. * Add stream.nim file. * Address comments. * Fix CI failure. * Minor change. * Address comments. * Fix windows CI failing test. * minor cleanup * spacess * more idiomatic connect * use stew/base10 Co-authored-by: Dmitriy Ryajov <dryajov@gmail.com> * Implement TLS in websocket. * Minor nit. * add testing keys * wip * fix test * wip * remove eth dep and add skipdirs * fix package structure * fix deps * check nim version * Fix CI failure. * Don't call `ws.stream.closeWait()` * always close both ends to complete the sequence * misc * don't fail on close * Fix windows CI. * fix linux x86 builds * use consistent connect pattern * move keys to better place * return dumbResponse * small cleanup Co-authored-by: Dmitriy Ryajov <dryajov@gmail.com>
280 lines
6.2 KiB
Nim
280 lines
6.2 KiB
Nim
import unittest
|
|
|
|
include ../ws/ws
|
|
include ../ws/random
|
|
|
|
# TODO: Fix Test.
|
|
|
|
var maskKey: array[4, char]
|
|
|
|
suite "Test data frames":
|
|
test "# 7bit length text":
|
|
check encodeFrame(Frame(
|
|
fin: false,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Text,
|
|
mask: false,
|
|
data: toBytes("hi there"),
|
|
maskKey: maskKey
|
|
)) == toBytes("\1\8hi there")
|
|
|
|
test "# 7bit length text fin bit":
|
|
check encodeFrame(Frame(
|
|
fin: true,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Text,
|
|
mask: false,
|
|
data: toBytes("hi there"),
|
|
maskKey: maskKey
|
|
)) == toBytes("\129\8hi there")
|
|
|
|
test "# 7bit length binary":
|
|
check encodeFrame(Frame(
|
|
fin: false,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Binary,
|
|
mask: false,
|
|
data: toBytes("hi there"),
|
|
maskKey: maskKey
|
|
)) == toBytes("\2\8hi there")
|
|
|
|
test "# 7bit length binary fin bit":
|
|
check encodeFrame(Frame(
|
|
fin: true,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Binary,
|
|
mask: false,
|
|
data: toBytes("hi there"),
|
|
maskKey: maskKey
|
|
)) == toBytes("\130\8hi there")
|
|
|
|
test "# 7bit length continuation":
|
|
check encodeFrame(Frame(
|
|
fin: false,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Cont,
|
|
mask: false,
|
|
data: toBytes("hi there"),
|
|
maskKey: maskKey
|
|
)) == toBytes("\0\8hi there")
|
|
|
|
test "# 7+16 length text":
|
|
var data = ""
|
|
for i in 0..32:
|
|
data.add "How are you this is the payload!!!"
|
|
|
|
check encodeFrame(Frame(
|
|
fin: false,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Text,
|
|
mask: false,
|
|
data: toBytes(data),
|
|
maskKey: maskKey
|
|
)) == toBytes("\1\126\4\98" & data)
|
|
|
|
test "# 7+16 length text fin bit":
|
|
var data = ""
|
|
for i in 0..32:
|
|
data.add "How are you this is the payload!!!"
|
|
|
|
check encodeFrame(Frame(
|
|
fin: false,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Text,
|
|
mask: false,
|
|
data: toBytes(data),
|
|
maskKey: maskKey
|
|
)) == toBytes("\1\126\4\98" & data)
|
|
|
|
test "# 7+16 length binary":
|
|
var data = ""
|
|
for i in 0..32:
|
|
data.add "How are you this is the payload!!!"
|
|
|
|
check encodeFrame(Frame(
|
|
fin: false,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Binary,
|
|
mask: false,
|
|
data: toBytes(data),
|
|
maskKey: maskKey
|
|
)) == toBytes("\2\126\4\98" & data)
|
|
|
|
test "# 7+16 length binary fin bit":
|
|
var data = ""
|
|
for i in 0..32:
|
|
data.add "How are you this is the payload!!!"
|
|
|
|
check encodeFrame(Frame(
|
|
fin: true,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Binary,
|
|
mask: false,
|
|
data: toBytes(data),
|
|
maskKey: maskKey
|
|
)) == toBytes("\130\126\4\98" & data)
|
|
|
|
test "# 7+16 length continuation":
|
|
var data = ""
|
|
for i in 0..32:
|
|
data.add "How are you this is the payload!!!"
|
|
|
|
check encodeFrame(Frame(
|
|
fin: false,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Cont,
|
|
mask: false,
|
|
data: toBytes(data),
|
|
maskKey: maskKey
|
|
)) == toBytes("\0\126\4\98" & data)
|
|
|
|
test "# 7+64 length text":
|
|
var data = ""
|
|
for i in 0..3200:
|
|
data.add "How are you this is the payload!!!"
|
|
|
|
check encodeFrame(Frame(
|
|
fin: false,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Text,
|
|
mask: false,
|
|
data: toBytes(data),
|
|
maskKey: maskKey
|
|
)) == toBytes("\1\127\0\0\0\0\0\1\169\34" & data)
|
|
|
|
test "# 7+64 length fin bit":
|
|
var data = ""
|
|
for i in 0..3200:
|
|
data.add "How are you this is the payload!!!"
|
|
|
|
check encodeFrame(Frame(
|
|
fin: true,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Text,
|
|
mask: false,
|
|
data: toBytes(data),
|
|
maskKey: maskKey
|
|
)) == toBytes("\129\127\0\0\0\0\0\1\169\34" & data)
|
|
|
|
test "# 7+64 length binary":
|
|
var data = ""
|
|
for i in 0..3200:
|
|
data.add "How are you this is the payload!!!"
|
|
|
|
check encodeFrame(Frame(
|
|
fin: false,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Binary,
|
|
mask: false,
|
|
data: toBytes(data),
|
|
maskKey: maskKey
|
|
)) == toBytes("\2\127\0\0\0\0\0\1\169\34" & data)
|
|
|
|
test "# 7+64 length binary fin bit":
|
|
var data = ""
|
|
for i in 0..3200:
|
|
data.add "How are you this is the payload!!!"
|
|
|
|
check encodeFrame(Frame(
|
|
fin: true,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Binary,
|
|
mask: false,
|
|
data: toBytes(data),
|
|
maskKey: maskKey
|
|
)) == toBytes("\130\127\0\0\0\0\0\1\169\34" & data)
|
|
|
|
test "# 7+64 length binary":
|
|
var data = ""
|
|
for i in 0..3200:
|
|
data.add "How are you this is the payload!!!"
|
|
|
|
check encodeFrame(Frame(
|
|
fin: false,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Cont,
|
|
mask: false,
|
|
data: toBytes(data),
|
|
maskKey: maskKey
|
|
)) == toBytes("\0\127\0\0\0\0\0\1\169\34" & data)
|
|
|
|
test "# masking":
|
|
let data = encodeFrame(Frame(
|
|
fin: true,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Text,
|
|
mask: true,
|
|
data: toBytes("hi there"),
|
|
maskKey: ['\xCF', '\xD8', '\x05', 'e']
|
|
))
|
|
|
|
check data == toBytes("\129\136\207\216\5e\167\177%\17\167\189w\0")
|
|
|
|
suite "Test control frames":
|
|
|
|
test "Close":
|
|
check encodeFrame(Frame(
|
|
fin: true,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Close,
|
|
mask: false,
|
|
data: @[3'u8, 232'u8] & toBytes("hi there"),
|
|
maskKey: maskKey
|
|
)) == toBytes("\136\10\3\232hi there")
|
|
|
|
test "Ping":
|
|
check encodeFrame(Frame(
|
|
fin: false,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Ping,
|
|
mask: false,
|
|
maskKey: maskKey
|
|
)) == toBytes("\9\0")
|
|
|
|
test "Pong":
|
|
check encodeFrame(Frame(
|
|
fin: false,
|
|
rsv1: false,
|
|
rsv2: false,
|
|
rsv3: false,
|
|
opcode: Opcode.Pong,
|
|
mask: false,
|
|
maskKey: maskKey
|
|
)) == toBytes("\10\0")
|