dotcloud/docker

package network

import (
    "context"
    "fmt"

    "github.com/docker/docker/api/types/network"
    "github.com/docker/docker/client"
    "gotest.tools/v3/assert/cmp"
)

// IsNetworkAvailable provides a comparison to check if a docker network is available
func IsNetworkAvailable(ctx context.Context, c client.NetworkAPIClient, name string) cmp.Comparison {
    return func() cmp.Result {
        networks, err := c.NetworkList(ctx, network.ListOptions{})
        if err != nil {
            return cmp.ResultFromError(err)
        }
        for _, network := range networks {
            if network.Name == name {
                return cmp.ResultSuccess
            }
        }
        return cmp.ResultFailure(fmt.Sprintf("could not find network %s", name))
    }
}

// IsNetworkNotAvailable provides a comparison to check if a docker network is not available
func IsNetworkNotAvailable(ctx context.Context, c client.NetworkAPIClient, name string) cmp.Comparison {
    return func() cmp.Result {
        networks, err := c.NetworkList(ctx, network.ListOptions{})
        if err != nil {
            return cmp.ResultFromError(err)
        }
        for _, network := range networks {
            if network.Name == name {
                return cmp.ResultFailure(fmt.Sprintf("network %s is still present", name))
            }
        }
        return cmp.ResultSuccess
    }
}