11package main
22
33import (
4- "log"
4+ stdlog "log"
55 "os"
66 "testing"
77 "time"
88
9+ "github.com/go-kit/kit/log"
910 "github.com/numbleroot/pluto/imap"
1011 "github.com/numbleroot/pluto/utils"
1112)
@@ -21,7 +22,7 @@ func TestMain(m *testing.M) {
2122 // Create needed test environment.
2223 testEnv , err = utils .CreateTestEnv ("./test-config.toml" )
2324 if err != nil {
24- log .Fatal (err )
25+ stdlog .Fatal (err )
2526 }
2627
2728 // Run all nodes in background.
@@ -51,15 +52,15 @@ func RunAllNodes(testEnv *utils.TestEnv, workerName string) {
5152 // Initialize storage node.
5253 storage , err := imap .InitStorage (testEnv .Config )
5354 if err != nil {
54- log .Fatal (err )
55+ stdlog .Fatal (err )
5556 }
5657
5758 go func () {
5859
5960 // Wait for shutdown signal on channel.
6061 <- testEnv .DownStorage
6162
62- log .Printf ("[utils.RunAllNodes] Closing storage socket.\n " )
63+ stdlog .Printf ("[utils.RunAllNodes] Closing storage socket.\n " )
6364
6465 // Shut down storage node.
6566 storage .MailSocket .Close ()
@@ -79,17 +80,17 @@ func RunAllNodes(testEnv *utils.TestEnv, workerName string) {
7980 go func () {
8081
8182 // Initialize workerName worker node.
82- worker , err := imap .InitWorker (testEnv .Config , workerName )
83+ worker , err := imap .InitWorker (log . NewNopLogger (), testEnv .Config , workerName )
8384 if err != nil {
84- log .Fatal (err )
85+ stdlog .Fatal (err )
8586 }
8687
8788 go func () {
8889
8990 // Wait for shutdown signal on channel.
9091 <- testEnv .DownWorker
9192
92- log .Printf ("[utils.RunAllNodes] Closing '%s' socket.\n " , workerName )
93+ stdlog .Printf ("[utils.RunAllNodes] Closing '%s' socket.\n " , workerName )
9394
9495 // Shut down worker node.
9596 worker .MailSocket .Close ()
@@ -110,21 +111,21 @@ func RunAllNodes(testEnv *utils.TestEnv, workerName string) {
110111
111112 authenticator , err := initAuthenticator (testEnv .Config )
112113 if err != nil {
113- log .Fatal (err )
114+ stdlog .Fatal (err )
114115 }
115116
116117 // Initialize distributor node.
117- distr , err := imap .InitDistributor (testEnv .Config , authenticator )
118+ distr , err := imap .InitDistributor (log . NewNopLogger (), testEnv .Config , authenticator )
118119 if err != nil {
119- log .Fatal (err )
120+ stdlog .Fatal (err )
120121 }
121122
122123 go func () {
123124
124125 // Wait for shutdown signal on channel.
125126 <- testEnv .DownDistr
126127
127- log .Printf ("[utils.RunAllNodes] Closing distributor socket.\n " )
128+ stdlog .Printf ("[utils.RunAllNodes] Closing distributor socket.\n " )
128129
129130 // Shut down distributor node.
130131 distr .Socket .Close ()
0 commit comments