From a27218112557e0c92da42c21964193411313028b Mon Sep 17 00:00:00 2001
From: Antoine Gelloz <antoine.gelloz@radiofrance.com>
Date: Tue, 8 Oct 2024 11:13:26 +0200
Subject: [PATCH] refact(buildGraph): new internal test to check the
 assignement of files to nodes

---
 pkg/dib/generate_dag_internal_test.go | 42 +++++++++++++++++++++++++--
 1 file changed, 40 insertions(+), 2 deletions(-)

diff --git a/pkg/dib/generate_dag_internal_test.go b/pkg/dib/generate_dag_internal_test.go
index 205d8052..7e7ab626 100644
--- a/pkg/dib/generate_dag_internal_test.go
+++ b/pkg/dib/generate_dag_internal_test.go
@@ -1,9 +1,47 @@
 package dib
 
-import "testing"
+import (
+	"testing"
+
+	"github.com/davecgh/go-spew/spew"
+	"github.com/radiofrance/dib/pkg/dag"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+const (
+	buildPath      = "../../test/fixtures/docker"
+	registryPrefix = "eu.gcr.io/my-test-repository"
+)
 
 func Test_buildGraph(t *testing.T) {
 	t.Parallel()
 
-	// Implement me.
+	graph, err := buildGraph(buildPath, registryPrefix)
+	require.NoError(t, err)
+	graph.WalkInDepth(func(node *dag.Node) {
+		switch node.Image.ShortName {
+		case "bullseye":
+			require.Len(t, node.Files, 5, spew.Sdump(node.Files))
+			assert.Contains(t, node.Files, buildPath+"/bullseye/Dockerfile")
+			assert.Contains(t, node.Files, buildPath+"/bullseye/external-parent/Dockerfile")
+			assert.Contains(t, node.Files, buildPath+"/bullseye/multistage/Dockerfile")
+			assert.Contains(t, node.Files, buildPath+"/bullseye/skipbuild/Dockerfile")
+			assert.Contains(t, node.Files, buildPath+"/bullseye/sub-image/Dockerfile")
+		case "kaniko":
+			require.Len(t, node.Files, 1, spew.Sdump(node.Files))
+			assert.Contains(t, node.Files, buildPath+"/bullseye/external-parent/Dockerfile")
+		case "multistage":
+			require.Len(t, node.Files, 1, spew.Sdump(node.Files))
+			assert.Contains(t, node.Files, buildPath+"/bullseye/multistage/Dockerfile")
+		case "skipbuild":
+			require.Len(t, node.Files, 1, spew.Sdump(node.Files))
+			assert.Contains(t, node.Files, buildPath+"/bullseye/skipbuild/Dockerfile")
+		case "sub-image":
+			require.Len(t, node.Files, 1, spew.Sdump(node.Files))
+			assert.Contains(t, node.Files, buildPath+"/bullseye/sub-image/Dockerfile")
+		default:
+			t.Errorf("unexpected image: %s", node.Image.ShortName)
+		}
+	})
 }