io.streams.tee: adding stream-element-type

mrjbq7 · May 1, 2024 · 973c929 · 973c929
1 parent abea585
commit 973c929
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/extra/io/streams/tee/tee-tests.factor b/extra/io/streams/tee/tee-tests.factor
@@ -1,12 +1,16 @@
-USING: io io.encodings.utf8 io.files io.streams.tee kernel
-tools.test ;
+USING: arrays io io.encodings.utf8 io.files io.streams.tee
+kernel tools.test ;
 
 { t } [
     "resource:LICENSE.txt" utf8 [
         [
-            utf8
-            [ tee-to-file-writer read-contents ]
-            [ file-contents ] 2bi =
+            utf8 [
+                tee-to-file-writer
+                readln 42 read read-contents 3array
+            ]
+            [
+                [ readln 42 read read-contents 3array ] with-file-reader
+            ] 2bi =
         ] with-test-file
     ] with-file-reader
 ] unit-test
diff --git a/extra/io/streams/tee/tee.factor b/extra/io/streams/tee/tee.factor
@@ -25,6 +25,8 @@ MACRO: tee2 ( read-quot write-quot -- quot )
 
 PRIVATE>
 
+M: tee-stream stream-element-type in>> stream-element-type ;
+
 M: tee-stream stream-read1
     [ stream-read1 ] [ stream-write1 ] tee1 ;