Add verbosity setting

This commit is contained in:
Aleksander Mistewicz 2019-09-28 14:24:54 +02:00
parent 21b236771b
commit 8053766092

View File

@ -22,23 +22,26 @@ func GenerateConfig() *config.NodeConfig {
return cfg return cfg
} }
func GetLoggerWithPrefix(prefix string) *log.Logger { func GetLoggerWithPrefix(prefix string, verbose bool) *log.Logger {
l := log.New(os.Stderr, prefix, log.Flags()) l := log.New(os.Stderr, prefix, log.Flags())
if !verbose {
return l
}
l.EnableLevel("info") l.EnableLevel("info")
l.EnableLevel("warn") l.EnableLevel("warn")
l.EnableLevel("error") l.EnableLevel("error")
return l return l
} }
func CreateAndConnectTwo(t testing.TB) (*Core, *Core) { func CreateAndConnectTwo(t testing.TB, verbose bool) (*Core, *Core) {
nodeA := Core{} nodeA := Core{}
_, err := nodeA.Start(GenerateConfig(), GetLoggerWithPrefix("A: ")) _, err := nodeA.Start(GenerateConfig(), GetLoggerWithPrefix("A: ", verbose))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
nodeB := Core{} nodeB := Core{}
_, err = nodeB.Start(GenerateConfig(), GetLoggerWithPrefix("B: ")) _, err = nodeB.Start(GenerateConfig(), GetLoggerWithPrefix("B: ", verbose))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -71,7 +74,7 @@ func WaitConnected(nodeA, nodeB *Core) bool {
} }
func TestCore_Start_Connect(t *testing.T) { func TestCore_Start_Connect(t *testing.T) {
CreateAndConnectTwo(t) CreateAndConnectTwo(t, true)
} }
func CreateEchoListener(t testing.TB, nodeA *Core, bufLen int, repeats int) chan struct{} { func CreateEchoListener(t testing.TB, nodeA *Core, bufLen int, repeats int) chan struct{} {
@ -114,7 +117,7 @@ func CreateEchoListener(t testing.TB, nodeA *Core, bufLen int, repeats int) chan
} }
func TestCore_Start_Transfer(t *testing.T) { func TestCore_Start_Transfer(t *testing.T) {
nodeA, nodeB := CreateAndConnectTwo(t) nodeA, nodeB := CreateAndConnectTwo(t, true)
msgLen := 1500 msgLen := 1500
done := CreateEchoListener(t, nodeA, msgLen, 1) done := CreateEchoListener(t, nodeA, msgLen, 1)
@ -151,7 +154,7 @@ func TestCore_Start_Transfer(t *testing.T) {
} }
func BenchmarkCore_Start_Transfer(b *testing.B) { func BenchmarkCore_Start_Transfer(b *testing.B) {
nodeA, nodeB := CreateAndConnectTwo(b) nodeA, nodeB := CreateAndConnectTwo(b, false)
msgLen := 1500 // typical MTU msgLen := 1500 // typical MTU
done := CreateEchoListener(b, nodeA, msgLen, b.N) done := CreateEchoListener(b, nodeA, msgLen, b.N)