simon/tiny_httpd
1
0
Fork 0
mirror of https://github.com/c-cube/tiny_httpd.git synced 2025-12-06 11:15:35 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: update unit tests for websockets

Browse source
This commit is contained in:
Simon Cruanes 2024-04-05 13:23:10 -04:00
parent 3393c13c00
commit 9329c95ce7
No known key found for this signature in database
GPG key ID: EBFFF6F283F3A2B4
2 changed files with 12 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/unit/ws/t_ws.ml
View file

@ -6,9 +6,9 @@ let read_file file : string =
let apply_masking = Tiny_httpd_ws.Private_.apply_masking let apply_masking = Tiny_httpd_ws.Private_.apply_masking
let decode ~key b = let decode ?(mask_offset = 0) ~key b =
let buf = Bytes.copy b in let buf = Bytes.copy b in
apply_masking ~mask_key:key buf 0 (Bytes.length buf); apply_masking ~mask_key:key ~mask_offset buf 0 (Bytes.length buf);
buf buf
let () = let () =

17
tests/unit/ws/t_ws_q.ml
View file

@ -2,9 +2,9 @@ open Test_util
let apply_masking = Tiny_httpd_ws.Private_.apply_masking let apply_masking = Tiny_httpd_ws.Private_.apply_masking
let decode ~key b = let decode ?(mask_offset = 0) ~key b =
let buf = Bytes.copy b in let buf = Bytes.copy b in
apply_masking ~mask_key:key buf 0 (Bytes.length buf); apply_masking ~mask_key:key ~mask_offset buf 0 (Bytes.length buf);
buf buf
let ppb b = Printf.sprintf "%S" (Bytes.unsafe_to_string b) let ppb b = Printf.sprintf "%S" (Bytes.unsafe_to_string b)
@ -13,12 +13,15 @@ let () =
add_qcheck add_qcheck
@@ QCheck.Test.make ~count:10_000 @@ QCheck.Test.make ~count:10_000
Q.( Q.(
pair (bytes_of_size (Gen.return 4)) (bytes_of_size Gen.(0 -- 6000)) triple
(bytes_of_size (Gen.return 4))
(option small_nat)
(bytes_of_size Gen.(0 -- 6000))
(* |> Q.add_stat ("b.size", fun (_k, b) -> Bytes.length b) *) (* |> Q.add_stat ("b.size", fun (_k, b) -> Bytes.length b) *)
|> Q.add_shrink_invariant (fun (k, _) -> Bytes.length k = 4)) |> Q.add_shrink_invariant (fun (k, _, _) -> Bytes.length k = 4))
(fun (key, b) -> (fun (key, mask_offset, b) ->
Q.assume (Bytes.length key = 4); Q.assume (Bytes.length key = 4);
let b2 = decode ~key b in let b2 = decode ~key ?mask_offset b in
let is_zero = let is_zero =
Bytes.equal key (Bytes.unsafe_of_string "\x00\x00\x00\x00") Bytes.equal key (Bytes.unsafe_of_string "\x00\x00\x00\x00")
@ -32,7 +35,7 @@ let () =
(ppb key) (ppb b) (ppb b2) (ppb key) (ppb b) (ppb b2)
); );
let b3 = decode ~key b2 in let b3 = decode ~key ?mask_offset b2 in
assert (Bytes.equal b b3); assert (Bytes.equal b b3);
true) true)