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)
}
}
func controlRouterFactory(userIDer multitenant.UserIDer, controlRouterURL string) (app.ControlRouter, error) {
if controlRouterURL == "local" {
return app.NewLocalControlRouter(), nil
}
parsed, err := url.Parse(controlRouterURL)
if err != nil {
return nil, err
}
if parsed.Scheme == "sqs" {
prefix := strings.TrimPrefix(parsed.Path, "/")
sqsConfig, err := awsConfigFromURL(parsed)
if err != nil {
return nil, err
}
return multitenant.NewSQSControlRouter(sqsConfig, userIDer, prefix), nil
}
return nil, fmt.Errorf("Invalid control router '%s'", controlRouterURL)
}