github.com/tompao/docker@v1.9.1/daemon/logger/copier_test.go (about) 1 package logger 2 3 import ( 4 "bytes" 5 "encoding/json" 6 "io" 7 "testing" 8 "time" 9 ) 10 11 type TestLoggerJSON struct { 12 *json.Encoder 13 } 14 15 func (l *TestLoggerJSON) Log(m *Message) error { return l.Encode(m) } 16 17 func (l *TestLoggerJSON) Close() error { return nil } 18 19 func (l *TestLoggerJSON) Name() string { return "json" } 20 21 type TestLoggerText struct { 22 *bytes.Buffer 23 } 24 25 func (l *TestLoggerText) Log(m *Message) error { 26 _, err := l.WriteString(m.ContainerID + " " + m.Source + " " + string(m.Line) + "\n") 27 return err 28 } 29 30 func (l *TestLoggerText) Close() error { return nil } 31 32 func (l *TestLoggerText) Name() string { return "text" } 33 34 func TestCopier(t *testing.T) { 35 stdoutLine := "Line that thinks that it is log line from docker stdout" 36 stderrLine := "Line that thinks that it is log line from docker stderr" 37 var stdout bytes.Buffer 38 var stderr bytes.Buffer 39 for i := 0; i < 30; i++ { 40 if _, err := stdout.WriteString(stdoutLine + "\n"); err != nil { 41 t.Fatal(err) 42 } 43 if _, err := stderr.WriteString(stderrLine + "\n"); err != nil { 44 t.Fatal(err) 45 } 46 } 47 48 var jsonBuf bytes.Buffer 49 50 jsonLog := &TestLoggerJSON{Encoder: json.NewEncoder(&jsonBuf)} 51 52 cid := "a7317399f3f857173c6179d44823594f8294678dea9999662e5c625b5a1c7657" 53 c := NewCopier(cid, 54 map[string]io.Reader{ 55 "stdout": &stdout, 56 "stderr": &stderr, 57 }, 58 jsonLog) 59 c.Run() 60 wait := make(chan struct{}) 61 go func() { 62 c.Wait() 63 close(wait) 64 }() 65 select { 66 case <-time.After(1 * time.Second): 67 t.Fatal("Copier failed to do its work in 1 second") 68 case <-wait: 69 } 70 dec := json.NewDecoder(&jsonBuf) 71 for { 72 var msg Message 73 if err := dec.Decode(&msg); err != nil { 74 if err == io.EOF { 75 break 76 } 77 t.Fatal(err) 78 } 79 if msg.Source != "stdout" && msg.Source != "stderr" { 80 t.Fatalf("Wrong Source: %q, should be %q or %q", msg.Source, "stdout", "stderr") 81 } 82 if msg.ContainerID != cid { 83 t.Fatalf("Wrong ContainerID: %q, expected %q", msg.ContainerID, cid) 84 } 85 if msg.Source == "stdout" { 86 if string(msg.Line) != stdoutLine { 87 t.Fatalf("Wrong Line: %q, expected %q", msg.Line, stdoutLine) 88 } 89 } 90 if msg.Source == "stderr" { 91 if string(msg.Line) != stderrLine { 92 t.Fatalf("Wrong Line: %q, expected %q", msg.Line, stderrLine) 93 } 94 } 95 } 96 }