#lang planet chongkai/sml

(* Auxiliary functions for test cases *)

infix 1 seq
fun e1 seq e2 = e2;
fun check b = if b then "OK" else "WRONG";
fun check' f = (if f () then "OK" else "WRONG") handle _ => "EXN";

fun range (from, to) p = 
    let open Int 
    in
	(from > to) orelse (p from) andalso (range (from+1, to) p)
    end;

fun checkrange bounds = check o range bounds;

fun tst0 s s' = print (s ^ "    \t" ^ s' ^ "\n");
fun tst  s b = tst0 s (check  b);
fun tst' s f = tst0 s (check' f);

fun tstrange s bounds = (tst s) o range bounds  

(* test/textio.sml
   PS 1995-11-22, 1996-04-18
*)

(*KILL 05/11/1997 11:03. tho.:
use "auxil.sml";
*)

val _ = print "Testing TextIO...\n";

local 
    open TextIO

    fun fileSize s = 
	let val is = openIn s
	in size (inputAll is) before closeIn is end;

    fun dup 0 s = s
      | dup n s = dup (n-1) (s^s)

    val longstring = dup 5 "abcdefg"  (* was 16 but because `limit stack` = 8192 kbytes on frigg
				       * I received a SIGSEGV for stack growth failure. *)
in

val empty  = openOut "empty.dat";
val small  = openOut "small1.dat";
val medium = openOut "medium.dat";
val text   = openOut "text.dat";

val test1 = 
    tst' "test1" (fn _ => 
	   (closeOut empty;
	    fileSize "empty.dat" = 0 
	    andalso fileSize "empty.dat" = 0));
    
val test2 = 
    tst' "test2" (fn _ => 
	   (output1(small, #"+");
	    closeOut small;
	    fileSize "small1.dat" = 1
	    andalso fileSize "small1.dat" = 1));

val test3 = 
    tst' "test3" (fn _ =>
	   let val small = openOut "small2.dat"
	   in 
	       output(small, "*");
	       closeOut small;
	       fileSize "small2.dat" = 1 andalso fileSize "small2.dat" = 1
	   end);

val test4 = 
    tst' "test4" (fn _ => 
	   (output(medium, longstring);
	    closeOut medium;
	    fileSize "medium.dat" = size longstring
	    andalso fileSize "medium.dat" = size longstring))

val test5 = 
    tst' "test5" (fn _ => 
	   let val small = openAppend "small2.dat"
	   in 
	       output(small, "1");
	       closeOut small;
	       fileSize "small2.dat" = 2 andalso fileSize "small2.dat" = 2
	   end);

val test6 = 
    tst' "test6" (fn _ => 
	   (output(text, "Line 1\n");
	    output(text, "Line 2\n");
	    output(text, "Line 3");
	    closeOut text;
	    fileSize "text.dat" = 20 andalso fileSize "text.dat" = 20));
    
(* Test that stdErr is flushed immediately, that flushOut works, and
 * that print flushes stdOut.  Assumes that stdOut is *not* flushed
 * immediately: *)

val _ = 
    let fun stdo s = output(stdOut, s)
	fun stde s = output(stdErr, s)
    in
	print "Two lines of output follow:\n";
	stdo "3"; stde "1"; stdo "4"; stde "2";
	flushOut stdOut;
	stde "  <--- this should read 1234\n";
	stdo "2"; stde "1"; print "3"; stde "4"; stdo "5";
	flushOut stdOut;
	stde " <--- this should read 12345\n"
    end;

val test7a = 
    tst' "test7a" (fn _ => 
	   let val is = openIn "empty.dat"
	   in 
	       (endOfStream is 
		andalso input1 is = NONE 
		andalso endOfStream is 
		andalso input1 is = NONE)
	       before closeIn is 
	   end);

val test7b = 
    tst' "test7b" (fn _ => 
	   let val is = openIn "small1.dat"
	   in 
	       (not (endOfStream is)
		andalso input1 is = SOME #"+" 
		andalso endOfStream is
		andalso input1 is = NONE 
		andalso input1 is = NONE)
	       before closeIn is 
	   end);

val test7c = 
    tst' "test7c" (fn _ => 
	   let val is = openIn "small2.dat"
	   in 
	       (not (endOfStream is)
		andalso input1 is = SOME #"*" 
		andalso not (endOfStream is)
		andalso input1 is = SOME #"1" 
		andalso endOfStream is
		andalso input1 is = NONE 
		andalso input1 is = NONE)
	       before closeIn is 
	   end);

val test8a =
    tst' "test8a" (fn _ =>
	   let val is = openIn "empty.dat"
	   in 
	       (inputN(is, 0) = ""
		andalso inputN(is, 1) = ""
		andalso inputN(is, 100) = ""
		andalso endOfStream is)
	       before closeIn is 
	   end);

val test8b =
    tst' "test8b" (fn _ =>
	   let val is = openIn "small1.dat"
	   in 
	       (inputN(is, 0) = ""
		andalso inputN(is, 1) = "+"
		andalso inputN(is, 100) = "")
	       before closeIn is 
	   end);

