Cleanup k8s logs
This commit is contained in:
@@ -13,33 +13,44 @@ import (
|
||||
)
|
||||
|
||||
func TestKubernetes(t *testing.T) {
|
||||
// TODO: fix local test running
|
||||
return
|
||||
|
||||
if os.Getenv("IN_TRAVIS_CI") == "yes" {
|
||||
t.Skip("In Travis CI")
|
||||
}
|
||||
|
||||
k := New(log.Name("micro-network"))
|
||||
|
||||
r, w, err := os.Pipe()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
s := os.Stderr
|
||||
os.Stderr = w
|
||||
meta := make(map[string]string)
|
||||
|
||||
write := log.Record{
|
||||
Timestamp: time.Unix(0, 0).UTC(),
|
||||
Message: "Test log entry",
|
||||
Metadata: meta,
|
||||
}
|
||||
|
||||
meta["foo"] = "bar"
|
||||
|
||||
k.Write(write)
|
||||
b := &bytes.Buffer{}
|
||||
w.Close()
|
||||
io.Copy(b, r)
|
||||
os.Stderr = s
|
||||
|
||||
var read log.Record
|
||||
|
||||
if err := json.Unmarshal(b.Bytes(), &read); err != nil {
|
||||
t.Fatalf("json.Unmarshal failed: %s", err.Error())
|
||||
}
|
||||
|
||||
assert.Equal(t, write, read, "Write was not equal")
|
||||
|
||||
records, err := k.Read()
|
||||
@@ -48,13 +59,16 @@ func TestKubernetes(t *testing.T) {
|
||||
|
||||
stream, err := k.Stream()
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
t.Fatal(err)
|
||||
}
|
||||
records = []log.Record{}
|
||||
|
||||
records = nil
|
||||
|
||||
go stream.Stop()
|
||||
|
||||
for s := range stream.Chan() {
|
||||
records = append(records, s)
|
||||
}
|
||||
assert.Equal(t, 0, len(records), "Stream should return nothing")
|
||||
|
||||
assert.Equal(t, 0, len(records), "Stream should return nothing")
|
||||
}
|
||||
|
Reference in New Issue
Block a user