chibi-scheme/tests/io-tests.scm

143 lines
4.4 KiB
Scheme

(cond-expand
(modules
(import (chibi io)
(only (scheme base) read-bytevector write-bytevector)
(only (chibi test) test-begin test test-end)))
(else #f))
(test-begin "io")
(define long-string (make-string 2000 #\a))
(test "input-string-port" 1025
(call-with-input-string (substring long-string 0 1025)
(lambda (in)
(let loop ((c (read-char in)) (i 0))
(cond ((eof-object? c) i)
((> i 1025) (error "read past eof"))
(else (loop (read-char in) (+ i 1))))))))
(test "read-line" '("abc" "def")
(call-with-input-string "abc\ndef\n"
(lambda (in) (let ((line (read-line in))) (list line (read-line in))))))
(test "read-line" '("abc" "def" "ghi")
(call-with-input-string "abcdef\nghi\n"
(lambda (in)
(let* ((line1 (read-line in 3))
(line2 (read-line in 3)))
(list line1 line2 (read-line in 3))))))
(test "read-line-to-eof" '("abc" "def")
(call-with-input-string "abc\ndef"
(lambda (in) (let ((line (read-line in))) (list line (read-line in))))))
(test "read-string" '("abc" "def")
(call-with-input-string "abcdef"
(lambda (in) (let ((str (read-string 3 in))) (list str (read-string 3 in))))))
(test "read-string-to-eof" '("abc" "de")
(call-with-input-string "abcde"
(lambda (in) (let ((str (read-string 3 in))) (list str (read-string 3 in))))))
(test "read-string!" '("abc" "def")
(call-with-input-string "abcdef"
(lambda (in)
(let* ((str1 (make-string 3))
(str2 (make-string 3)))
(read-string! str1 3 in)
(read-string! str2 3 in)
(list str1 str2)))))
(test "read-string!-to-eof" '("abc" "de ")
(call-with-input-string "abcde"
(lambda (in)
(let* ((str1 (make-string 3))
(str2 (make-string 3 #\space)))
(read-string! str1 3 in)
(read-string! str2 3 in)
(list str1 str2)))))
(test "null-output-port" #t
(let ((out (make-null-output-port)))
(write 1 out)
(close-output-port out)
#t))
(test "null-input-port" #t
(let ((in (make-null-input-port)))
(let ((res (eof-object? (read-char in))))
(close-input-port in)
res)))
(define (string-upcase str)
(list->string (map char-upcase (string->list str))))
(test "upcase-input-port" "ABC"
(call-with-input-string "abc"
(lambda (in)
(let ((in (make-filtered-input-port string-upcase in)))
(let ((res (read-line in)))
(close-input-port in)
res)))))
(test "upcase-output-port" "ABC"
(call-with-output-string
(lambda (out)
(let ((out (make-filtered-output-port string-upcase out)))
(display "abc" out)
(close-output-port out)))))
(define (strings->input-port str-ls)
(make-generated-input-port
(lambda ()
(and (pair? str-ls)
(let ((res (car str-ls)))
(set! str-ls (cdr str-ls))
res)))))
(test "abcdef" (read-line (strings->input-port '("abcdef"))))
(test "abcdef" (read-line (strings->input-port '("abc" "def"))))
(test "abcdef" (read-line (strings->input-port '("a" "b" "c" "d" "e" "f"))))
(test "日本語" (read-line (strings->input-port '("日本語"))))
(test "日本語" (read-line (strings->input-port '("日" "本" "語"))))
(test "abc"
(let ((in (strings->input-port
(list "日本語" (make-string 4087 #\-) "abc"))))
(read-string 4090 in)
(read-line in)))
(test "abc"
(let ((in (strings->input-port
(list "日本語" (make-string 4093 #\-) "abc"))))
(read-string 4096 in)
(read-line in)))
(let ((in (make-custom-binary-input-port
(let ((i 0))
(lambda (bv start end)
(do ((j start (+ j 1)))
((= j end))
(bytevector-u8-set! bv j (modulo (+ j i) 256)))
(if (> end 0)
(set! i (bytevector-u8-ref bv (- end 1))))
(- end start))))))
(test #u8(0 1 2 3) (read-bytevector 4 in))
(test #u8(4 5 6 7) (read-bytevector 4 in))
(test 7 (bytevector-u8-ref (read-bytevector 256 in) 255))
(test 6 (bytevector-u8-ref (read-bytevector 1024 in) 1022)))
(let* ((sum 0)
(out (make-custom-binary-output-port
(lambda (bv start end)
(do ((i start (+ i 1))
(x 0 (+ x (bytevector-u8-ref bv i))))
((= i end) (set! sum x)))))))
(write-bytevector #u8(0 1 2 3) out)
(flush-output out)
(test 6 sum)
(write-bytevector #u8(100) out)
(flush-output out)
(test 106 sum))
(test-end)