55 lines
1010 B
Go
55 lines
1010 B
Go
|
package agent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/hashicorp/logutils"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type MockStreamClient struct {
|
||
|
|
headers []*responseHeader
|
||
|
|
objs []interface{}
|
||
|
|
err error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *MockStreamClient) Send(h *responseHeader, o interface{}) error {
|
||
|
|
m.headers = append(m.headers, h)
|
||
|
|
m.objs = append(m.objs, o)
|
||
|
|
return m.err
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRPCLogStream(t *testing.T) {
|
||
|
|
sc := &MockStreamClient{}
|
||
|
|
filter := LevelFilter()
|
||
|
|
filter.MinLevel = logutils.LogLevel("INFO")
|
||
|
|
|
||
|
|
ls := newLogStream(sc, filter, 42, log.New(os.Stderr, "", log.LstdFlags))
|
||
|
|
defer ls.Stop()
|
||
|
|
|
||
|
|
log := "[DEBUG] this is a test log"
|
||
|
|
log2 := "[INFO] This should pass"
|
||
|
|
ls.HandleLog(log)
|
||
|
|
ls.HandleLog(log2)
|
||
|
|
|
||
|
|
time.Sleep(5 * time.Millisecond)
|
||
|
|
|
||
|
|
if len(sc.headers) != 1 {
|
||
|
|
t.Fatalf("expected 1 messages!")
|
||
|
|
}
|
||
|
|
for _, h := range sc.headers {
|
||
|
|
if h.Seq != 42 {
|
||
|
|
t.Fatalf("bad seq")
|
||
|
|
}
|
||
|
|
if h.Error != "" {
|
||
|
|
t.Fatalf("bad err")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
obj1 := sc.objs[0].(*logRecord)
|
||
|
|
if obj1.Log != log2 {
|
||
|
|
t.Fatalf("bad event %#v", obj1)
|
||
|
|
}
|
||
|
|
}
|