Skip to content
Snippets Groups Projects
testing.go 1.6 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";
    
    Russ Cox's avatar
    Russ Cox committed
    )
    
    var chatty bool;
    func init() {
    	flag.Bool("chatty", false, &chatty, "chatty");
    }
    
    
    export type T struct {
    	errors	string;
    	failed	bool;
    	ch	*chan *T;
    }
    
    func (t *T) Fail() {
    	t.failed = true
    }
    
    func (t *T) FailNow() {
    	t.Fail();
    	t.ch <- t;
    	sys.goexit();
    }
    
    func (t *T) Log(args ...) {
    	t.errors += "\t" + fmt.sprintln(args);
    }
    
    func (t *T) Logf(format string, args ...) {
    	t.errors += 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;
    
    	f *(*T);
    }
    
    func TRunner(t *T, test *Test) {
    	test.f(t);
    	t.ch <- t;
    
    Rob Pike's avatar
    Rob Pike committed
    }
    
    export func Main(tests *[]Test) {
    	ok := true;
    
    	if len(tests) == 0 {
    		println("gotest: 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
    		if chatty {
    			println("=== RUN ", tests[i].name);
    		}
    
    		t := new(T);
    		t.ch = new(chan *T);
    		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;
    		} else if chatty {
    
    			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
    }