Function getTestPackageName

coderd/database/dbtestutil/broker.go:197–233

getTestPackageName returns the package name of the test that called it.

(t TBSubset)

// getTestPackageName returns the package name of the test that called it.
func getTestPackageName(t TBSubset) string {
	packageName := "unknown"
	// Ask runtime.Callers for up to 100 program counters, including runtime.Callers itself.
	pc := make([]uintptr, 100)
	n := runtime.Callers(0, pc)
	if n == 0 {
		// No PCs available. This can happen if the first argument to
		// runtime.Callers is large.
		//
		// Return now to avoid processing the zero Frame that would
		// otherwise be returned by frames.Next below.
		t.Logf("could not determine test package name: no PCs available")
		return packageName
	}

	pc = pc[:n] // pass only valid pcs to runtime.CallersFrames
	frames := runtime.CallersFrames(pc)

	// Loop to get frames.
	// A fixed number of PCs can expand to an indefinite number of Frames.
	for {
		frame, more := frames.Next()

		if strings.HasPrefix(frame.Function, "github.com/coder/coder/v2/") {
			packageName = strings.SplitN(strings.TrimPrefix(frame.Function, "github.com/coder/coder/v2/"), ".", 2)[0]
		}
		if strings.HasPrefix(frame.Function, "testing") {
			break
		}

		// Check whether there are more frames to process after this one.
		if !more {
			break
		}
	}
	return packageName
}