package api import ( "connectrpc.com/connect" "context" "github.com/stretchr/testify/assert" "testing" log "github.com/sirupsen/logrus" apiv1 "github.com/OliveTin/OliveTin/gen/olivetin/api/v1" apiv1connect "github.com/OliveTin/OliveTin/gen/olivetin/api/v1/apiv1connect" config "github.com/OliveTin/OliveTin/internal/config" "github.com/OliveTin/OliveTin/internal/executor" "net/http" "net/http/httptest" ) func getNewTestServerAndClient(t *testing.T, injectedConfig *config.Config) (*httptest.Server, apiv1connect.OliveTinApiServiceClient) { ex := executor.DefaultExecutor(injectedConfig) ex.RebuildActionMap() path, handler := GetNewHandler(ex) path = "/api" + path mux := http.NewServeMux() mux.HandleFunc(path, func(w http.ResponseWriter, r *http.Request) { log.Infof("HTTP Request: %s %s", r.Method, r.URL.Path) http.StripPrefix("/api/", handler) }) log.Infof("API path is %s", path) httpclient := &http.Client{} ts := httptest.NewServer(mux) client := apiv1connect.NewOliveTinApiServiceClient(httpclient, ts.URL+"/api") log.Infof("Test server URL is %s", ts.URL+path) return ts, client } func TestGetActionsAndStart(t *testing.T) { cfg := config.DefaultConfig() btn1 := &config.Action{} btn1.Title = "blat" btn1.ID = "blat" btn1.Shell = "echo 'test'" cfg.Actions = append(cfg.Actions, btn1) ex := executor.DefaultExecutor(cfg) ex.RebuildActionMap() conn, client := getNewTestServerAndClient(t, cfg) respInit, err := client.Init(context.Background(), connect.NewRequest(&apiv1.InitRequest{})) respGetReady, err := client.GetReadyz(context.Background(), connect.NewRequest(&apiv1.GetReadyzRequest{})) if err != nil { t.Errorf("GetDashboardComponentsRequest: %v", err) return } log.Infof("GetReadyz response: %v", respGetReady.Msg) assert.Equal(t, true, true, "sayHello Failed") // assert.Equal(t, 1, len(respGb.Msg.Actions), "Got 1 action button back") log.Printf("Response: %+v", respInit) respSa, err := client.StartAction(context.Background(), connect.NewRequest(&apiv1.StartActionRequest{ // ActionId: "blat" })) assert.NotNil(t, err, "Error 404 after start action") assert.Nil(t, respSa, "Nil response for non existing action") defer conn.Close() }