97 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			Factor
		
	
	
			
		
		
	
	
			97 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			Factor
		
	
	
USING: accessors io io.streams.string kernel math namespaces
 | 
						|
parser sbufs sequences strings tools.test words ;
 | 
						|
IN: io.tests
 | 
						|
 | 
						|
{ f } [
 | 
						|
    "vocab:io/test/no-trailing-eol.factor" run-file
 | 
						|
    "foo" "io.tests" lookup-word
 | 
						|
] unit-test
 | 
						|
 | 
						|
! Make sure we use correct to_c_string form when writing
 | 
						|
{ } [ "\0" write ] unit-test
 | 
						|
 | 
						|
! Test default input stream protocol methods
 | 
						|
 | 
						|
TUPLE: up-to-13-reader { i fixnum initial: 0 } ;
 | 
						|
INSTANCE: up-to-13-reader input-stream
 | 
						|
 | 
						|
M: up-to-13-reader stream-element-type drop +byte+ ; inline
 | 
						|
M: up-to-13-reader stream-read1
 | 
						|
    [ dup 1 + ] change-i drop
 | 
						|
    dup 13 >= [ drop f ] when ; inline
 | 
						|
 | 
						|
{ B{ 0 1 2 } } [ 3 up-to-13-reader new stream-read ] unit-test
 | 
						|
{ B{ 0 1 2 } } [ 3 up-to-13-reader new stream-read-partial ] unit-test
 | 
						|
 | 
						|
{ B{ 0 1 2 3 4 5 6 7 8 9 10 11 12 } f }
 | 
						|
[ up-to-13-reader new [ 20 swap stream-read ] [ 20 swap stream-read ] bi ] unit-test
 | 
						|
 | 
						|
{
 | 
						|
    T{ slice f 0 8 B{ 0 1 2 3 4 5 6 7 } } t
 | 
						|
    T{ slice f 0 5 B{ 8 9 10 11 12 205 206 207 } } t
 | 
						|
    T{ slice f 0 0 B{ 8 9 10 11 12 205 206 207 } } f
 | 
						|
} [
 | 
						|
    up-to-13-reader new
 | 
						|
    [ B{ 200 201 202 203 204 205 206 207 } swap stream-read-into ]
 | 
						|
    [ B{ 200 201 202 203 204 205 206 207 } swap stream-read-into ]
 | 
						|
    [ B{ 200 201 202 203 204 205 206 207 } swap stream-read-into ] tri
 | 
						|
] unit-test
 | 
						|
 | 
						|
{
 | 
						|
    B{ 0 1 2 3 4 5 6 7 8 } 9
 | 
						|
    B{ 10 11 12 } f
 | 
						|
    f f
 | 
						|
} [
 | 
						|
    up-to-13-reader new
 | 
						|
    [ "\t" swap stream-read-until ]
 | 
						|
    [ "\t" swap stream-read-until ]
 | 
						|
    [ "\t" swap stream-read-until ] tri
 | 
						|
] unit-test
 | 
						|
 | 
						|
{ B{ 0 1 2 3 4 5 6 7 8 9 } B{ 11 12 } f } [
 | 
						|
    up-to-13-reader new
 | 
						|
    [ stream-readln ] [ stream-readln ] [ stream-readln ] tri
 | 
						|
] unit-test
 | 
						|
 | 
						|
! Test default output stream protocol methods
 | 
						|
 | 
						|
TUPLE: dumb-writer vector ;
 | 
						|
INSTANCE: dumb-writer output-stream
 | 
						|
 | 
						|
: <dumb-writer> ( -- x ) BV{ } clone dumb-writer boa ; inline
 | 
						|
 | 
						|
M: dumb-writer stream-element-type drop +byte+ ; inline
 | 
						|
M: dumb-writer stream-write1 vector>> push ; inline
 | 
						|
 | 
						|
{ BV{ 11 22 33 } } [
 | 
						|
    <dumb-writer>
 | 
						|
    [ B{ 11 22 33 } swap stream-write ]
 | 
						|
    [ vector>> ] bi
 | 
						|
] unit-test
 | 
						|
 | 
						|
{ BV{ 11 22 33 10 } } [
 | 
						|
    <dumb-writer>
 | 
						|
    [ B{ 11 22 33 } swap stream-write ]
 | 
						|
    [ stream-nl ]
 | 
						|
    [ vector>> ] tri
 | 
						|
] unit-test
 | 
						|
 | 
						|
{ SBUF" asdf" }
 | 
						|
[ "asdf" <string-reader> 4 <sbuf> [ stream-copy ] keep ] unit-test
 | 
						|
 | 
						|
{ "asdf" }
 | 
						|
[
 | 
						|
    [
 | 
						|
        [ "asdf" error-stream get stream-write ] with-error>output
 | 
						|
    ] with-string-writer
 | 
						|
] unit-test
 | 
						|
 | 
						|
{ "asdf" }
 | 
						|
[
 | 
						|
    <string-writer> [
 | 
						|
        [
 | 
						|
            [ "asdf" output-stream get stream-write ] with-output>error
 | 
						|
        ] with-error-stream
 | 
						|
    ] keep >string
 | 
						|
] unit-test
 |