giles
b67709dab5
Some checks failed
Test, Build, and Deploy / test-build-deploy (push) Failing after 37s
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
95 lines
3.3 KiB
Plaintext
95 lines
3.3 KiB
Plaintext
;; lib/dream/tests/websocket.sx — upgrade, send/receive/close, broadcast.
|
|
|
|
(define dream-ws-pass 0)
|
|
(define dream-ws-fail 0)
|
|
(define dream-ws-fails (list))
|
|
|
|
(define
|
|
dream-ws-test
|
|
(fn
|
|
(name actual expected)
|
|
(if
|
|
(= actual expected)
|
|
(set! dream-ws-pass (+ dream-ws-pass 1))
|
|
(begin
|
|
(set! dream-ws-fail (+ dream-ws-fail 1))
|
|
(append! dream-ws-fails {:name name :actual actual :expected expected})))))
|
|
|
|
;; ── upgrade response ───────────────────────────────────────────────
|
|
(define dream-ws-echo (fn (ws) (dream-text "unused")))
|
|
(define
|
|
dream-ws-up
|
|
((dream-websocket dream-ws-echo) (dream-request "GET" "/ws" {} "")))
|
|
(dream-ws-test "upgrade status 101" (dream-status dream-ws-up) 101)
|
|
(dream-ws-test "is a websocket response" (dream-websocket? dream-ws-up) true)
|
|
(dream-ws-test
|
|
"plain response is not ws"
|
|
(dream-websocket? (dream-html "x"))
|
|
false)
|
|
(dream-ws-test
|
|
"upgrade header"
|
|
(dream-resp-header dream-ws-up "upgrade")
|
|
"websocket")
|
|
|
|
;; ── basic send / receive / close on a mock ─────────────────────────
|
|
(define dream-ws-w1 (dream-mock-ws (list "hi" "there")))
|
|
(dream-ws-test "open initially" (dream-ws-open? dream-ws-w1) true)
|
|
(dream-ws-test "receive first" (dream-receive dream-ws-w1) "hi")
|
|
(dream-ws-test "receive second" (dream-receive dream-ws-w1) "there")
|
|
(dream-ws-test "receive empty -> nil" (dream-receive dream-ws-w1) nil)
|
|
(dream-send dream-ws-w1 "out1")
|
|
(dream-send dream-ws-w1 "out2")
|
|
(dream-ws-test
|
|
"sent recorded"
|
|
(dream-ws-sent dream-ws-w1)
|
|
(list "out1" "out2"))
|
|
(dream-close dream-ws-w1)
|
|
(dream-ws-test "closed flag" (dream-ws-closed? dream-ws-w1) true)
|
|
(dream-ws-test "open? false after close" (dream-ws-open? dream-ws-w1) false)
|
|
|
|
;; ── echo handler driven over the upgrade response ──────────────────
|
|
(define
|
|
dream-ws-echo-h
|
|
(fn
|
|
(ws)
|
|
(let
|
|
((m (dream-receive ws)))
|
|
(if
|
|
(nil? m)
|
|
(dream-close ws)
|
|
(begin (dream-send ws (str "echo:" m)) (dream-ws-echo-h ws))))))
|
|
(define
|
|
dream-ws-echo-up
|
|
((dream-websocket dream-ws-echo-h)
|
|
(dream-request "GET" "/ws" {} "")))
|
|
(define dream-ws-echo-conn (dream-mock-ws (list "a" "b" "c")))
|
|
(dream-ws-run dream-ws-echo-up dream-ws-echo-conn)
|
|
(dream-ws-test
|
|
"echo all messages"
|
|
(dream-ws-sent dream-ws-echo-conn)
|
|
(list "echo:a" "echo:b" "echo:c"))
|
|
(dream-ws-test
|
|
"echo closes at end"
|
|
(dream-ws-closed? dream-ws-echo-conn)
|
|
true)
|
|
|
|
;; ── broadcast to a room ────────────────────────────────────────────
|
|
(define dream-ws-c1 (dream-mock-ws (list)))
|
|
(define dream-ws-c2 (dream-mock-ws (list)))
|
|
(define dream-ws-c3 (dream-mock-ws (list)))
|
|
(dream-ws-broadcast (list dream-ws-c1 dream-ws-c2 dream-ws-c3) "hello room")
|
|
(dream-ws-test
|
|
"broadcast c1"
|
|
(dream-ws-sent dream-ws-c1)
|
|
(list "hello room"))
|
|
(dream-ws-test
|
|
"broadcast c2"
|
|
(dream-ws-sent dream-ws-c2)
|
|
(list "hello room"))
|
|
(dream-ws-test
|
|
"broadcast c3"
|
|
(dream-ws-sent dream-ws-c3)
|
|
(list "hello room"))
|
|
|
|
(define dream-ws-tests-run! (fn () {:total (+ dream-ws-pass dream-ws-fail) :passed dream-ws-pass :failed dream-ws-fail :fails dream-ws-fails}))
|