mirror of
https://github.com/weaveworks/scope.git
synced 2026-03-03 18:20:27 +00:00
41193b428e
It is not a singleton anymore. Instead it is an object with a registry backend. The default registry backend is provided, which is equivalent to what used to be before. Custom backend can be provided for testing purposes. The registry also supports batch operations to remove and add handlers as an atomic step.
87 lines
2.4 KiB
Go
87 lines
2.4 KiB
Go
package docker_test
|
|
|
|
import (
|
|
"io"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/weaveworks/scope/common/xfer"
|
|
"github.com/weaveworks/scope/probe/controls"
|
|
"github.com/weaveworks/scope/probe/docker"
|
|
"github.com/weaveworks/scope/report"
|
|
"github.com/weaveworks/scope/test"
|
|
)
|
|
|
|
func TestControls(t *testing.T) {
|
|
mdc := newMockClient()
|
|
setupStubs(mdc, func() {
|
|
hr := controls.NewDefaultHandlerRegistry()
|
|
registry, _ := docker.NewRegistry(10*time.Second, nil, false, "", hr)
|
|
defer registry.Stop()
|
|
|
|
for _, tc := range []struct{ command, result string }{
|
|
{docker.StopContainer, "stopped"},
|
|
{docker.StartContainer, "started"},
|
|
{docker.RestartContainer, "restarted"},
|
|
{docker.PauseContainer, "paused"},
|
|
{docker.UnpauseContainer, "unpaused"},
|
|
} {
|
|
result := hr.HandleControlRequest(xfer.Request{
|
|
Control: tc.command,
|
|
NodeID: report.MakeContainerNodeID("a1b2c3d4e5"),
|
|
})
|
|
if !reflect.DeepEqual(result, xfer.Response{
|
|
Error: tc.result,
|
|
}) {
|
|
t.Error(result)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
type mockPipe struct{}
|
|
|
|
func (mockPipe) Ends() (io.ReadWriter, io.ReadWriter) { return nil, nil }
|
|
func (mockPipe) CopyToWebsocket(io.ReadWriter, xfer.Websocket) error { return nil }
|
|
func (mockPipe) Close() error { return nil }
|
|
func (mockPipe) Closed() bool { return false }
|
|
func (mockPipe) OnClose(func()) {}
|
|
|
|
func TestPipes(t *testing.T) {
|
|
oldNewPipe := controls.NewPipe
|
|
defer func() { controls.NewPipe = oldNewPipe }()
|
|
controls.NewPipe = func(_ controls.PipeClient, _ string) (string, xfer.Pipe, error) {
|
|
return "pipeid", mockPipe{}, nil
|
|
}
|
|
|
|
mdc := newMockClient()
|
|
setupStubs(mdc, func() {
|
|
hr := controls.NewDefaultHandlerRegistry()
|
|
registry, _ := docker.NewRegistry(10*time.Second, nil, false, "", hr)
|
|
defer registry.Stop()
|
|
|
|
test.Poll(t, 100*time.Millisecond, true, func() interface{} {
|
|
_, ok := registry.GetContainer("ping")
|
|
return ok
|
|
})
|
|
|
|
for _, tc := range []string{
|
|
docker.AttachContainer,
|
|
docker.ExecContainer,
|
|
} {
|
|
result := hr.HandleControlRequest(xfer.Request{
|
|
Control: tc,
|
|
NodeID: report.MakeContainerNodeID("ping"),
|
|
})
|
|
want := xfer.Response{
|
|
Pipe: "pipeid",
|
|
RawTTY: true,
|
|
}
|
|
if !reflect.DeepEqual(result, want) {
|
|
t.Errorf("diff %s: %s", tc, test.Diff(want, result))
|
|
}
|
|
}
|
|
})
|
|
}
|