kubelatte-ce

1
package lib
2

3
import (
4
	"fmt"
5
	"github.com/stretchr/testify/assert"
6
	"go.uber.org/zap"
7
	"net/http"
8
	"os"
9
	"testing"
10
)
11

12
func TestMain(m *testing.M) {
13
	LoggersInit("debug")
14
	os.Exit(m.Run())
15
}
16

17
func TestParseAndSetReverseLevel(t *testing.T) {
18

19
	AtomLevel = zap.NewAtomicLevel()
20

21
	t.Parallel()
22
	tests := []struct {
23
		level   string
24
		wantErr bool
25
	}{
26
		{level: "debug"},
27
		{level: "info"},
28
		{level: "warn"},
29
		{level: "error"},
30
		{level: "dpanic"},
31
		{level: "panic"},
32
		{level: "fatal"},
33
		{level: "err", wantErr: true},
34
	}
35
	for _, tt := range tests {
36
		t.Run(tt.level, func(t *testing.T) {
37
			if err := parseAndSetReverseLevel(tt.level); (err != nil) != tt.wantErr {
38
				t.Errorf("parseAndSetReverseLevel() error = %v, wantErr %v", err, tt.wantErr)
39
			}
40
		})
41
	}
42
}
43

44
func TestHandle(t *testing.T) {
45
	db := IsDebugMode()
46
	mux := http.ServeMux{}
47
	mux.Handle("/logging", AtomLevel)
48
	ZapLogger.Info(fmt.Sprintf("See log level by path %s", fmt.Sprintf("http://localhost%s/logging", ":8080")))
49
	assert.Equal(t, false, db)
50
}
51