github.com/phuslu/log@v1.0.100/multi_test.go

package log

import (
	"errors"
	"io"
	"os"
	"path/filepath"
	"testing"
)

func TestMultiWriter(t *testing.T) {
	w := &MultiWriter{
		InfoWriter:    &FileWriter{Filename: "file-info.log"},
		WarnWriter:    &FileWriter{Filename: "file-warn.log"},
		ErrorWriter:   &FileWriter{Filename: "file-error.log"},
		ConsoleWriter: &ConsoleWriter{ColorOutput: true},
		ConsoleLevel:  ErrorLevel,
	}

	for _, level := range []string{"trace", "debug", "info", "warning", "error", "fatal", "panic", "hahaha"} {
		_, err := wlprintf(w, ParseLevel(level), `{"ts":1234567890,"level":"%s","caller":"test.go:42","error":"i am test error","foo":"bar","n":42,"message":"hello json mutli writer"}`+"\n", level)
		if err != nil {
			t.Errorf("test json mutli writer error: %+v", err)
		}
		_, err = wlprintf(w, ParseLevel(level), `{"time":"2019-07-10T05:35:54.277Z","level":"%s","caller":"test.go:42","error":"i am test error","foo":"bar","n":42,"message":"hello json mutli writer"}`+"\n", level)
		if err != nil {
			t.Errorf("test json mutli writer error: %+v", err)
		}
		_, err = wlprintf(w, ParseLevel(level), `{"time":"2019-07-10T05:35:54.277+08:00","level":"%s","caller":"test.go:42","error":"i am test error","foo":"bar","n":42,"message":"hello json mutli writer"}`+"\n", level)
		if err != nil {
			t.Errorf("test json mutli writer error: %+v", err)
		}
	}

	if err := w.Close(); err != nil {
		t.Errorf("test close mutli writer error: %+v", err)
	}

	matches, _ := filepath.Glob("file-*.*.log")
	for i := range matches {
		err := os.Remove(matches[i])
		if err != nil {
			t.Fatalf("os remove %s error: %+v", matches[i], err)
		}
	}
}

type errorWriter struct {
	io.WriteCloser
}

var errorWriterOK = errors.New("errorWriter return OK")

func (ew errorWriter) Write(p []byte) (n int, err error) {
	n, _ = ew.WriteCloser.Write(p)
	err = errorWriterOK
	return
}

func (ew errorWriter) Close() (err error) {
	ew.WriteCloser.Close()
	err = errorWriterOK
	return
}

func TestMultiWriterError(t *testing.T) {
	file, err := os.OpenFile(os.DevNull, os.O_WRONLY, 0644)
	if err != nil {
		t.Errorf("open null file error: %+v", err)
	}

	w := &MultiWriter{
		InfoWriter:    IOWriter{errorWriter{file}},
		WarnWriter:    IOWriter{errorWriter{file}},
		ErrorWriter:   IOWriter{errorWriter{file}},
		ConsoleWriter: &ConsoleWriter{ColorOutput: true},
		ConsoleLevel:  TraceLevel,
	}

	for _, level := range []string{"trace", "debug", "info", "warning", "error", "fatal", "panic", "hahaha"} {
		_, err := wlprintf(w, ParseLevel(level), `{"time":"2019-07-10T05:35:54.277Z","level":"%s","caller":"test.go:42","error":"i am test error","foo":"bar","n":42,"message":"hello json mutli writer"}`+"\n", level)
		if err == nil {
			t.Errorf("test json error writer error: %+v", err)
		}
	}

	if err := w.Close(); err != nil {
		t.Errorf("test close error writer error: %+v", err)
	}
}

func TestMultiWriterEntry(t *testing.T) {
	w := &MultiWriter{
		ConsoleWriter: &ConsoleWriter{
			ColorOutput: true,
		},
		ConsoleLevel: InfoLevel,
	}

	var err error
	for _, level := range []string{"trace", "debug", "info", "warning", "error", "fatal", "panic", "hahaha"} {
		_, err = wlprintf(w, ParseLevel(level), `{"time":"2019-07-10T05:35:54.277Z","level":"%s","caller":"test.go:42","error":"i am test error","foo":"bar","n":42,"message":"hello json mutli writer"}`+"\n", level)
		if err != nil {
			t.Errorf("test json mutli writer error: %+v", err)
		}
	}

	if err := w.Close(); err != nil {
		t.Errorf("test close mutli writer error: %+v", err)
	}
}

func TestMultiEntryWriter(t *testing.T) {
	w := &MultiEntryWriter{
		&FileWriter{Filename: "file-1.log"},
		&FileWriter{Filename: "file-2.log"},
		&ConsoleWriter{ColorOutput: true},
	}

	for _, level := range []string{"trace", "debug", "info", "warning", "error", "fatal", "panic", "hahaha"} {
		_, err := wlprintf(w, ParseLevel(level), `{"ts":1234567890,"level":"%s","caller":"test.go:42","error":"i am test error","foo":"bar","n":42,"message":"hello json mutli writer"}`+"\n", level)
		if err != nil {
			t.Errorf("test json mutli writer error: %+v", err)
		}
		_, err = wlprintf(w, ParseLevel(level), `{"time":"2019-07-10T05:35:54.277Z","level":"%s","caller":"test.go:42","error":"i am test error","foo":"bar","n":42,"message":"hello json mutli writer"}`+"\n", level)
		if err != nil {
			t.Errorf("test json mutli writer error: %+v", err)
		}
		_, err = wlprintf(w, ParseLevel(level), `{"time":"2019-07-10T05:35:54.277+08:00","level":"%s","caller":"test.go:42","error":"i am test error","foo":"bar","n":42,"message":"hello json mutli writer"}`+"\n", level)
		if err != nil {
			t.Errorf("test json mutli writer error: %+v", err)
		}
	}

	if err := w.Close(); err != nil {
		t.Errorf("test close mutli writer error: %+v", err)
	}

	matches, _ := filepath.Glob("file-*.*.log")
	for i := range matches {
		err := os.Remove(matches[i])
		if err != nil {
			t.Fatalf("os remove %s error: %+v", matches[i], err)
		}
	}
}

type errorEntryWriter struct {
	io.WriteCloser
}

var errorEntryWriterOK = errors.New("errorEntryWriter return OK")

func (ew errorEntryWriter) Write(p []byte) (n int, err error) {
	n, _ = ew.WriteCloser.Write(p)
	err = errorEntryWriterOK
	return
}

func (ew errorEntryWriter) Close() (err error) {
	ew.WriteCloser.Close()
	err = errorEntryWriterOK
	return
}

func TestMultiEntryWriterError(t *testing.T) {
	file, err := os.OpenFile(os.DevNull, os.O_WRONLY, 0644)
	if err != nil {
		t.Errorf("open null file error: %+v", err)
	}

	w := &MultiEntryWriter{
		IOWriter{errorEntryWriter{file}},
		IOWriter{errorEntryWriter{file}},
		&ConsoleWriter{ColorOutput: true},
	}

	for _, level := range []string{"trace", "debug", "info", "warning", "error", "fatal", "panic", "hahaha"} {
		_, err := wlprintf(w, ParseLevel(level), `{"time":"2019-07-10T05:35:54.277Z","level":"%s","caller":"test.go:42","error":"i am test error","foo":"bar","n":42,"message":"hello json mutli writer"}`+"\n", level)
		if err == nil {
			t.Errorf("test json error writer error: %+v", err)
		}
	}

	if err := w.Close(); err != nil {
		t.Errorf("test close error writer error: %+v", err)
	}
}