package service // import "github.com/docker/docker/integration/service"

import (
	"context"
	"testing"

	"github.com/docker/docker/api/types"
	"github.com/docker/docker/api/types/network"
	"github.com/docker/docker/client"
	"github.com/docker/docker/integration/internal/container"
	"github.com/docker/docker/integration/internal/swarm"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestDockerNetworkConnectAlias(t *testing.T) {
	defer setupTest(t)()
	d := swarm.NewSwarm(t, testEnv)
	defer d.Stop(t)
	client, err := client.NewClientWithOpts(client.WithHost((d.Sock())))
	require.NoError(t, err)
	ctx := context.Background()

	name := "test-alias"
	_, err = client.NetworkCreate(ctx, name, types.NetworkCreate{
		Driver:     "overlay",
		Attachable: true,
	})
	require.NoError(t, err)

	container.Create(t, ctx, client, container.WithName("ng1"), func(c *container.TestContainerConfig) {
		c.NetworkingConfig = &network.NetworkingConfig{
			map[string]*network.EndpointSettings{
				name: {},
			},
		}
	})

	err = client.NetworkConnect(ctx, name, "ng1", &network.EndpointSettings{
		Aliases: []string{
			"aaa",
		},
	})
	require.NoError(t, err)

	err = client.ContainerStart(ctx, "ng1", types.ContainerStartOptions{})
	require.NoError(t, err)

	ng1, err := client.ContainerInspect(ctx, "ng1")
	require.NoError(t, err)
	assert.Equal(t, len(ng1.NetworkSettings.Networks[name].Aliases), 2)
	assert.Equal(t, ng1.NetworkSettings.Networks[name].Aliases[0], "aaa")

	container.Create(t, ctx, client, container.WithName("ng2"), func(c *container.TestContainerConfig) {
		c.NetworkingConfig = &network.NetworkingConfig{
			map[string]*network.EndpointSettings{
				name: {},
			},
		}
	})

	err = client.NetworkConnect(ctx, name, "ng2", &network.EndpointSettings{
		Aliases: []string{
			"bbb",
		},
	})
	require.NoError(t, err)

	err = client.ContainerStart(ctx, "ng2", types.ContainerStartOptions{})
	require.NoError(t, err)

	ng2, err := client.ContainerInspect(ctx, "ng2")
	require.NoError(t, err)
	assert.Equal(t, len(ng2.NetworkSettings.Networks[name].Aliases), 2)
	assert.Equal(t, ng2.NetworkSettings.Networks[name].Aliases[0], "bbb")
}