mirror of
https://github.com/lorsanstand/HomeOps-Hub.git
synced 2026-06-19 17:45:17 +03:00
feat: list containers
This commit is contained in:
@@ -0,0 +1,31 @@
|
||||
package docker_service
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/moby/moby/api/types"
|
||||
"github.com/moby/moby/api/types/container"
|
||||
)
|
||||
|
||||
type dockerAPI interface {
|
||||
Ping(ctx context.Context) (types.Ping, error)
|
||||
ContainerList(ctx context.Context, opts container.ListOptions) ([]container.Summary, error)
|
||||
}
|
||||
|
||||
type DockerService struct {
|
||||
dockerClient dockerAPI
|
||||
}
|
||||
|
||||
func NewDockerService(api dockerAPI) *DockerService {
|
||||
return &DockerService{dockerClient: api}
|
||||
}
|
||||
|
||||
func (d *DockerService) CheckDockerDaemon(ctx context.Context) error {
|
||||
_, err := d.dockerClient.Ping(ctx)
|
||||
return err
|
||||
}
|
||||
|
||||
func (d *DockerService) ContainersList(ctx context.Context) ([]container.Summary, error) {
|
||||
ContainersList, err := d.dockerClient.ContainerList(ctx, container.ListOptions{})
|
||||
return ContainersList, err
|
||||
}
|
||||
@@ -0,0 +1,99 @@
|
||||
package docker_service
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"testing"
|
||||
|
||||
"github.com/moby/moby/api/types"
|
||||
"github.com/moby/moby/api/types/container"
|
||||
)
|
||||
|
||||
var testError error = errors.New("test")
|
||||
|
||||
type DockerMock struct {
|
||||
pingErr error
|
||||
containers []container.Summary
|
||||
containerErr error
|
||||
}
|
||||
|
||||
func (d DockerMock) Ping(ctx context.Context) (types.Ping, error) {
|
||||
return types.Ping{}, d.pingErr
|
||||
}
|
||||
|
||||
func (d DockerMock) ContainerList(ctx context.Context, _ container.ListOptions) ([]container.Summary, error) {
|
||||
return d.containers, d.containerErr
|
||||
}
|
||||
|
||||
func TestCheckDockerDaemon(t *testing.T) {
|
||||
api := DockerMock{containerErr: nil, pingErr: nil, containers: []container.Summary{}}
|
||||
docker := NewDockerService(api)
|
||||
|
||||
if err := docker.CheckDockerDaemon(context.TODO()); err != nil {
|
||||
t.Errorf("check daemon failed: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestCheckDaemonFailed(t *testing.T) {
|
||||
api := DockerMock{containerErr: nil, pingErr: testError, containers: []container.Summary{}}
|
||||
docker := NewDockerService(api)
|
||||
|
||||
if err := docker.CheckDockerDaemon(context.TODO()); !errors.Is(err, testError) {
|
||||
t.Errorf("the error does not match the one originally specified: %v received: %v", testError, err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestContainersList(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
containers := []container.Summary{
|
||||
{ID: "123", Image: "postgres:latest"},
|
||||
{ID: "456", Image: "nginx:latest"},
|
||||
}
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
mock DockerMock
|
||||
wantLen int
|
||||
wantErr error
|
||||
}{
|
||||
{
|
||||
name: "success",
|
||||
mock: DockerMock{
|
||||
pingErr: nil,
|
||||
containers: containers,
|
||||
containerErr: nil,
|
||||
},
|
||||
wantLen: len(containers),
|
||||
wantErr: nil,
|
||||
},
|
||||
{
|
||||
name: "docker error",
|
||||
mock: DockerMock{
|
||||
pingErr: nil,
|
||||
containers: nil,
|
||||
containerErr: testError,
|
||||
},
|
||||
wantLen: 0,
|
||||
wantErr: testError,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
tt := tt
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
svc := NewDockerService(tt.mock)
|
||||
|
||||
got, err := svc.ContainersList(context.Background())
|
||||
if !errors.Is(err, tt.wantErr) {
|
||||
t.Fatalf("expected error %v, got: %v", tt.wantErr, err)
|
||||
}
|
||||
|
||||
if tt.wantLen != len(got) {
|
||||
t.Fatalf("expected %d containers, got: %d", tt.wantLen, len(got))
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
@@ -1 +1,21 @@
|
||||
package hub_service
|
||||
|
||||
import (
|
||||
"github.com/lorsanstand/HomeOps-Hub/internal/agent/rpc"
|
||||
"github.com/lorsanstand/HomeOps-Hub/internal/agent/service/docker_service"
|
||||
"github.com/rs/zerolog"
|
||||
)
|
||||
|
||||
type HubService struct {
|
||||
docker *docker_service.DockerService
|
||||
log zerolog.Logger
|
||||
hubConn *rpc.Connection
|
||||
}
|
||||
|
||||
func NewHubService(docker *docker_service.DockerService, log zerolog.Logger) *HubService {
|
||||
return &HubService{docker: docker, log: log}
|
||||
}
|
||||
|
||||
func (h *HubService) GatherInfoSystem() {
|
||||
|
||||
}
|
||||
|
||||
@@ -0,0 +1,15 @@
|
||||
package hub_service
|
||||
|
||||
type AgentRegistrationData struct {
|
||||
AgentID string
|
||||
AgentName string
|
||||
Config AgentConfig
|
||||
}
|
||||
|
||||
type AgentConfig struct {
|
||||
System string
|
||||
Docker bool
|
||||
Hostname string
|
||||
Version string
|
||||
Arch string
|
||||
}
|
||||
Reference in New Issue
Block a user