mirror of
https://github.com/octoleo/restic.git
synced 2024-12-24 11:55:28 +00:00
142 lines
3.6 KiB
Go
142 lines
3.6 KiB
Go
|
package logrus
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"errors"
|
||
|
"fmt"
|
||
|
"strings"
|
||
|
"testing"
|
||
|
"time"
|
||
|
)
|
||
|
|
||
|
func TestFormatting(t *testing.T) {
|
||
|
tf := &TextFormatter{DisableColors: true}
|
||
|
|
||
|
testCases := []struct {
|
||
|
value string
|
||
|
expected string
|
||
|
}{
|
||
|
{`foo`, "time=\"0001-01-01T00:00:00Z\" level=panic test=foo\n"},
|
||
|
}
|
||
|
|
||
|
for _, tc := range testCases {
|
||
|
b, _ := tf.Format(WithField("test", tc.value))
|
||
|
|
||
|
if string(b) != tc.expected {
|
||
|
t.Errorf("formatting expected for %q (result was %q instead of %q)", tc.value, string(b), tc.expected)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestQuoting(t *testing.T) {
|
||
|
tf := &TextFormatter{DisableColors: true}
|
||
|
|
||
|
checkQuoting := func(q bool, value interface{}) {
|
||
|
b, _ := tf.Format(WithField("test", value))
|
||
|
idx := bytes.Index(b, ([]byte)("test="))
|
||
|
cont := bytes.Contains(b[idx+5:], []byte("\""))
|
||
|
if cont != q {
|
||
|
if q {
|
||
|
t.Errorf("quoting expected for: %#v", value)
|
||
|
} else {
|
||
|
t.Errorf("quoting not expected for: %#v", value)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
checkQuoting(false, "")
|
||
|
checkQuoting(false, "abcd")
|
||
|
checkQuoting(false, "v1.0")
|
||
|
checkQuoting(false, "1234567890")
|
||
|
checkQuoting(false, "/foobar")
|
||
|
checkQuoting(false, "foo_bar")
|
||
|
checkQuoting(false, "foo@bar")
|
||
|
checkQuoting(false, "foobar^")
|
||
|
checkQuoting(false, "+/-_^@f.oobar")
|
||
|
checkQuoting(true, "foobar$")
|
||
|
checkQuoting(true, "&foobar")
|
||
|
checkQuoting(true, "x y")
|
||
|
checkQuoting(true, "x,y")
|
||
|
checkQuoting(false, errors.New("invalid"))
|
||
|
checkQuoting(true, errors.New("invalid argument"))
|
||
|
|
||
|
// Test for quoting empty fields.
|
||
|
tf.QuoteEmptyFields = true
|
||
|
checkQuoting(true, "")
|
||
|
checkQuoting(false, "abcd")
|
||
|
checkQuoting(true, errors.New("invalid argument"))
|
||
|
}
|
||
|
|
||
|
func TestEscaping(t *testing.T) {
|
||
|
tf := &TextFormatter{DisableColors: true}
|
||
|
|
||
|
testCases := []struct {
|
||
|
value string
|
||
|
expected string
|
||
|
}{
|
||
|
{`ba"r`, `ba\"r`},
|
||
|
{`ba'r`, `ba'r`},
|
||
|
}
|
||
|
|
||
|
for _, tc := range testCases {
|
||
|
b, _ := tf.Format(WithField("test", tc.value))
|
||
|
if !bytes.Contains(b, []byte(tc.expected)) {
|
||
|
t.Errorf("escaping expected for %q (result was %q instead of %q)", tc.value, string(b), tc.expected)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestEscaping_Interface(t *testing.T) {
|
||
|
tf := &TextFormatter{DisableColors: true}
|
||
|
|
||
|
ts := time.Now()
|
||
|
|
||
|
testCases := []struct {
|
||
|
value interface{}
|
||
|
expected string
|
||
|
}{
|
||
|
{ts, fmt.Sprintf("\"%s\"", ts.String())},
|
||
|
{errors.New("error: something went wrong"), "\"error: something went wrong\""},
|
||
|
}
|
||
|
|
||
|
for _, tc := range testCases {
|
||
|
b, _ := tf.Format(WithField("test", tc.value))
|
||
|
if !bytes.Contains(b, []byte(tc.expected)) {
|
||
|
t.Errorf("escaping expected for %q (result was %q instead of %q)", tc.value, string(b), tc.expected)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestTimestampFormat(t *testing.T) {
|
||
|
checkTimeStr := func(format string) {
|
||
|
customFormatter := &TextFormatter{DisableColors: true, TimestampFormat: format}
|
||
|
customStr, _ := customFormatter.Format(WithField("test", "test"))
|
||
|
timeStart := bytes.Index(customStr, ([]byte)("time="))
|
||
|
timeEnd := bytes.Index(customStr, ([]byte)("level="))
|
||
|
timeStr := customStr[timeStart+5+len("\"") : timeEnd-1-len("\"")]
|
||
|
if format == "" {
|
||
|
format = time.RFC3339
|
||
|
}
|
||
|
_, e := time.Parse(format, (string)(timeStr))
|
||
|
if e != nil {
|
||
|
t.Errorf("time string \"%s\" did not match provided time format \"%s\": %s", timeStr, format, e)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
checkTimeStr("2006-01-02T15:04:05.000000000Z07:00")
|
||
|
checkTimeStr("Mon Jan _2 15:04:05 2006")
|
||
|
checkTimeStr("")
|
||
|
}
|
||
|
|
||
|
func TestDisableTimestampWithColoredOutput(t *testing.T) {
|
||
|
tf := &TextFormatter{DisableTimestamp: true, ForceColors: true}
|
||
|
|
||
|
b, _ := tf.Format(WithField("test", "test"))
|
||
|
if strings.Contains(string(b), "[0000]") {
|
||
|
t.Error("timestamp not expected when DisableTimestamp is true")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// TODO add tests for sorting etc., this requires a parser for the text
|
||
|
// formatter output.
|