core-chain/p2p/testlog_test.go

package p2p

import (
	"testing"

	"github.com/ethereum/go-ethereum/logger"
)

type testLogger struct{ t *testing.T }

func testlog(t *testing.T) testLogger {
	logger.Reset()
	l := testLogger{t}
	logger.AddLogSystem(l)
	return l
}

func (l testLogger) LogPrint(msg logger.LogMsg) {
	l.t.Logf("%s", msg.String())
}

func (testLogger) detach() {
	logger.Flush()
	logger.Reset()
}