updated tests

Author: CamrynCarter

cmd/hauler/cli/store/add_test.go

@@ -368,6 +368,108 @@ func TestRewriteReference(t *testing.T) {
 		// condition fires → registry reverts to host, no library/ to strip
 		assertAnnotationsInStore(t, s, "newrepo/img:v2", host+"/newrepo/img:v2")
 	})
+
+	// The library/-detection must look at rawRewrite's path (not the
+	// go-containerregistry-normalized newRepo, which always carries "library/" for
+	// single-segment repos), so that a rewrite which explicitly asks for
+	// "library/..." is honored instead of being unconditionally stripped.
+
+	t.Run("path-only rewrite with explicit library/ prefix is preserved", func(t *testing.T) {
+		s := newTestStore(t)
+		seedStoreDescriptor(t, s, map[string]string{
+			ocispec.AnnotationRefName:     "library/nginx:latest",
+			consts.ContainerdImageNameKey: "index.docker.io/library/nginx:latest",
+		})
+
+		oldRef, _ := name.NewTag("nginx:latest")
+		newRef, _ := name.NewTag("library/nginx:v2")
+		rawRewrite := "library/nginx:v2"
+
+		if err := rewriteReference(ctx, s, oldRef, newRef, rawRewrite); err != nil {
+			t.Fatalf("rewriteReference: %v", err)
+		}
+		// rewriteRepo (derived from rawRewrite) starts with "library/" → must be kept
+		assertAnnotationsInStore(t, s, "library/nginx:v2", "index.docker.io/library/nginx:v2")
+	})
+
+	t.Run("leading slash rewrite with explicit library/ prefix is preserved", func(t *testing.T) {
+		s := newTestStore(t)
+		seedStoreDescriptor(t, s, map[string]string{
+			ocispec.AnnotationRefName:     "library/nginx:latest",
+			consts.ContainerdImageNameKey: "index.docker.io/library/nginx:latest",
+		})
+
+		oldRef, _ := name.NewTag("nginx:latest")
+		newRef, _ := name.NewTag("library/nginx:v2")
+		// AddImageCmd passes the pre-trim rewrite string through as rawRewrite, so a
+		// leading "/" must still be handled correctly here.
+		rawRewrite := "/library/nginx:v2"
+
+		if err := rewriteReference(ctx, s, oldRef, newRef, rawRewrite); err != nil {
+			t.Fatalf("rewriteReference: %v", err)
+		}
+		assertAnnotationsInStore(t, s, "library/nginx:v2", "index.docker.io/library/nginx:v2")
+	})
+}
+
+func TestRewriteChartReference(t *testing.T) {
+	ctx := newTestContext(t)
+
+	// A chart rewritten to a bare single-segment name must not keep an erroneous
+	// "library/" prefix picked up from go-containerregistry's docker hub
+	// normalization, unless the rewrite explicitly asked for one.
+
+	t.Run("path-only rewrite strips library/ prefix from docker hub normalization", func(t *testing.T) {
+		s := newTestStore(t)
+		seedStoreDescriptor(t, s, map[string]string{
+			ocispec.AnnotationRefName: "library/mychart:1.0.0",
+		})
+
+		ref, _ := name.NewTag("mychart:1.0.0")
+		if err := rewriteChartReference(ctx, s, ref, "mychart:2.0.0"); err != nil {
+			t.Fatalf("rewriteChartReference: %v", err)
+		}
+		assertArtifactInStore(t, s, "mychart:2.0.0")
+	})
+
+	t.Run("explicit library/ prefix in rewrite is preserved", func(t *testing.T) {
+		s := newTestStore(t)
+		seedStoreDescriptor(t, s, map[string]string{
+			ocispec.AnnotationRefName: "library/mychart:1.0.0",
+		})
+
+		ref, _ := name.NewTag("mychart:1.0.0")
+		if err := rewriteChartReference(ctx, s, ref, "library/mychart:2.0.0"); err != nil {
+			t.Fatalf("rewriteChartReference: %v", err)
+		}
+		assertArtifactInStore(t, s, "library/mychart:2.0.0")
+	})
+
+	t.Run("leading slash rewrite with explicit library/ prefix is preserved", func(t *testing.T) {
+		s := newTestStore(t)
+		seedStoreDescriptor(t, s, map[string]string{
+			ocispec.AnnotationRefName: "library/mychart:1.0.0",
+		})
+
+		ref, _ := name.NewTag("mychart:1.0.0")
+		if err := rewriteChartReference(ctx, s, ref, "/library/mychart:2.0.0"); err != nil {
+			t.Fatalf("rewriteChartReference: %v", err)
+		}
+		assertArtifactInStore(t, s, "library/mychart:2.0.0")
+	})
+
+	t.Run("rewrite omitting tag inherits the source tag", func(t *testing.T) {
+		s := newTestStore(t)
+		seedStoreDescriptor(t, s, map[string]string{
+			ocispec.AnnotationRefName: "library/mychart:1.0.0",
+		})
+
+		ref, _ := name.NewTag("mychart:1.0.0")
+		if err := rewriteChartReference(ctx, s, ref, "myneworg/mychart"); err != nil {
+			t.Fatalf("rewriteChartReference: %v", err)
+		}
+		assertArtifactInStore(t, s, "myneworg/mychart:1.0.0")
+	})
 }
 
 // --------------------------------------------------------------------------