val test8c =
    tst' "test8c" (fn _ =>
	   let val is = openIn "small1.dat"
	   in 
	       (inputN(is, 0) = ""
		andalso inputN(is, 100) = "+"
		andalso inputN(is, 100) = "")
	       before closeIn is 
	   end);

val test8d =
    tst' "test8d" (fn _ =>
	   let val is = openIn "small2.dat"
	   in 
	       (inputN(is, 0) = ""
		andalso inputN(is, 1) = "*"
		andalso inputN(is, 100) = "1"
		andalso inputN(is, 100) = "")
	       before closeIn is 
	   end);

val test8e =
    tst' "test8e" (fn _ =>
	   let val is = openIn "medium.dat"
	   in 
	       (inputN(is, 0) = ""
		andalso inputN(is, 15) = "abcdefgabcdefga"
		andalso inputN(is, 15) = "bcdefgabcdefgab"
		andalso inputN(is, 0) = ""
		andalso not (endOfStream is))
	       before closeIn is 
	   end);
   
val test8f =
    tst' "test8f" (fn _ =>
	   let val is = openIn "medium.dat"
	   in 
	       (inputN(is, 500000) = longstring
		andalso inputN(is, 100) = ""
		andalso endOfStream is)
	       before closeIn is 
	   end);
   
val test9a =
    tst' "test9a" (fn _ =>
	   let val is = openIn "empty.dat"
	   in 
	       (lookahead is = NONE 
		andalso input is = "" 
		andalso lookahead is = NONE
		andalso input is = "")
	       before closeIn is 
	   end);

val test9b =
    tst' "test9b" (fn _ =>
	   let val is = openIn "small1.dat"
	   in 
	       (lookahead is = SOME #"+"
		andalso input is = "+" 
		andalso input is = ""
		andalso lookahead is = NONE)
	       before closeIn is 
	   end);

val test9c =
    tst' "test9c" (fn _ =>
	   let val is = openIn "small2.dat"
	   in 
	       (lookahead is = SOME #"*"
		andalso input is = "*1" 
		andalso input is = ""
		andalso lookahead is = NONE)
	       before closeIn is 
	   end);

val test9d =
    tst' "test9d" (fn _ =>
	   let val is = openIn "small2.dat"
	   in 
	       (input is = "*1" 
		andalso input is = "")
	       before closeIn is 
	   end);

val test9e =
    tst' "test9e" (fn _ =>
	   let val is = openIn "medium.dat"
	   in 
	       lookahead is = SOME #"a"
	       andalso String.substring(input is, 0, 15) = "abcdefgabcdefga"
	       before closeIn is 
	   end);
   
val test10 = 
    tst' "test10" (fn _ =>
	   let val is = openIn "medium.dat"
	   in 
	       (lookahead is = SOME #"a"
		andalso input1 is = SOME #"a"
		andalso lookahead is = SOME #"b"
		andalso input1 is = SOME #"b"
		andalso lookahead is = SOME #"c")
	       before closeIn is 
	   end);

val test11 = 
    tst' "test11" (fn _ =>
	   let val is = openIn "medium.dat"
	   in 
	       (lookahead is = SOME #"a"
		andalso inputN(is, 5) = "abcde"
		andalso lookahead is = SOME #"f"
		andalso inputN(is, 4) = "fgab"
		andalso lookahead is = SOME #"c")
	       before closeIn is 
	   end);

val test12a =
    tst' "test12a" (fn _ => 
	   let val is = openIn "empty.dat"
	   in 
	       (inputLine is = NONE
		andalso inputLine is = NONE)
	       before closeIn is
	   end);

val test12b =
    tst' "test12b" (fn _ => 
	   let val is = openIn "small1.dat"
	   in 
	       (inputLine is = SOME "+\n"
		andalso inputLine is = NONE)
	       before closeIn is
	   end);

val test12c =
    tst' "test12c" (fn _ => 
	   let val is = openIn "text.dat"
	   in 
	       (inputLine is = SOME "Line 1\n"
	       andalso inputLine is = SOME "Line 2\n"
	       andalso inputLine is = SOME "Line 3\n"
	       andalso inputLine is = NONE)
	       before closeIn is
	   end);

val test12d =
    tst' "test12d" (fn _ => 
	   let val is = openIn "medium.dat"
	   in 
	       (inputLine is = SOME (longstring ^ "\n")
	       andalso inputLine is = NONE)
	       before closeIn is
	   end);

(* Test that outputSubstr works *)

val _ = 
    let fun stdo s i n = outputSubstr(stdOut, Substring.substring(s, i, n))
	fun stde s = output(stdErr, s)
	val abcde = "abcde"
    in
	print "Two lines of output follow:\n";
	stdo abcde 0 1; stdo abcde 1 3;
	stdo "" 0 0; stdo abcde 0 0; stdo abcde 5 0; stdo abcde 3 0;
	stdo abcde 4 1;
	flushOut stdOut;
	stde " <--- this should read abcde\n";
	stdo abcde 0 5;
	flushOut stdOut;
	stde " <--- this should read abcde\n"
    end;

end