github/weave-scope
1
0
Fork 0
mirror of https://github.com/weaveworks/scope.git synced 2026-07-28 09:41:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Extend the node control rewriting test

Browse Source 
Just to make sure that all the node controls are rewritten, even those
that don't have a counterpart in topology controls.
This commit is contained in:
Krzesimir Nowak
2016-07-21 11:52:01 +02:00
parent 1f5dbb776f
commit 0116b963e3
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
probe/plugins/registry_internal_test.go
View File

@@ -762,6 +762,14 @@ func TestRegistryRegistersHandlers(t *testing.T) {
"/control": {http.StatusOK, mustMarshal(PluginResponse{})},
}),
}.file(),
mockPlugin{
t: t,
Name: "testPlugin2",
Handler: mapStringHandler(testResponseMap{
"/report": {http.StatusOK, mustMarshal(testReport(topologyWithControls("pod", "node2", []int{1, 2}, []int{1}), pluginSpec("testPlugin2", "reporter", "controller")))},
"/control": {http.StatusOK, mustMarshal(PluginResponse{})},
}),
}.file(),
)
defer restore(t)
@@ -775,11 +783,14 @@ func TestRegistryRegistersHandlers(t *testing.T) {
defer r.Close()
r.Report()
if len(testBackend.handlers) != 1 {
t.Fatalf("Expected only one registered handler, got %d", len(testBackend.handlers))
expectedLen := 3
if len(testBackend.handlers) != expectedLen {
t.Fatalf("Expected %d registered handler, got %d", expectedLen, len(testBackend.handlers))
}
fakeIDs := []string{
fakeControlID("testPlugin", controlID(1)),
fakeControlID("testPlugin2", controlID(1)),
fakeControlID("testPlugin2", controlID(2)),
}
for _, fakeID := range fakeIDs {
if _, found := testBackend.Handler(fakeID); !found {