package gui

import (
	"sort"
	"testing"

	"github.com/docker/docker/api/types"
	"github.com/jesseduffield/lazydocker/pkg/commands"
	"github.com/stretchr/testify/assert"
)

func sampleContainers() []*commands.Container {
	return []*commands.Container{
		{
			ID:   "1",
			Name: "1",
			Container: types.Container{
				State: "exited",
			},
		},
		{
			ID:   "2",
			Name: "2",
			Container: types.Container{
				State: "running",
			},
		},
		{
			ID:   "3",
			Name: "3",
			Container: types.Container{
				State: "running",
			},
		},
		{
			ID:   "4",
			Name: "4",
			Container: types.Container{
				State: "created",
			},
		},
	}
}

func expectedPerStatusContainers() []*commands.Container {
	return []*commands.Container{
		{
			ID:   "2",
			Name: "2",
			Container: types.Container{
				State: "running",
			},
		},
		{
			ID:   "3",
			Name: "3",
			Container: types.Container{
				State: "running",
			},
		},
		{
			ID:   "1",
			Name: "1",
			Container: types.Container{
				State: "exited",
			},
		},
		{
			ID:   "4",
			Name: "4",
			Container: types.Container{
				State: "created",
			},
		},
	}
}

func expectedLegacySortedContainers() []*commands.Container {
	return []*commands.Container{
		{
			ID:   "1",
			Name: "1",
			Container: types.Container{
				State: "exited",
			},
		},
		{
			ID:   "2",
			Name: "2",
			Container: types.Container{
				State: "running",
			},
		},
		{
			ID:   "3",
			Name: "3",
			Container: types.Container{
				State: "running",
			},
		},
		{
			ID:   "4",
			Name: "4",
			Container: types.Container{
				State: "created",
			},
		},
	}
}

func assertEqualContainers(t *testing.T, left *commands.Container, right *commands.Container) {
	t.Helper()
	assert.Equal(t, left.Container.State, right.Container.State)
	assert.Equal(t, left.Container.ID, right.Container.ID)
	assert.Equal(t, left.Name, right.Name)
}

func TestSortContainers(t *testing.T) {
	actual := sampleContainers()

	expected := expectedPerStatusContainers()

	sort.Slice(actual, func(i, j int) bool {
		return sortContainers(actual[i], actual[j], false)
	})

	assert.Equal(t, len(actual), len(expected))

	for i := 0; i < len(actual); i++ {
		assertEqualContainers(t, expected[i], actual[i])
	}
}

func TestLegacySortedContainers(t *testing.T) {
	actual := sampleContainers()

	expected := expectedLegacySortedContainers()

	sort.Slice(actual, func(i, j int) bool {
		return sortContainers(actual[i], actual[j], true)
	})

	assert.Equal(t, len(actual), len(expected))

	for i := 0; i < len(actual); i++ {
		assertEqualContainers(t, expected[i], actual[i])
	}
}