From cfe8f19986d2c11ca98e2aa185d05e5f5b2cf7e4 Mon Sep 17 00:00:00 2001
From: Philip Laine <philip.laine@gmail.com>
Date: Mon, 6 May 2024 17:17:44 +0200
Subject: [PATCH] test: add tests for remove copies from components to enable
 refactoring (#2473)

## Description

This change adds tests for remove copies from components to enable
refactoring in the future.

## Related Issue

Relates to #2337

## Checklist before merging

- [x] Test, docs, adr added or updated as needed
- [x] [Contributor Guide
Steps](https://github.com/defenseunicorns/zarf/blob/main/.github/CONTRIBUTING.md#developer-workflow)
followed

Co-authored-by: razzle <razzle@defenseunicorns.com>
Co-authored-by: Austin Abro <37223396+AustinAbro321@users.noreply.github.com>
---
 src/pkg/packager/creator/differential_test.go | 66 +++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100644 src/pkg/packager/creator/differential_test.go

diff --git a/src/pkg/packager/creator/differential_test.go b/src/pkg/packager/creator/differential_test.go
new file mode 100644
index 0000000000..cc616f92ed
--- /dev/null
+++ b/src/pkg/packager/creator/differential_test.go
@@ -0,0 +1,66 @@
+// SPDX-License-Identifier: Apache-2.0
+// SPDX-FileCopyrightText: 2021-Present The Zarf Authors
+
+package creator
+
+import (
+	"testing"
+
+	"github.com/defenseunicorns/zarf/src/types"
+	"github.com/stretchr/testify/require"
+)
+
+func TestRemoveCopiesFromComponents(t *testing.T) {
+	components := []types.ZarfComponent{
+		{
+			Images: []string{
+				"example.com/include-image-tag:latest",
+				"example.com/image-with-tag:v1",
+				"example.com/diff-image-with-tag:v1",
+				"example.com/image-with-digest@sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
+				"example.com/diff-image-with-digest@sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
+				"example.com/image-with-tag-and-digest:v1@sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
+				"example.com/diff-image-with-tag-and-digest:v1@sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
+			},
+			Repos: []string{
+				"https://example.com/no-ref.git",
+				"https://example.com/branch.git@refs/heads/main",
+				"https://example.com/tag.git@v1",
+				"https://example.com/diff-tag.git@v1",
+				"https://example.com/commit.git@524980951ff16e19dc25232e9aea8fd693989ba6",
+				"https://example.com/diff-commit.git@524980951ff16e19dc25232e9aea8fd693989ba6",
+			},
+		},
+	}
+	loadedDiffData := types.DifferentialData{
+		DifferentialImages: map[string]bool{
+			"example.com/include-image-tag:latest": true,
+			"example.com/diff-image-with-tag:v1":   true,
+			"example.com/diff-image-with-digest@sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855":            true,
+			"example.com/diff-image-with-tag-and-digest:v1@sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855": true,
+		},
+		DifferentialRepos: map[string]bool{
+			"https://example.com/no-ref.git":                                               true,
+			"https://example.com/branch.git@refs/heads/main":                               true,
+			"https://example.com/diff-tag.git@v1":                                          true,
+			"https://example.com/diff-commit.git@524980951ff16e19dc25232e9aea8fd693989ba6": true,
+		},
+	}
+	diffComponents, err := removeCopiesFromComponents(components, &loadedDiffData)
+	require.NoError(t, err)
+
+	expectedImages := []string{
+		"example.com/include-image-tag:latest",
+		"example.com/image-with-tag:v1",
+		"example.com/image-with-digest@sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
+		"example.com/image-with-tag-and-digest:v1@sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
+	}
+	require.ElementsMatch(t, expectedImages, diffComponents[0].Images)
+	expectedRepos := []string{
+		"https://example.com/no-ref.git",
+		"https://example.com/branch.git@refs/heads/main",
+		"https://example.com/tag.git@v1",
+		"https://example.com/commit.git@524980951ff16e19dc25232e9aea8fd693989ba6",
+	}
+	require.ElementsMatch(t, expectedRepos, diffComponents[0].Repos)
+}