Unlock 30% off on Manual Testing Annual Plans this Holiday Season.

Copied to Clipboard!

How to use TestMain method of template Package

Best Gauge code snippet using template.TestMain

testmain.go

Source:testmain.go

1// Copyright 2013 The Go Authors. All rights reserved.2// Use of this source code is governed by a BSD-style3// license that can be found in the LICENSE file.4package ssa5// CreateTestMainPackage synthesizes a main package that runs all the6// tests of the supplied packages.7// It is closely coupled to $GOROOT/src/cmd/go/test.go and $GOROOT/src/testing.8//9// TODO(adonovan): throws this all away now that x/tools/go/packages10// provides access to the actual synthetic test main files.11import (12	"bytes"13	"fmt"14	"go/ast"15	"go/parser"16	"go/types"17	"log"18	"os"19	"strings"20	"text/template"21)22// FindTests returns the Test, Benchmark, and Example functions23// (as defined by "go test") defined in the specified package,24// and its TestMain function, if any.25//26// Deprecated: Use golang.org/x/tools/go/packages to access synthetic27// testmain packages.28func FindTests(pkg *Package) (tests, benchmarks, examples []*Function, main *Function) {29	prog := pkg.Prog30	// The first two of these may be nil: if the program doesn't import "testing",31	// it can't contain any tests, but it may yet contain Examples.32	var testSig *types.Signature                              // func(*testing.T)33	var benchmarkSig *types.Signature                         // func(*testing.B)34	var exampleSig = types.NewSignature(nil, nil, nil, false) // func()35	// Obtain the types from the parameters of testing.MainStart.36	if testingPkg := prog.ImportedPackage("testing"); testingPkg != nil {37		mainStart := testingPkg.Func("MainStart")38		params := mainStart.Signature.Params()39		testSig = funcField(params.At(1).Type())40		benchmarkSig = funcField(params.At(2).Type())41		// Does the package define this function?42		//   func TestMain(*testing.M)43		if f := pkg.Func("TestMain"); f != nil {44			sig := f.Type().(*types.Signature)45			starM := mainStart.Signature.Results().At(0).Type() // *testing.M46			if sig.Results().Len() == 0 &&47				sig.Params().Len() == 1 &&48				types.Identical(sig.Params().At(0).Type(), starM) {49				main = f50			}51		}52	}53	// TODO(adonovan): use a stable order, e.g. lexical.54	for _, mem := range pkg.Members {55		if f, ok := mem.(*Function); ok &&56			ast.IsExported(f.Name()) &&57			strings.HasSuffix(prog.Fset.Position(f.Pos()).Filename, "_test.go") {58			switch {59			case testSig != nil && isTestSig(f, "Test", testSig):60				tests = append(tests, f)61			case benchmarkSig != nil && isTestSig(f, "Benchmark", benchmarkSig):62				benchmarks = append(benchmarks, f)63			case isTestSig(f, "Example", exampleSig):64				examples = append(examples, f)65			default:66				continue67			}68		}69	}70	return71}72// Like isTest, but checks the signature too.73func isTestSig(f *Function, prefix string, sig *types.Signature) bool {74	return isTest(f.Name(), prefix) && types.Identical(f.Signature, sig)75}76// Given the type of one of the three slice parameters of testing.Main,77// returns the function type.78func funcField(slice types.Type) *types.Signature {79	return slice.(*types.Slice).Elem().Underlying().(*types.Struct).Field(1).Type().(*types.Signature)80}81// isTest tells whether name looks like a test (or benchmark, according to prefix).82// It is a Test (say) if there is a character after Test that is not a lower-case letter.83// We don't want TesticularCancer.84// Plundered from $GOROOT/src/cmd/go/test.go85func isTest(name, prefix string) bool {86	if !strings.HasPrefix(name, prefix) {87		return false88	}89	if len(name) == len(prefix) { // "Test" is ok90		return true91	}92	return ast.IsExported(name[len(prefix):])93}94// CreateTestMainPackage creates and returns a synthetic "testmain"95// package for the specified package if it defines tests, benchmarks or96// executable examples, or nil otherwise.  The new package is named97// "main" and provides a function named "main" that runs the tests,98// similar to the one that would be created by the 'go test' tool.99//100// Subsequent calls to prog.AllPackages include the new package.101// The package pkg must belong to the program prog.102//103// Deprecated: Use golang.org/x/tools/go/packages to access synthetic104// testmain packages.105func (prog *Program) CreateTestMainPackage(pkg *Package) *Package {106	if pkg.Prog != prog {107		log.Fatal("Package does not belong to Program")108	}109	// Template data110	var data struct {111		Pkg                         *Package112		Tests, Benchmarks, Examples []*Function113		Main                        *Function114		Go18                        bool115	}116	data.Pkg = pkg117	// Enumerate tests.118	data.Tests, data.Benchmarks, data.Examples, data.Main = FindTests(pkg)119	if data.Main == nil &&120		data.Tests == nil && data.Benchmarks == nil && data.Examples == nil {121		return nil122	}123	// Synthesize source for testmain package.124	path := pkg.Pkg.Path() + "$testmain"125	tmpl := testmainTmpl126	if testingPkg := prog.ImportedPackage("testing"); testingPkg != nil {127		// In Go 1.8, testing.MainStart's first argument is an interface, not a func.128		data.Go18 = types.IsInterface(testingPkg.Func("MainStart").Signature.Params().At(0).Type())129	} else {130		// The program does not import "testing", but FindTests131		// returned non-nil, which must mean there were Examples132		// but no Test, Benchmark, or TestMain functions.133		// We'll simply call them from testmain.main; this will134		// ensure they don't panic, but will not check any135		// "Output:" comments.136		// (We should not execute an Example that has no137		// "Output:" comment, but it's impossible to tell here.)138		tmpl = examplesOnlyTmpl139	}140	var buf bytes.Buffer141	if err := tmpl.Execute(&buf, data); err != nil {142		log.Fatalf("internal error expanding template for %s: %v", path, err)143	}144	if false { // debugging145		fmt.Fprintln(os.Stderr, buf.String())146	}...

