Skip to content

Fix the mock agent repo missing public key fetch logic #96

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
iAziz786 merged 1 commit into from
Oct 1, 2025
Merged

Fix the mock agent repo missing public key fetch logic #96

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
27 changes: 18 additions & 9 deletions app/service/agent/registration_test.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,14 +12,15 @@ import (
)

type mockAgentRepository struct {
createFunc func(ctx context.Context, agent *agent.Agent) error
updateFunc func(ctx context.Context, agent *agent.Agent) error
findByFingerprintFunc func(ctx context.Context, fp string) (*agent.Agent, error)
findByIDFunc func(ctx context.Context, id uint) (*agent.Agent, error)
addTagsFunc func(ctx context.Context, agentID uint, tags []agent.AgentTag) error
updateTagsFunc func(ctx context.Context, agentID uint, tags []agent.AgentTag) error
addRegistrationFunc func(ctx context.Context, registration *agent.AgentRegistration) error
transactionFunc func(ctx context.Context, fn func(agent.Repository) error) error
createFunc func(ctx context.Context, agent *agent.Agent) error
updateFunc func(ctx context.Context, agent *agent.Agent) error
findByFingerprintFunc func(ctx context.Context, fp string) (*agent.Agent, error)
findByIDFunc func(ctx context.Context, id uint) (*agent.Agent, error)
getPublicKeyByAgentID func(ctx context.Context, agentID string) (string, error)
addTagsFunc func(ctx context.Context, agentID uint, tags []agent.AgentTag) error
updateTagsFunc func(ctx context.Context, agentID uint, tags []agent.AgentTag) error
addRegistrationFunc func(ctx context.Context, registration *agent.AgentRegistration) error
transactionFunc func(ctx context.Context, fn func(agent.Repository) error) error
}

func (m *mockAgentRepository) Create(ctx context.Context, a *agent.Agent) error {
Expand Down Expand Up @@ -52,6 +53,13 @@ func (m *mockAgentRepository) FindByID(ctx context.Context, id uint) (*agent.Age
return nil, nil
}

func (m *mockAgentRepository) GetPublicKeyByAgentID(ctx context.Context, agentID string) (string, error) {
if m.getPublicKeyByAgentID != nil {
return m.getPublicKeyByAgentID(ctx, agentID)
}
return "", nil
}

func (m *mockAgentRepository) AddTags(ctx context.Context, agentID uint, tags []agent.AgentTag) error {
if m.addTagsFunc != nil {
return m.addTagsFunc(ctx, agentID, tags)
Expand Down Expand Up @@ -319,4 +327,5 @@ func TestRegistrationService(t *testing.T) {
assert.Equal(t, "database connection failed", failedRegistration.Error)
assert.Equal(t, uint(0), failedRegistration.AgentID) // No agent ID since creation failed
})
}
}