refactor: defer close connection or file

agent/internal/service/agent_service/agent_test.go

@@ -2,8 +2,12 @@ package agent_service
 
 import (
 	"context"
+	"errors"
+	"testing"
 
+	"github.com/lorsanstand/HomeOps-Hub/agent/internal/utils/config_yaml"
 	"github.com/lorsanstand/HomeOps-Hub/shared/domain"
+	"github.com/rs/zerolog"
 )
 
 type CollectorMock struct {
@@ -18,8 +22,88 @@ func (c *CollectorMock) GatherInfoSystem() (domain.HostInfo, []domain.Capability
 type ConnectionMock struct {
 	regAgentErr error
 	regResp     domain.RegisterAgentResponse
+	regData     domain.RegisterAgentRequest
 }
 
 func (c *ConnectionMock) RegisterAgent(ctx context.Context, RegisterData domain.RegisterAgentRequest) (domain.RegisterAgentResponse, error) {
+	c.regData = RegisterData
 	return c.regResp, c.regAgentErr
 }
+
+type SettingsMock struct {
+	insertErr error
+	agentID   string
+	countUse  int
+}
+
+func (s *SettingsMock) InsertAgentID(agentID string) error {
+	s.agentID = agentID
+	s.countUse++
+	return s.insertErr
+}
+
+func (s *SettingsMock) GetAgentID() string {
+	return s.agentID
+}
+
+func TestAgentService_RegisterAgentConn(t *testing.T) {
+	t.Parallel()
+
+	tests := []struct {
+		name             string
+		wantErr          error
+		insertAgentIDUse int
+		settings         SettingsMock
+		collector        CollectorMock
+		conn             ConnectionMock
+		cfg              config_yaml.AgentConfig
+	}{
+		{
+			name:             "success",
+			wantErr:          nil,
+			insertAgentIDUse: 1,
+			settings:         SettingsMock{agentID: "", insertErr: nil},
+			collector: CollectorMock{
+				host: domain.HostInfo{System: "Linux", Hostname: "test", Arch: "x64"},
+				caps: []domain.Capability{
+					{Available: true, Version: "0", Name: "testCaps", Reason: ""},
+				},
+			},
+			conn: ConnectionMock{regAgentErr: nil, regResp: domain.RegisterAgentResponse{AgentID: "123", Heartbeat: 4}},
+			cfg:  config_yaml.AgentConfig{AppName: "test"},
+		},
+	}
+
+	for _, tt := range tests {
+		tt := tt
+		t.Run(tt.name, func(t *testing.T) {
+			t.Parallel()
+			ctx := context.Background()
+
+			svc := NewAgentService(&tt.collector,
+				&tt.conn,
+				&tt.settings,
+				&tt.cfg,
+				zerolog.New(nil),
+			)
+
+			err := svc.RegisterAgentConn(ctx)
+			if !errors.Is(err, tt.wantErr) {
+				t.Fatalf("expected error %v, got: %v", tt.wantErr, err)
+			}
+
+			if tt.insertAgentIDUse != tt.settings.countUse {
+				t.Errorf("expected count insert agent id %v, got: %v", tt.insertAgentIDUse, tt.settings.countUse)
+			}
+
+			if tt.settings.agentID != tt.conn.regResp.AgentID {
+				t.Errorf("expected insert agent id %v, got: %v", tt.conn.regResp.AgentID, tt.settings.agentID)
+			}
+
+			if tt.cfg.AppName != tt.conn.regData.AgentName {
+				t.Fatalf("expected agent name %v, got: %v", tt.cfg.AppName, tt.conn.regData.AgentName)
+			}
+
+		})
+	}
+}