TestMain

Using AI Code Generation

1import (2func TestMain(m *testing.M) {3	fmt.Println("before test")4	code := m.Run()5	fmt.Println("after test")6	os.Exit(code)7}8func TestA(t *testing.T) {9	fmt.Println("test a")10}11func TestB(t *testing.T) {12	fmt.Println("test b")13}14import (15func TestA(t *testing.T) {16	t.Log("test a")17}18func TestB(t *testing.T) {19	t.Log("test b")20}21--- PASS: TestA (0.00s)22--- PASS: TestB (0.00s)23import (24func TestA(t *testing.T) {25	t.Log("test a")26}27func TestB(t *testing.T) {28	t.Log("test b")29}30--- PASS: TestA (0.00s)31import (32func TestA(t *testing.T) {33	t.Log("test a")34}35func TestB(t *testing.T) {36	t.Log("test b")37}

TestMain

Using AI Code Generation

1import (2func TestMain(m *testing.M) {3    m.Run()4}5import (6func TestMain(m *testing.M) {7    m.Run()8}9import (10func TestMain(m *testing.M) {11    m.Run()12}13import (14func TestMain(m *testing.M) {15    m.Run()16}17import (18func TestMain(m *testing.M) {19    m.Run()20}21import (22func TestMain(m *testing.M) {23    m.Run()24}25import (26func TestMain(m *testing.M) {27    m.Run()28}29import (30func TestMain(m *testing.M) {31    m.Run()32}33import (34func TestMain(m *testing.M) {35    m.Run()36}

TestMain

Using AI Code Generation

1import (2func TestMain(m *testing.M) {3    fmt.Println("TestMain")4    m.Run()5}6import (7func TestMain(m *testing.M) {8    fmt.Println("TestMain")9    m.Run()10}11import (12func TestMain(m *testing.M) {13    fmt.Println("TestMain")14    m.Run()15}16import (17func TestMain(m *testing.M) {18    fmt.Println("TestMain")19    m.Run()20}21import (22func TestMain(m *testing.M) {23    fmt.Println("TestMain")24    m.Run()25}26import (27func TestMain(m *testing.M) {28    fmt.Println("TestMain")29    m.Run()30}31import (32func TestMain(m *testing.M) {33    fmt.Println("TestMain")34    m.Run()35}36import (37func TestMain(m *testing.M) {38    fmt.Println("TestMain")39    m.Run()40}41import (42func TestMain(m *testing.M) {43    fmt.Println("TestMain")44    m.Run()45}46import (47func TestMain(m *testing.M

TestMain

Using AI Code Generation

1import (2func TestMain(m *testing.M) {3	fmt.Println("TestMain")4	m.Run()5}6func TestOne(t *testing.T) {7	fmt.Println("TestOne")8}9func TestTwo(t *testing.T) {10	fmt.Println("TestTwo")11}12func TestThree(t *testing.T) {13	fmt.Println("TestThree")14}

TestMain

Using AI Code Generation

1import (2func TestMain(m *testing.M) {3	fmt.Println("Before test")4	m.Run()5	fmt.Println("After test")6}7func TestA(t *testing.T) {8	fmt.Println("Testing A")9}10func TestB(t *testing.T) {11	fmt.Println("Testing B")12}13func TestC(t *testing.T) {14	fmt.Println("Testing C")15}16import (17func TestA(t *testing.T) {18	t.Run("A1", func(t *testing.T) {19		fmt.Println("A1")20	})21	t.Run("A2", func(t *testing.T) {22		fmt.Println("A2")23	})24	t.Run("A3", func(t *testing.T) {25		fmt.Println("A3")26	})27}28import (29func TestA(t *testing.T) {30	t.Run("A1", func(t *testing.T) {31		fmt.Println("A1")32	})33	t.Run("A2", func(t *testing.T) {34		fmt.Println("A2")35	})36	t.Run("A3", func(t *testing.T) {37		fmt.Println("A3")38	})39}

TestMain

Using AI Code Generation

1import (2func TestMain(m *testing.M) {3	fmt.Println("TestMain")4}5func TestMain2(m *testing.M) {6	fmt.Println("TestMain2")7}8func TestMain3(m *testing.M) {9	fmt.Println("TestMain3")10}11func TestMain4(m *testing.M) {12	fmt.Println("TestMain4")13}14func TestMain5(m *testing.M) {15	fmt.Println("TestMain5")16}17func TestMain6(m *testing.M) {18	fmt.Println("TestMain6")19}20import (21func TestMain(m *testing.M) {22	fmt.Println("TestMain")23}24func TestMain2(m *testing.M) {25	fmt.Println("TestMain2")26}27func TestMain3(m *testing.M) {28	fmt.Println("TestMain3")29}30func TestMain4(m *testing.M) {31	fmt.Println("TestMain4")32}33func TestMain5(m *testing.M) {34	fmt.Println("TestMain5")35}36func TestMain6(m *testing.M) {37	fmt.Println("TestMain6")38}39import (40func TestMain(m *testing.M) {41	fmt.Println("TestMain")42}43func TestMain2(m *testing.M) {44	fmt.Println("TestMain2")45}46func TestMain3(m *testing.M) {47	fmt.Println("TestMain3")48}49func TestMain4(m *testing.M) {50	fmt.Println("TestMain4")51}52func TestMain5(m *testing.M) {53	fmt.Println("TestMain5")54}55func TestMain6(m *testing.M) {56	fmt.Println("TestMain6")57}58import (

TestMain

Using AI Code Generation

1import (2func TestMain(m *testing.M) {3	fmt.Println("TestMain: Before the test")4	m.Run()5	fmt.Println("TestMain: After the test")6}7func TestA(t *testing.T) {8	fmt.Println("TestA: Before the test")9	t.Run("TestA1", TestA1)10	t.Run("TestA2", TestA2)11	fmt.Println("TestA: After the test")12}13func TestA1(t *testing.T) {14	fmt.Println("TestA1: Before the test")15	fmt.Println("TestA1: After the test")16}17func TestA2(t *testing.T) {18	fmt.Println("TestA2: Before the test")19	fmt.Println("TestA2: After the test")20}21import (22func TestMain(m *testing.M) {23	fmt.Println("TestMain: Before the test")24	m.Run()25	fmt.Println("TestMain: After the test")26}27func TestA(t *testing.T) {28	fmt.Println("TestA: Before the test")29	TestA1(t)30	TestA2(t)31	fmt.Println("TestA: After the test")32}33func TestA1(t *testing.T) {34	fmt.Println("TestA1: Before the test")35	fmt.Println("TestA1: After the test")36}37func TestA2(t *testing.T) {38	fmt.Println("TestA2: Before the test")39	fmt.Println("TestA2: After the test")40}41import (42func TestMain(m *testing.M) {

TestMain

Using AI Code Generation

1import (2func TestMain(m *testing.M) {3    fmt.Println("TestMain")4    m.Run()5}6func TestHelloWorld(t *testing.T) {7    fmt.Println("TestHelloWorld")8}9func TestHelloWorld2(t *testing.T) {10    fmt.Println("TestHelloWorld2")11}12--- PASS: TestHelloWorld (0.00s)13--- PASS: TestHelloWorld2 (0.00s)

TestMain

Using AI Code Generation

1func TestMain(m *testing.M) {2    os.Exit(m.Run())3}4func Test(t *testing.T) {5}6func Benchmark(b *testing.B) {7}8func Example() {9}10func ExampleMain() {11}12func ExampleOutput() {13}14func ExampleError() {15}16func ExampleName() {17}18func ExampleF() {19}20func ExampleS() {21}22func ExampleNew() {23}24func ExampleNewError() {25}26func ExampleNewName() {27}28func ExampleNewF() {

TestMain

Using AI Code Generation

1import (2func TestMain(t *testing.T) {3}4import (5func TestMain(t *testing.T) {6}7--- PASS: TestMain (0.00 seconds)8--- PASS: TestMain (0.00 seconds)9--- PASS: TestMain (0.00 seconds)10--- PASS: TestMain (0.00 seconds)11--- PASS: TestMain (0.00 seconds)

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.