mirror of
https://github.com/weaveworks/scope.git
synced 2026-03-04 18:51:17 +00:00
b6dfe25499
* Hacky working prototype. * Operate with time.Duration offset instead of fixed timestamp. * Polished the backend code. * Made a nicer UI component. * Small refactorings of the websockets code. * Fixed the backend tests. * Better websocketing and smoother transitions * Small styling refactoring. * Detecting empty topologies. * Improved error messaging. * Addressed some of David's comments. * Moved nodesDeltaBuffer to a global state to fix the paused status rendering bug. * Small styling changes * Changed the websocket global state variables a bit. * Polishing & refactoring. * More polishing. * Final refactoring. * Addressed a couple of bugs. * Hidden the timeline control behind Cloud context and a feature flag. * Addressed most of @davkal's comments. * Added mixpanel tracking.
133 lines
2.8 KiB
Go
133 lines
2.8 KiB
Go
package app_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"golang.org/x/net/context"
|
|
|
|
"github.com/weaveworks/common/mtime"
|
|
"github.com/weaveworks/common/test"
|
|
"github.com/weaveworks/scope/app"
|
|
"github.com/weaveworks/scope/report"
|
|
"github.com/weaveworks/scope/test/reflect"
|
|
)
|
|
|
|
func TestCollector(t *testing.T) {
|
|
ctx := context.Background()
|
|
window := 10 * time.Second
|
|
c := app.NewCollector(window)
|
|
|
|
now := time.Now()
|
|
mtime.NowForce(now)
|
|
defer mtime.NowReset()
|
|
|
|
r1 := report.MakeReport()
|
|
r1.Endpoint.AddNode(report.MakeNode("foo"))
|
|
|
|
r2 := report.MakeReport()
|
|
r2.Endpoint.AddNode(report.MakeNode("foo"))
|
|
|
|
have, err := c.Report(ctx, mtime.Now())
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if want := report.MakeReport(); !reflect.DeepEqual(want, have) {
|
|
t.Error(test.Diff(want, have))
|
|
}
|
|
|
|
c.Add(ctx, r1, nil)
|
|
have, err = c.Report(ctx, mtime.Now())
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if want := r1; !reflect.DeepEqual(want, have) {
|
|
t.Error(test.Diff(want, have))
|
|
}
|
|
|
|
timeBefore := mtime.Now()
|
|
mtime.NowForce(now.Add(time.Second))
|
|
|
|
c.Add(ctx, r2, nil)
|
|
merged := report.MakeReport()
|
|
merged = merged.Merge(r1)
|
|
merged = merged.Merge(r2)
|
|
have, err = c.Report(ctx, mtime.Now())
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if want := merged; !reflect.DeepEqual(want, have) {
|
|
t.Error(test.Diff(want, have))
|
|
}
|
|
|
|
// Since the timestamp given is before r2 was added,
|
|
// it shouldn't be included in the final report.
|
|
have, err = c.Report(ctx, timeBefore)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if want := r1; !reflect.DeepEqual(want, have) {
|
|
t.Error(test.Diff(want, have))
|
|
}
|
|
}
|
|
|
|
func TestCollectorExpire(t *testing.T) {
|
|
now := time.Now()
|
|
mtime.NowForce(now)
|
|
defer mtime.NowReset()
|
|
|
|
ctx := context.Background()
|
|
window := 10 * time.Second
|
|
c := app.NewCollector(window)
|
|
|
|
// 1st check the collector is empty
|
|
have, err := c.Report(ctx, mtime.Now())
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if want := report.MakeReport(); !reflect.DeepEqual(want, have) {
|
|
t.Error(test.Diff(want, have))
|
|
}
|
|
|
|
// Now check an added report is returned
|
|
r1 := report.MakeReport()
|
|
r1.Endpoint.AddNode(report.MakeNode("foo"))
|
|
c.Add(ctx, r1, nil)
|
|
have, err = c.Report(ctx, mtime.Now())
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if want := r1; !reflect.DeepEqual(want, have) {
|
|
t.Error(test.Diff(want, have))
|
|
}
|
|
|
|
// Finally move time forward to expire the report
|
|
mtime.NowForce(now.Add(window))
|
|
have, err = c.Report(ctx, mtime.Now())
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if want := report.MakeReport(); !reflect.DeepEqual(want, have) {
|
|
t.Error(test.Diff(want, have))
|
|
}
|
|
}
|
|
|
|
func TestCollectorWait(t *testing.T) {
|
|
ctx := context.Background()
|
|
window := time.Millisecond
|
|
c := app.NewCollector(window)
|
|
|
|
waiter := make(chan struct{}, 1)
|
|
c.WaitOn(ctx, waiter)
|
|
defer c.UnWait(ctx, waiter)
|
|
c.(interface {
|
|
Broadcast()
|
|
}).Broadcast()
|
|
|
|
select {
|
|
case <-waiter:
|
|
default:
|
|
t.Fatal("Didn't unblock")
|
|
}
|
|
}
|