mirror of
https://github.com/weaveworks/scope.git
synced 2026-02-14 18:09:59 +00:00
5f7f74bf1b
- Add interfaces to allow for alternative implementations for Collector, ControlRouter and PipeRouter. - Pass contexts on http handlers to these interfaces. Although not used by the current (local, in-memory) implementations, the idea is this will be used to pass headers to implementations which support multitenancy (by, for instance, putting an authenticating reverse proxy in form of the app, and then inspecting the headers of the request for a used id).
74 lines
1.6 KiB
Go
74 lines
1.6 KiB
Go
package app_test
|
|
|
|
import (
|
|
"net"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/ugorji/go/codec"
|
|
|
|
"github.com/weaveworks/scope/app"
|
|
"github.com/weaveworks/scope/common/xfer"
|
|
"github.com/weaveworks/scope/probe/appclient"
|
|
)
|
|
|
|
func TestControl(t *testing.T) {
|
|
router := mux.NewRouter()
|
|
app.RegisterControlRoutes(router, app.NewLocalControlRouter())
|
|
server := httptest.NewServer(router)
|
|
defer server.Close()
|
|
|
|
ip, port, err := net.SplitHostPort(strings.TrimPrefix(server.URL, "http://"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
probeConfig := appclient.ProbeConfig{
|
|
ProbeID: "foo",
|
|
}
|
|
controlHandler := xfer.ControlHandlerFunc(func(req xfer.Request) xfer.Response {
|
|
if req.NodeID != "nodeid" {
|
|
t.Fatalf("'%s' != 'nodeid'", req.NodeID)
|
|
}
|
|
|
|
if req.Control != "control" {
|
|
t.Fatalf("'%s' != 'control'", req.Control)
|
|
}
|
|
|
|
return xfer.Response{
|
|
Value: "foo",
|
|
}
|
|
})
|
|
client, err := appclient.NewAppClient(probeConfig, ip+":"+port, ip+":"+port, controlHandler)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
client.ControlConnection()
|
|
defer client.Stop()
|
|
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
httpClient := http.Client{
|
|
Timeout: 1 * time.Second,
|
|
}
|
|
resp, err := httpClient.Post(server.URL+"/api/control/foo/nodeid/control", "", nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
var response xfer.Response
|
|
decoder := codec.NewDecoder(resp.Body, &codec.JsonHandle{})
|
|
if err := decoder.Decode(&response); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if response.Value != "foo" {
|
|
t.Fatalf("'%s' != 'foo'", response.Value)
|
|
}
|
|
}
|