cmd/hauler/cli/store/info_test.go

@@ -215,6 +215,82 @@ func TestNewItem(t *testing.T) {
 	}
 }
 
+func TestResolveDisplayReference(t *testing.T) {
+	// ContainerdImageNameKey already holds the fully-qualified reference exactly as
+	// computed by rewriteReference (see add.go), so it must be returned verbatim.
+	// Re-parsing it through the reference package would re-trigger
+	// go-containerregistry's docker hub "library/" normalization for a
+	// single-segment repo, undoing a rewrite like "hello-world-custom" back to
+	// "library/hello-world-custom".
+	t.Run("ContainerdImageNameKey is used verbatim, without re-injecting library/", func(t *testing.T) {
+		desc := ocispec.Descriptor{
+			Annotations: map[string]string{
+				consts.ContainerdImageNameKey: "index.docker.io/hello-world-custom:v2",
+				ocispec.AnnotationRefName:     "hello-world-custom:v2",
+			},
+		}
+		got, err := resolveDisplayReference(desc)
+		if err != nil {
+			t.Fatalf("resolveDisplayReference: %v", err)
+		}
+		if want := "index.docker.io/hello-world-custom:v2"; got != want {
+			t.Errorf("got %q, want %q", got, want)
+		}
+	})
+
+	t.Run("falls back to AnnotationRefName parsed when ContainerdImageNameKey absent", func(t *testing.T) {
+		desc := ocispec.Descriptor{
+			Annotations: map[string]string{
+				ocispec.AnnotationRefName: "hello-world-custom:v2",
+			},
+		}
+		got, err := resolveDisplayReference(desc)
+		if err != nil {
+			t.Fatalf("resolveDisplayReference: %v", err)
+		}
+		if want := "hauler/hello-world-custom:v2"; got != want {
+			t.Errorf("got %q, want %q", got, want)
+		}
+	})
+
+	t.Run("returns error when fallback ref cannot be parsed", func(t *testing.T) {
+		desc := ocispec.Descriptor{Annotations: map[string]string{}}
+		if _, err := resolveDisplayReference(desc); err == nil {
+			t.Fatal("expected error, got nil")
+		}
+	})
+}
+
+func TestNewItem_ReferenceUsesContainerdImageNameVerbatim(t *testing.T) {
+	desc := ocispec.Descriptor{
+		Annotations: map[string]string{
+			consts.ContainerdImageNameKey: "index.docker.io/hello-world-custom:v2",
+			ocispec.AnnotationRefName:     "hello-world-custom:v2",
+		},
+	}
+	m := ocispec.Manifest{Config: ocispec.Descriptor{MediaType: consts.DockerConfigJSON}}
+	o := &flags.InfoOpts{TypeFilter: "all"}
+
+	got := newItem(nil, desc, m, "linux/amd64", o)
+	if want := "index.docker.io/hello-world-custom:v2"; got.Reference != want {
+		t.Errorf("got Reference %q, want %q", got.Reference, want)
+	}
+}
+
+func TestFallbackItem_ReferenceUsesContainerdImageNameVerbatim(t *testing.T) {
+	desc := ocispec.Descriptor{
+		Annotations: map[string]string{
+			consts.ContainerdImageNameKey: "index.docker.io/hello-world-custom:v2",
+			ocispec.AnnotationRefName:     "hello-world-custom:v2",
+		},
+	}
+
+	got := fallbackItem(desc, "linux/amd64", store.BlobResult{})
+	if want := "index.docker.io/hello-world-custom:v2"; got.Reference != want {
+		t.Errorf("got Reference %q, want %q", got.Reference, want)
+	}
+}
+
 func TestInfoCmd(t *testing.T) {
 	ctx := newTestContext(t)
 	s := newTestStore(t)