chore: add test to ensure the package versions are added correctly

pkg/anchor/docker_test.go

@@ -1,7 +1,9 @@
 package anchor
 
 import (
+	"fmt"
 	"reflect"
+	"strings"
 	"testing"
 )
 
@@ -13,3 +15,43 @@ func TestParseCommand(t *testing.T) {
 		t.Errorf("Expected %v but got %v", expected, actual)
 	}
 }
+
+func TestAppendPackageVersions(t *testing.T) {
+	file := `# hadolint ignore=DL3008
+RUN apt-get update \
+  && apt-get install \
+    --no-install-recommends -y \
+    # We just need curl and wget
+    curl wget \
+  && rm -rf /var/lib/apt/lists/* \
+  && apt-get clean`
+	input := strings.NewReader(file)
+	nodes := Parse(input)
+	architecture := "amd64"
+
+	packageMap := map[string]string{
+		"curl": "7.68.0",
+		"wget": "1.20.3",
+	}
+
+	expected := fmt.Sprintf(`# hadolint ignore=DL3008
+RUN dpkg --add-architecture %s && apt-get update && apt-get update \
+  && apt-get install \
+    --no-install-recommends -y \
+    # We just need curl and wget
+    curl=%s wget=%s \
+  && rm -rf /var/lib/apt/lists/* \
+  && apt-get clean
+`, architecture, packageMap["curl"], packageMap["wget"])
+
+	node := nodes[0]
+	appendPackageVersions(&node, packageMap, architecture)
+	nodes[0] = node
+
+	w := &strings.Builder{}
+	nodes.Write(w)
+	result := w.String()
+	if result != expected {
+		t.Errorf("Expected:\n%v\ngot:\n%v", expected, result)
+	}
+}

pkg/anchor/packages.go

@@ -56,8 +56,8 @@ func parsePackageVersions(s string) (map[string]string, error) {
 			}
 
 			if _, ok := versions[currentPackage]; ok {
-        // We have already seen this package, so we can skip it
-        currentPackage = ""
+				// We have already seen this package, so we can skip it
+				currentPackage = ""
 				continue
 			}
 			versions[currentPackage] = strings.Split(line, ": ")[1]

pkg/anchor/parser_test.go

@@ -143,7 +143,12 @@ RUN apt-get update \
 			input := strings.NewReader(tc.input)
 			result := Parse(input)
 			if len(result) != len(tc.expected) {
-				t.Fatalf("%s: Different lengths, expected:\n%v\n got:\n%v", tc.name, tc.expected, result)
+				t.Fatalf(
+					"%s: Different lengths, expected:\n%v\n got:\n%v",
+					tc.name,
+					tc.expected,
+					result,
+				)
 			}
 			for i := range result {
 				if tc.expected[i].Command != result[i].Command {