Tests are working again

scheduler/task/check_swarm_status.go

@@ -0,0 +1,69 @@
+package task
+
+import (
+	"context"
+	"log"
+
+	"github.com/docker/docker/api/types/swarm"
+	"github.com/play-with-docker/play-with-docker/docker"
+	"github.com/play-with-docker/play-with-docker/event"
+	"github.com/play-with-docker/play-with-docker/pwd/types"
+)
+
+type DockerSwarmStatus struct {
+	IsManager bool   `json:"is_manager"`
+	IsWorker  bool   `json:"is_worker"`
+	Instance  string `json:"instance"`
+}
+
+type checkSwarmStatus struct {
+	event   event.EventApi
+	factory docker.FactoryApi
+}
+
+var CheckSwarmStatusEvent event.EventType
+
+func init() {
+	CheckSwarmStatusEvent = event.NewEventType("instance docker swarm status")
+}
+
+func (t *checkSwarmStatus) Name() string {
+	return "CheckSwarmStatus"
+}
+
+func (t *checkSwarmStatus) Run(ctx context.Context, instance *types.Instance) error {
+	dockerClient, err := t.factory.GetForInstance(instance.SessionId, instance.Name)
+	if err != nil {
+		log.Println(err)
+		return err
+	}
+
+	status, err := getDockerSwarmStatus(ctx, dockerClient)
+	if err != nil {
+		log.Println(err)
+		return err
+	}
+	status.Instance = instance.Name
+
+	t.event.Emit(CheckSwarmStatusEvent, instance.SessionId, status)
+	return nil
+}
+
+func NewCheckSwarmStatus(e event.EventApi, f docker.FactoryApi) *checkSwarmStatus {
+	return &checkSwarmStatus{event: e, factory: f}
+}
+
+func getDockerSwarmStatus(ctx context.Context, client docker.DockerApi) (DockerSwarmStatus, error) {
+	status := DockerSwarmStatus{}
+	info, err := client.GetDaemonInfo()
+	if err != nil {
+		return status, err
+	}
+
+	if info.Swarm.LocalNodeState != swarm.LocalNodeStateInactive && info.Swarm.LocalNodeState != swarm.LocalNodeStateLocked {
+		status.IsManager = info.Swarm.ControlAvailable
+		status.IsWorker = !info.Swarm.ControlAvailable
+	}
+
+	return status, nil
+}