more tests

agent/tactical/rpc/rpc.go

@@ -36,7 +36,7 @@ var (
 	installWinUpdateLocker uint32
 )
 
-func RunRPC(version string) {
+func RunRPC(version string) error {
 	config := config.NewAgentConfig()
 	go service.RunAsService(version)
 	var wg sync.WaitGroup
@@ -45,6 +45,7 @@ func RunRPC(version string) {
 	server := fmt.Sprintf("tls://%s:4222", config.APIURL)
 	nc, err := nats.Connect(server, opts...)
 	if err != nil {
+		return err
 	}
 
 	nc.Subscribe(config.AgentID, func(msg *nats.Msg) {
@@ -457,11 +458,12 @@ func RunRPC(version string) {
 	nc.Flush()
 
 	if err := nc.LastError(); err != nil {
-		//a.Logger.Errorln(err)
-		os.Exit(1)
+		return err
 	}
 
 	wg.Wait()
+
+	return nil
 }
 
 func Start(version string, _ ksvc.Service) error {

agent/tactical/rpc/rpc_test.go

@@ -0,0 +1,31 @@
+package rpc_test
+
+import (
+	"errors"
+	"testing"
+
+	"github.com/amidaware/rmmagent/agent/tactical/rpc"
+)
+
+func TestRunRPC(t *testing.T) {
+	testTable := []struct {
+		name string
+		expectedError error
+		version string
+	}{
+		{
+			name: "Run RPC",
+			expectedError: nil, 
+			version: "development",
+		},
+	}
+
+	for _, tt := range testTable {
+		t.Run(tt.name, func(t *testing.T) {
+			err := rpc.RunRPC(tt.version)
+			if !errors.Is(tt.expectedError, err) {
+				t.Errorf("expected (%v), got (%v)", tt.expectedError, err)
+			}
+		})
+	}
+}