Function assertQueryStats

proxy/stats_test.go:238–256

assertQueryStats asserts the statistics using the provided parameters.

(
	tb testing.TB,
	stats *proxy.QueryStatistics,
	wantMainCount int,
	wantMainErr assert.BoolAssertionFunc,
	wantFallbackCount int,
	wantFallbackErr assert.BoolAssertionFunc,
)

// assertQueryStats asserts the statistics using the provided parameters.
func assertQueryStats(
	tb testing.TB,
	stats *proxy.QueryStatistics,
	wantMainCount int,
	wantMainErr assert.BoolAssertionFunc,
	wantFallbackCount int,
	wantFallbackErr assert.BoolAssertionFunc,
) {
	tb.Helper()

	main := stats.Main()
	assert.Lenf(tb, main, wantMainCount, "main stats count")

	fallback := stats.Fallback()
	assert.Lenf(tb, fallback, wantFallbackCount, "fallback stats count")

	wantMainErr(tb, isErrorInStats(main), "main err")
	wantFallbackErr(tb, isErrorInStats(fallback), "fallback err")
}

// isErrorInStats is a helper function for tests that returns true if the
// upstream statistics contain an DNS lookup error.