57 lines
1.0 KiB
Go
57 lines
1.0 KiB
Go
package kubernetes
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"io"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/micro/go-micro/debug/log"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestKubernetes(t *testing.T) {
|
|
k := New()
|
|
|
|
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),
|
|
Value: "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")
|
|
|
|
_, err = k.Read()
|
|
assert.Error(t, err, "Read should be unimplemented")
|
|
|
|
stream, err := k.Stream()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
records := []log.Record{}
|
|
go stream.Stop()
|
|
for s := range stream.Chan() {
|
|
records = append(records, s)
|
|
}
|
|
assert.Equal(t, 0, len(records), "Stream should return nothing")
|
|
|
|
}
|