Skip to content

Commit aa3431b

Browse files
committed
Use stdlog and go-kit as log in main_test.go
1 parent 585551b commit aa3431b

File tree

1 file changed

+12
-11
lines changed

1 file changed

+12
-11
lines changed

main_test.go

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,12 @@
11
package main
22

33
import (
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

Comments
 (0)