calc - unit-tests

main_test.go

func TestAdd(t *testing.T) {
	tests := []struct {
		a, b, expected int
	}{
		{0, 0, 0},
		{1, 1, 2},
		{-2, 1, -1},
		{-1, -10, -11},
	}

	for i := range tests {
		test := &tests[i]
		got := add(test.a, test.b)
		if test.expected != got {
			t.Fatalf("expected %d but got %d", test.expected, got)
		}
	}
}

func TestMain(m *testing.M) {
	cover.ParseAndStripTestFlags()

	// Make sure we have the opportunity to flush the coverage report to disk when
	// terminating the process.
	exit.AtExit(cover.FlushProfiles)

	// If the test binary name is "calc", we've are being asked to run the
	// coverage-instrumented program.
	if path.Base(os.Args[0]) == "calc" {
		main()
		exit.Exit(0)
	}

	os.Exit(m.Run())
}