Use Sirupsen/logrus with uppercase
[govpp.git] / vendor / github.com / Sirupsen / logrus / text_formatter_test.go
diff --git a/vendor/github.com/Sirupsen/logrus/text_formatter_test.go b/vendor/github.com/Sirupsen/logrus/text_formatter_test.go
new file mode 100644 (file)
index 0000000..9793b5f
--- /dev/null
@@ -0,0 +1,87 @@
+package logrus
+
+import (
+       "bytes"
+       "errors"
+       "strings"
+       "testing"
+       "time"
+)
+
+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(tf.QuoteCharacter))
+               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(true, "/foobar")
+       checkQuoting(true, "x y")
+       checkQuoting(true, "x,y")
+       checkQuoting(false, errors.New("invalid"))
+       checkQuoting(true, errors.New("invalid argument"))
+
+       // Test for custom quote character.
+       tf.QuoteCharacter = "`"
+       checkQuoting(false, "")
+       checkQuoting(false, "abcd")
+       checkQuoting(true, "/foobar")
+       checkQuoting(true, errors.New("invalid argument"))
+
+       // Test for multi-character quotes.
+       tf.QuoteCharacter = "§~±"
+       checkQuoting(false, "abcd")
+       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 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(customFormatter.QuoteCharacter) : timeEnd-1-len(customFormatter.QuoteCharacter)]
+               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.