✨ Command registration

pkg/nex/command_test.go

@@ -0,0 +1,46 @@
+package nex_test
+
+import (
+	"fmt"
+	"git.solsynth.dev/hypernet/nexus/pkg/nex"
+	"git.solsynth.dev/hypernet/nexus/pkg/proto"
+	"net/http"
+	"testing"
+	"time"
+)
+
+func TestHandleCommand(t *testing.T) {
+	conn, err := nex.NewNexusConn("127.0.0.1:7001", &proto.ServiceInfo{
+		Id:       "echo01",
+		Type:     "echo",
+		Label:    "Echo",
+		GrpcAddr: "127.0.0.1:6001",
+		HttpAddr: nil,
+	})
+	if err != nil {
+		t.Fatal(fmt.Errorf("unable to connect nexus: %v", err))
+	}
+
+	if err := conn.RegisterService(); err != nil {
+		t.Fatal(fmt.Errorf("unable to register service: %v", err))
+	}
+
+	err = conn.AddCommand("echo", "get", nil, func(ctx *nex.CommandCtx) error {
+		return ctx.Write(ctx.Read(), http.StatusOK)
+	})
+	if err != nil {
+		t.Fatal(fmt.Errorf("unable to add command: %v", err))
+		return
+	}
+
+	go func() {
+		err := conn.RunCommands("127.0.0.1:6001")
+		if err != nil {
+			t.Error(fmt.Errorf("unable to run commands: %v", err))
+			return
+		}
+	}()
+
+	t.Log("Waiting 10 seconds for calling command...")
+	time.Sleep(time.Second * 10)
+}