2023-01-07 21:04:16 +00:00
|
|
|
# Copyright (c) 2023 Calvin Rose & contributors
|
2020-04-18 17:12:27 +00:00
|
|
|
#
|
|
|
|
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
|
|
# of this software and associated documentation files (the "Software"), to
|
|
|
|
# deal in the Software without restriction, including without limitation the
|
|
|
|
# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
|
|
|
|
# sell copies of the Software, and to permit persons to whom the Software is
|
|
|
|
# furnished to do so, subject to the following conditions:
|
|
|
|
#
|
|
|
|
# The above copyright notice and this permission notice shall be included in
|
|
|
|
# all copies or substantial portions of the Software.
|
|
|
|
#
|
|
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
|
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
|
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
|
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
|
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
|
|
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
|
|
|
# IN THE SOFTWARE.
|
|
|
|
|
|
|
|
(import ./helper :prefix "" :exit true)
|
2023-06-01 22:04:07 +00:00
|
|
|
(start-suite)
|
2020-04-18 17:12:27 +00:00
|
|
|
|
2021-01-12 00:32:56 +00:00
|
|
|
# Subprocess
|
2023-06-01 22:04:07 +00:00
|
|
|
# 5e1a8c86f
|
2021-01-12 00:32:56 +00:00
|
|
|
(def janet (dyn :executable))
|
|
|
|
|
2021-01-12 00:00:31 +00:00
|
|
|
(repeat 10
|
2021-01-16 21:11:07 +00:00
|
|
|
|
2021-01-12 00:32:56 +00:00
|
|
|
(let [p (os/spawn [janet "-e" `(print "hello")`] :p {:out :pipe})]
|
2021-01-12 00:00:31 +00:00
|
|
|
(os/proc-wait p)
|
2021-01-16 21:11:07 +00:00
|
|
|
(def x (:read (p :out) :all))
|
2023-06-01 22:04:07 +00:00
|
|
|
(assert (deep= "hello" (string/trim x))
|
|
|
|
"capture stdout from os/spawn pre close."))
|
2021-01-12 00:00:31 +00:00
|
|
|
|
2021-01-12 00:32:56 +00:00
|
|
|
(let [p (os/spawn [janet "-e" `(print "hello")`] :p {:out :pipe})]
|
2021-01-12 00:00:31 +00:00
|
|
|
(def x (:read (p :out) 1024))
|
|
|
|
(os/proc-wait p)
|
2023-06-01 22:04:07 +00:00
|
|
|
(assert (deep= "hello" (string/trim x))
|
|
|
|
"capture stdout from os/spawn post close."))
|
2021-01-12 00:00:31 +00:00
|
|
|
|
2023-06-01 22:04:07 +00:00
|
|
|
(let [p (os/spawn [janet "-e" `(file/read stdin :line)`] :px
|
|
|
|
{:in :pipe})]
|
2021-01-16 21:11:07 +00:00
|
|
|
(:write (p :in) "hello!\n")
|
2021-01-12 00:00:31 +00:00
|
|
|
(assert-no-error "pipe stdin to process" (os/proc-wait p))))
|
|
|
|
|
2023-06-01 22:04:07 +00:00
|
|
|
(let [p (os/spawn [janet "-e" `(print (file/read stdin :line))`] :px
|
|
|
|
{:in :pipe :out :pipe})]
|
2021-01-16 21:11:07 +00:00
|
|
|
(:write (p :in) "hello!\n")
|
|
|
|
(def x (:read (p :out) 1024))
|
|
|
|
(assert-no-error "pipe stdin to process 2" (os/proc-wait p))
|
|
|
|
(assert (= "hello!" (string/trim x)) "round trip pipeline in process"))
|
|
|
|
|
2021-07-11 09:21:55 +00:00
|
|
|
(let [p (os/spawn [janet "-e" `(do (ev/sleep 30) (os/exit 24)`] :p)]
|
|
|
|
(os/proc-kill p)
|
|
|
|
(def retval (os/proc-wait p))
|
|
|
|
(assert (not= retval 24) "Process was *not* terminated by parent"))
|
|
|
|
|
2023-05-16 11:27:52 +00:00
|
|
|
(let [p (os/spawn [janet "-e" `(do (ev/sleep 30) (os/exit 24)`] :p)]
|
2023-05-16 16:47:38 +00:00
|
|
|
(os/proc-kill p false :term)
|
2023-05-16 11:27:52 +00:00
|
|
|
(def retval (os/proc-wait p))
|
|
|
|
(assert (not= retval 24) "Process was *not* terminated by parent"))
|
|
|
|
|
2021-01-12 00:32:56 +00:00
|
|
|
# Parallel subprocesses
|
2023-06-01 22:04:07 +00:00
|
|
|
# 5e1a8c86f
|
2021-01-12 00:32:56 +00:00
|
|
|
(defn calc-1
|
|
|
|
"Run subprocess, read from stdout, then wait on subprocess."
|
|
|
|
[code]
|
2023-06-01 22:04:07 +00:00
|
|
|
(let [p (os/spawn [janet "-e" (string `(printf "%j" ` code `)`)] :px
|
|
|
|
{:out :pipe})]
|
2021-01-12 00:32:56 +00:00
|
|
|
(os/proc-wait p)
|
|
|
|
(def output (:read (p :out) :all))
|
|
|
|
(parse output)))
|
|
|
|
|
|
|
|
(assert
|
|
|
|
(deep=
|
|
|
|
(ev/gather
|
|
|
|
(calc-1 "(+ 1 2 3 4)")
|
|
|
|
(calc-1 "(+ 5 6 7 8)")
|
|
|
|
(calc-1 "(+ 9 10 11 12)"))
|
|
|
|
@[10 26 42]) "parallel subprocesses 1")
|
|
|
|
|
|
|
|
(defn calc-2
|
2023-06-01 22:04:07 +00:00
|
|
|
``
|
|
|
|
Run subprocess, wait on subprocess, then read from stdout. Read only up
|
|
|
|
to 10 bytes instead of :all
|
|
|
|
``
|
2021-01-12 00:32:56 +00:00
|
|
|
[code]
|
2023-06-01 22:04:07 +00:00
|
|
|
(let [p (os/spawn [janet "-e" (string `(printf "%j" ` code `)`)] :px
|
|
|
|
{:out :pipe})]
|
2021-01-12 00:32:56 +00:00
|
|
|
(def output (:read (p :out) 10))
|
|
|
|
(os/proc-wait p)
|
|
|
|
(parse output)))
|
|
|
|
|
|
|
|
(assert
|
|
|
|
(deep=
|
|
|
|
(ev/gather
|
|
|
|
(calc-2 "(+ 1 2 3 4)")
|
|
|
|
(calc-2 "(+ 5 6 7 8)")
|
|
|
|
(calc-2 "(+ 9 10 11 12)"))
|
|
|
|
@[10 26 42]) "parallel subprocesses 2")
|
|
|
|
|
2021-01-16 13:18:07 +00:00
|
|
|
# File piping
|
2023-06-01 22:04:07 +00:00
|
|
|
# a1cc5ca04
|
2021-01-16 13:18:07 +00:00
|
|
|
(assert-no-error "file writing 1"
|
|
|
|
(with [f (file/temp)]
|
|
|
|
(os/execute [janet "-e" `(repeat 20 (print :hello))`] :p {:out f})))
|
|
|
|
|
|
|
|
(assert-no-error "file writing 2"
|
|
|
|
(with [f (file/open "unique.txt" :w)]
|
|
|
|
(os/execute [janet "-e" `(repeat 20 (print :hello))`] :p {:out f})
|
|
|
|
(file/flush f)))
|
|
|
|
|
2023-06-01 22:04:07 +00:00
|
|
|
# Issue #593
|
|
|
|
# a1cc5ca04
|
|
|
|
(assert-no-error "file writing 3"
|
|
|
|
(def outfile (file/open "unique.txt" :w))
|
|
|
|
(os/execute [janet "-e" "(pp (seq [i :range (1 10)] i))"] :p
|
|
|
|
{:out outfile})
|
|
|
|
(file/flush outfile)
|
|
|
|
(file/close outfile)
|
|
|
|
(os/rm "unique.txt"))
|
2023-05-24 14:39:49 +00:00
|
|
|
|
2023-06-01 22:04:07 +00:00
|
|
|
# each-line iterator
|
|
|
|
# 70f13f1
|
2023-05-24 14:39:49 +00:00
|
|
|
(assert-no-error "file/lines iterator"
|
|
|
|
(def outstream (os/open "unique.txt" :wct))
|
|
|
|
(def buf1 "123\n456\n")
|
|
|
|
(defer (:close outstream)
|
|
|
|
(:write outstream buf1))
|
|
|
|
(var buf2 "")
|
2023-05-26 15:50:26 +00:00
|
|
|
(with [f (file/open "unique.txt" :r)]
|
|
|
|
(each line (file/lines f)
|
|
|
|
(set buf2 (string buf2 line))))
|
2023-05-24 14:39:49 +00:00
|
|
|
(assert (= buf1 buf2) "file/lines iterator")
|
|
|
|
(os/rm "unique.txt"))
|
|
|
|
|
2021-07-24 08:30:00 +00:00
|
|
|
# Ensure that the stream created by os/open works
|
2023-06-01 22:04:07 +00:00
|
|
|
# e8a86013d
|
2021-07-24 08:30:00 +00:00
|
|
|
(assert-no-error "File writing 4.1"
|
|
|
|
(def outstream (os/open "unique.txt" :wct))
|
|
|
|
(defer (:close outstream)
|
|
|
|
(:write outstream "123\n")
|
|
|
|
(:write outstream "456\n"))
|
|
|
|
# Cast to string to enable comparison
|
2023-06-01 22:04:07 +00:00
|
|
|
(assert (= "123\n456\n" (string (slurp "unique.txt")))
|
|
|
|
"File writing 4.2")
|
2021-07-24 08:30:00 +00:00
|
|
|
(os/rm "unique.txt"))
|
2022-01-07 00:35:30 +00:00
|
|
|
|
|
|
|
# Test that the stream created by os/open can be read from
|
2023-06-01 22:04:07 +00:00
|
|
|
# 8d8a6534e
|
2022-01-07 02:52:21 +00:00
|
|
|
(comment
|
|
|
|
(assert-no-error "File reading 1.1"
|
|
|
|
(def outstream (os/open "unique.txt" :wct))
|
|
|
|
(defer (:close outstream)
|
|
|
|
(:write outstream "123\n")
|
|
|
|
(:write outstream "456\n"))
|
2022-01-21 03:58:22 +00:00
|
|
|
|
2022-01-07 02:52:21 +00:00
|
|
|
(def outstream (os/open "unique.txt" :r))
|
|
|
|
(defer (:close outstream)
|
2023-06-01 22:04:07 +00:00
|
|
|
(assert (= "123\n456\n" (string (:read outstream :all)))
|
|
|
|
"File reading 1.2"))
|
2022-01-07 02:52:21 +00:00
|
|
|
(os/rm "unique.txt")))
|
2022-01-21 03:58:22 +00:00
|
|
|
|
2023-06-01 22:04:07 +00:00
|
|
|
# ev/gather
|
|
|
|
# 4f2d1cdc0
|
2021-01-13 03:35:28 +00:00
|
|
|
(assert (deep= @[1 2 3] (ev/gather 1 2 3)) "ev/gather 1")
|
|
|
|
(assert (deep= @[] (ev/gather)) "ev/gather 2")
|
|
|
|
(assert-error "ev/gather 3" (ev/gather 1 2 (error 3)))
|
|
|
|
|
2023-06-04 13:59:10 +00:00
|
|
|
(var cancel-counter 0)
|
|
|
|
(assert-error "ev/gather 4.1" (ev/gather
|
|
|
|
(defer (++ cancel-counter) (ev/take (ev/chan)))
|
|
|
|
(defer (++ cancel-counter) (ev/take (ev/chan)))
|
|
|
|
(error :oops)))
|
|
|
|
(assert (= cancel-counter 2) "ev/gather 4.2")
|
|
|
|
|
2020-04-18 17:12:27 +00:00
|
|
|
# Net testing
|
2023-06-01 22:04:07 +00:00
|
|
|
# 2904c19ed
|
2021-01-12 00:32:56 +00:00
|
|
|
(repeat 10
|
2021-01-12 00:00:31 +00:00
|
|
|
|
|
|
|
(defn handler
|
|
|
|
"Simple handler for connections."
|
|
|
|
[stream]
|
|
|
|
(defer (:close stream)
|
|
|
|
(def id (gensym))
|
|
|
|
(def b @"")
|
|
|
|
(net/read stream 1024 b)
|
|
|
|
(net/write stream b)
|
|
|
|
(buffer/clear b)))
|
2020-04-18 17:12:27 +00:00
|
|
|
|
2021-01-12 00:00:31 +00:00
|
|
|
(def s (net/server "127.0.0.1" "8000" handler))
|
|
|
|
(assert s "made server 1")
|
2020-04-18 17:12:27 +00:00
|
|
|
|
2021-01-12 00:00:31 +00:00
|
|
|
(defn test-echo [msg]
|
|
|
|
(with [conn (net/connect "127.0.0.1" "8000")]
|
|
|
|
(net/write conn msg)
|
|
|
|
(def res (net/read conn 1024))
|
|
|
|
(assert (= (string res) msg) (string "echo " msg))))
|
2020-04-18 17:12:27 +00:00
|
|
|
|
2021-01-12 00:00:31 +00:00
|
|
|
(test-echo "hello")
|
|
|
|
(test-echo "world")
|
|
|
|
(test-echo (string/repeat "abcd" 200))
|
2020-04-18 17:12:27 +00:00
|
|
|
|
2021-01-12 00:00:31 +00:00
|
|
|
(:close s))
|
2020-04-18 17:12:27 +00:00
|
|
|
|
2021-10-30 15:56:40 +00:00
|
|
|
# Test on both server and client
|
2023-06-01 22:04:07 +00:00
|
|
|
# 504411e
|
2021-10-30 15:56:40 +00:00
|
|
|
(defn names-handler
|
|
|
|
[stream]
|
|
|
|
(defer (:close stream)
|
2022-06-19 15:01:10 +00:00
|
|
|
# prevent immediate close
|
|
|
|
(ev/read stream 1)
|
|
|
|
(def [host port] (net/localname stream))
|
|
|
|
(assert (= host "127.0.0.1") "localname host server")
|
|
|
|
(assert (= port 8000) "localname port server")))
|
2021-10-30 15:56:40 +00:00
|
|
|
|
2021-09-07 12:12:43 +00:00
|
|
|
# Test localname and peername
|
2023-06-01 22:04:07 +00:00
|
|
|
# 077bf5eba
|
2022-06-19 14:31:43 +00:00
|
|
|
(repeat 10
|
2021-09-28 17:05:28 +00:00
|
|
|
(with [s (net/server "127.0.0.1" "8000" names-handler)]
|
2022-06-19 14:31:43 +00:00
|
|
|
(repeat 10
|
2021-09-28 17:05:28 +00:00
|
|
|
(with [conn (net/connect "127.0.0.1" "8000")]
|
2022-06-19 15:01:10 +00:00
|
|
|
(def [host port] (net/peername conn))
|
|
|
|
(assert (= host "127.0.0.1") "peername host client ")
|
|
|
|
(assert (= port 8000) "peername port client")
|
|
|
|
# let server close
|
|
|
|
(ev/write conn " "))))
|
2021-09-28 17:05:28 +00:00
|
|
|
(gccollect))
|
2021-09-07 12:12:43 +00:00
|
|
|
|
2020-11-14 20:28:10 +00:00
|
|
|
# Create pipe
|
2023-06-01 22:04:07 +00:00
|
|
|
# 12f09ad2d
|
2020-11-14 20:28:10 +00:00
|
|
|
(var pipe-counter 0)
|
2020-11-15 16:12:10 +00:00
|
|
|
(def chan (ev/chan 10))
|
|
|
|
(let [[reader writer] (os/pipe)]
|
|
|
|
(ev/spawn
|
|
|
|
(while (ev/read reader 3)
|
|
|
|
(++ pipe-counter))
|
|
|
|
(assert (= 20 pipe-counter) "ev/pipe 1")
|
|
|
|
(ev/give chan 1))
|
2020-11-14 20:28:10 +00:00
|
|
|
|
2020-11-15 16:12:10 +00:00
|
|
|
(for i 0 10
|
|
|
|
(ev/write writer "xxx---"))
|
2020-11-14 20:28:10 +00:00
|
|
|
|
2020-11-15 16:12:10 +00:00
|
|
|
(ev/close writer)
|
|
|
|
(ev/take chan))
|
2020-11-14 20:28:10 +00:00
|
|
|
|
2023-06-01 22:04:07 +00:00
|
|
|
# cff52ded5
|
2020-11-17 00:46:59 +00:00
|
|
|
(var result nil)
|
2021-01-11 21:55:12 +00:00
|
|
|
(var fiber nil)
|
|
|
|
(set fiber
|
2021-01-12 00:32:56 +00:00
|
|
|
(ev/spawn
|
2021-01-11 21:55:12 +00:00
|
|
|
(set result (protect (ev/sleep 10)))
|
2020-11-17 00:46:59 +00:00
|
|
|
(assert (= result '(false "boop")) "ev/cancel 1")))
|
2021-01-11 21:55:12 +00:00
|
|
|
(ev/sleep 0)
|
2020-11-17 00:46:59 +00:00
|
|
|
(ev/cancel fiber "boop")
|
|
|
|
|
2023-06-01 22:04:07 +00:00
|
|
|
# f0dbc2e
|
2021-01-12 00:54:50 +00:00
|
|
|
(assert (os/execute [janet "-e" `(+ 1 2 3)`] :xp) "os/execute self")
|
2021-01-11 17:10:23 +00:00
|
|
|
|
2021-07-31 00:26:42 +00:00
|
|
|
# Test some channel
|
2023-06-01 22:04:07 +00:00
|
|
|
# e76b8da26
|
2021-07-31 00:26:42 +00:00
|
|
|
(def c1 (ev/chan))
|
|
|
|
(def c2 (ev/chan))
|
|
|
|
(def arr @[])
|
|
|
|
(ev/spawn
|
|
|
|
(while (def x (ev/take c1))
|
|
|
|
(array/push arr x))
|
|
|
|
(ev/chan-close c2))
|
|
|
|
(for i 0 1000
|
|
|
|
(ev/give c1 i))
|
|
|
|
(ev/chan-close c1)
|
|
|
|
(ev/take c2)
|
|
|
|
(assert (= (slice arr) (slice (range 1000))) "ev/chan-close 1")
|
|
|
|
|
|
|
|
(def c1 (ev/chan))
|
|
|
|
(def c2 (ev/chan))
|
|
|
|
(def arr @[])
|
|
|
|
(ev/spawn
|
|
|
|
(while (def x (ev/take c1))
|
|
|
|
(array/push arr x))
|
|
|
|
(ev/sleep 0.1)
|
|
|
|
(ev/chan-close c2))
|
|
|
|
(for i 0 100
|
|
|
|
(ev/give c1 i))
|
|
|
|
(ev/chan-close c1)
|
|
|
|
(ev/select c2)
|
|
|
|
(assert (= (slice arr) (slice (range 100))) "ev/chan-close 2")
|
|
|
|
|
|
|
|
(def c1 (ev/chan))
|
|
|
|
(def c2 (ev/chan))
|
|
|
|
(def arr @[])
|
|
|
|
(ev/spawn
|
|
|
|
(while (def x (ev/take c1))
|
|
|
|
(array/push arr x))
|
|
|
|
(ev/chan-close c2))
|
|
|
|
(for i 0 100
|
|
|
|
(ev/give c1 i))
|
|
|
|
(ev/chan-close c1)
|
|
|
|
(ev/rselect c2)
|
|
|
|
(assert (= (slice arr) (slice (range 100))) "ev/chan-close 3")
|
|
|
|
|
2022-01-21 03:58:22 +00:00
|
|
|
# threaded channels
|
2023-06-01 22:04:07 +00:00
|
|
|
# 868cdb9
|
2022-01-21 03:58:22 +00:00
|
|
|
(def ch (ev/thread-chan 2))
|
|
|
|
(def att (ev/thread-chan 109))
|
|
|
|
(assert att "`att` was nil after creation")
|
|
|
|
(ev/give ch att)
|
|
|
|
(ev/do-thread
|
2023-06-01 22:04:07 +00:00
|
|
|
(assert (ev/take ch)
|
|
|
|
"channel packing bug for threaded abstracts on threaded channels."))
|
2022-01-21 03:58:22 +00:00
|
|
|
|
2022-11-10 22:32:54 +00:00
|
|
|
# marshal channels
|
2023-06-01 22:04:07 +00:00
|
|
|
# 76be8006a
|
2022-11-10 22:32:54 +00:00
|
|
|
(def ch (ev/chan 10))
|
|
|
|
(ev/give ch "hello")
|
|
|
|
(ev/give ch "world")
|
|
|
|
(def ch2 (-> ch marshal unmarshal))
|
|
|
|
(def item1 (ev/take ch2))
|
|
|
|
(def item2 (ev/take ch2))
|
|
|
|
(assert (= item1 "hello"))
|
|
|
|
(assert (= item2 "world"))
|
|
|
|
|
2023-06-07 18:56:31 +00:00
|
|
|
# ev/take, suspended, channel closed
|
|
|
|
(def ch (ev/chan))
|
|
|
|
(ev/go |(ev/chan-close ch))
|
|
|
|
(assert (= (ev/take ch) nil))
|
|
|
|
|
|
|
|
# ev/give, suspended, channel closed
|
|
|
|
(def ch (ev/chan))
|
|
|
|
(ev/go |(ev/chan-close ch))
|
|
|
|
(assert (= (ev/give ch 1) nil))
|
|
|
|
|
|
|
|
# ev/select, suspended take operation, channel closed
|
|
|
|
(def ch (ev/chan))
|
|
|
|
(ev/go |(ev/chan-close ch))
|
|
|
|
(assert (= (ev/select ch) [:close ch]))
|
|
|
|
|
|
|
|
# ev/select, suspended give operation, channel closed
|
|
|
|
(def ch (ev/chan))
|
|
|
|
(ev/go |(ev/chan-close ch))
|
|
|
|
(assert (= (ev/select [ch 1]) [:close ch]))
|
|
|
|
|
2020-04-18 17:12:27 +00:00
|
|
|
(end-suite)
|
2023-06-01 22:04:07 +00:00
|
|
|
|