package graph

import (
	"testing"

	"github.com/stretchr/testify/require"

	"edge-infra.dev/pkg/f8n/warehouse/oci"
)

func TestDot(t *testing.T) {
	t.Parallel()

	tcs := map[string]struct {
		a              oci.Artifact
		depth          int
		providerLabels bool
	}{
		"shoot_0":                 {shoot, 0, false},
		"shoot_1":                 {shoot, 1, false},
		"shoot_2":                 {shoot, 2, false},
		"shoot_3":                 {shoot, 3, false},
		"shoot_0_provider_labels": {shoot, 0, true},
		"shoot_1_provider_labels": {shoot, 1, true},
		"shoot_2_provider_labels": {shoot, 2, true},
		"shoot_3_provider_labels": {shoot, 3, true},
	}

	for name, tc := range tcs {
		tc := tc
		t.Run(name, func(t *testing.T) {
			t.Parallel()

			g, err := buildGraph(tc.a, tc.depth)
			require.NoError(t, err)

			_, err = g.toDot()
			require.NoError(t, err)
		})
	}
}