more tests

agent/tactical/service/service.go

@@ -24,7 +24,7 @@ func RunAsService(version string) {
 	wg.Wait()
 }
 
-func AgentSvc(version string) {
+func AgentSvc(version string) error {
 	config := config.NewAgentConfig()
 	go shared.GetPython(false)
 	utils.CreateTRMMTempDir()
@@ -35,6 +35,7 @@ func AgentSvc(version string) {
 	server := fmt.Sprintf("tls://%s:4222", config.APIURL)
 	nc, err := nats.Connect(server, opts...)
 	if err != nil {
+		return err
 	}
 
 	for _, s := range natsCheckin {

agent/tactical/service/service_test.go

@@ -0,0 +1,36 @@
+package service_test
+
+import (
+	"errors"
+	"testing"
+
+	"github.com/amidaware/rmmagent/agent/tactical/service"
+)
+
+func TestAgentSvc(t *testing.T) {
+	testTable := []struct {
+		name          string
+		expectedError error
+		version       string
+	}{
+		{
+			name:          "Agent Svc",
+			expectedError: nil,
+			version:       "2.0.4",
+		},
+		{
+			name:          "Agent Svc Error",
+			expectedError: nil,
+			version:       "bad version",
+		},
+	}
+
+	for _, tt := range testTable {
+		t.Run(tt.name, func(t *testing.T) {
+			err := service.AgentSvc(tt.version)
+			if errors.Is(tt.expectedError, err) {
+				t.Errorf("expected (%v), got (%v)", tt.expectedError, err)
+			}
+		})
+	}
+}