03a828ef51
Merge in DNS/adguard-home from golibs-errors to master
Squashed commit of the following:
commit 5aba278a31c5a213bd9e08273ce7277c57713b22
Author: Ainar Garipov <A.Garipov@AdGuard.COM>
Date: Mon May 24 17:05:18 2021 +0300
all: imp code
commit f447eb875b81779fa9e391d98c31c1eeba7ef323
Author: Ainar Garipov <A.Garipov@AdGuard.COM>
Date: Mon May 24 15:33:45 2021 +0300
replace agherr with golibs' errors
47 lines
1.0 KiB
Go
47 lines
1.0 KiB
Go
// Package aghtest contains utilities for testing.
|
|
package aghtest
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/AdguardTeam/golibs/log"
|
|
)
|
|
|
|
// DiscardLogOutput runs tests with discarded logger output.
|
|
func DiscardLogOutput(m *testing.M) {
|
|
// TODO(e.burkov): Refactor code and tests to not use the global mutable
|
|
// logger.
|
|
log.SetOutput(io.Discard)
|
|
|
|
os.Exit(m.Run())
|
|
}
|
|
|
|
// ReplaceLogWriter moves logger output to w and uses Cleanup method of t to
|
|
// revert changes.
|
|
func ReplaceLogWriter(t *testing.T, w io.Writer) {
|
|
stdWriter := log.Writer()
|
|
t.Cleanup(func() {
|
|
log.SetOutput(stdWriter)
|
|
})
|
|
log.SetOutput(w)
|
|
}
|
|
|
|
// ReplaceLogLevel sets logging level to l and uses Cleanup method of t to
|
|
// revert changes.
|
|
func ReplaceLogLevel(t *testing.T, l log.Level) {
|
|
switch l {
|
|
case log.INFO, log.DEBUG, log.ERROR:
|
|
// Go on.
|
|
default:
|
|
t.Fatalf("wrong l value (must be one of %v, %v, %v)", log.INFO, log.DEBUG, log.ERROR)
|
|
}
|
|
|
|
stdLevel := log.GetLevel()
|
|
t.Cleanup(func() {
|
|
log.SetLevel(stdLevel)
|
|
})
|
|
log.SetLevel(l)
|
|
}
|