mirror of
https://github.com/lorsanstand/HomeOps-Hub.git
synced 2026-06-19 20:05:17 +03:00
110 lines
2.7 KiB
Go
110 lines
2.7 KiB
Go
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 {
|
|
host domain.HostInfo
|
|
caps []domain.Capability
|
|
}
|
|
|
|
func (c *CollectorMock) GatherInfoSystem() (domain.HostInfo, []domain.Capability) {
|
|
return c.host, c.caps
|
|
}
|
|
|
|
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)
|
|
}
|
|
|
|
})
|
|
}
|
|
}
|