Skip to content
Snippets Groups Projects
testing.go 1.86 KiB
Newer Older
  • Learn to ignore specific revisions
  • Rob Pike's avatar
    Rob Pike committed
    // Copyright 2009 The Go Authors. All rights reserved.
    // Use of this source code is governed by a BSD-style
    // license that can be found in the LICENSE file.
    
    package testing
    
    
    Russ Cox's avatar
    Russ Cox committed
    import (
    
    	"fmt";
    	"flag";
    
    var chatty = flag.Bool("chatty", false, "chatty")
    
    Russ Cox's avatar
    Russ Cox committed
    
    
    // Insert tabs after newlines - but not the last one
    
    Rob Pike's avatar
    Rob Pike committed
    func tabify(s string) string {
    
    	for i := 0; i < len(s) - 1; i++ {	// -1 because if last char is newline, don't bother
    		if s[i] == '\n' {
    
    Rob Pike's avatar
    Rob Pike committed
    			return s[0:i+1] + "\t" + tabify(s[i+1:len(s)]);
    
    export type T struct {
    	errors	string;
    	failed	bool;
    
    }
    
    func (t *T) Fail() {
    	t.failed = true
    }
    
    func (t *T) FailNow() {
    	t.Fail();
    	t.ch <- t;
    	sys.goexit();
    }
    
    func (t *T) Log(args ...) {
    
    Rob Pike's avatar
    Rob Pike committed
    	t.errors += "\t" + tabify(fmt.Sprintln(args));
    
    }
    
    func (t *T) Logf(format string, args ...) {
    
    Rob Pike's avatar
    Rob Pike committed
    	t.errors += tabify(fmt.Sprintf("\t" + format, args));
    
    	l := len(t.errors);
    	if l > 0 && t.errors[l-1] != '\n' {
    		t.errors += "\n"
    	}
    }
    
    func (t *T) Error(args ...) {
    	t.Log(args);
    	t.Fail();
    }
    
    func (t *T) Errorf(format string, args ...) {
    	t.Logf(format, args);
    	t.Fail();
    }
    
    func (t *T) Fatal(args ...) {
    	t.Log(args);
    	t.FailNow();
    }
    
    func (t *T) Fatalf(format string, args ...) {
    	t.Logf(format, args);
    	t.FailNow();
    }
    
    
    Rob Pike's avatar
    Rob Pike committed
    export type Test struct {
    	name string;
    
    Rob Pike's avatar
    Rob Pike committed
    func tRunner(t *T, test *Test) {
    
    	test.f(t);
    	t.ch <- t;
    
    Rob Pike's avatar
    Rob Pike committed
    }
    
    
    Russ Cox's avatar
    Russ Cox committed
    export func Main(tests []Test) {
    
    Russ Cox's avatar
    Russ Cox committed
    	flag.Parse();
    
    Rob Pike's avatar
    Rob Pike committed
    	ok := true;
    
    	if len(tests) == 0 {
    
    		println("testing: warning: no tests to run");
    
    Rob Pike's avatar
    Rob Pike committed
    	for i := 0; i < len(tests); i++ {
    
    Russ Cox's avatar
    Russ Cox committed
    			println("=== RUN ", tests[i].name);
    		}
    
    Russ Cox's avatar
    Russ Cox committed
    		t := new(T);
    		t.ch = make(chan *T);
    
    Rob Pike's avatar
    Rob Pike committed
    		go tRunner(t, &tests[i]);
    
    		<-t.ch;
    		if t.failed {
    			println("--- FAIL:", tests[i].name);
    			print(t.errors);
    
    Russ Cox's avatar
    Russ Cox committed
    			ok = false;
    
    			println("--- PASS:", tests[i].name);
    			print(t.errors);
    
    Rob Pike's avatar
    Rob Pike committed
    		}
    	}
    	if !ok {
    
    		println("FAIL");
    
    Rob Pike's avatar
    Rob Pike committed
    		sys.exit(1);
    	}
    
    Russ Cox's avatar
    Russ Cox committed
    	println("PASS");
    
    Rob Pike's avatar
    Rob Pike committed
    }