ci(ut): add ut for test coverage

Signed-off-by: imeoer <yansong.ys@antgroup.com>

Author: imeoer

contrib/nydusify/pkg/backend/registry_test.go

@@ -0,0 +1,72 @@
+package backend
+
+import (
+	"context"
+	"io"
+	"os"
+	"reflect"
+	"testing"
+
+	"github.com/agiledragon/gomonkey/v2"
+	"github.com/opencontainers/go-digest"
+	ocispec "github.com/opencontainers/image-spec/specs-go/v1"
+	"github.com/stretchr/testify/require"
+
+	"github.com/dragonflyoss/nydus/contrib/nydusify/pkg/remote"
+	"github.com/dragonflyoss/nydus/contrib/nydusify/pkg/utils"
+)
+
+func TestRegistryUpload(t *testing.T) {
+	tmp, err := os.CreateTemp(t.TempDir(), "blob-*")
+	require.NoError(t, err)
+	_, err = tmp.WriteString("blob-data")
+	require.NoError(t, err)
+	require.NoError(t, tmp.Close())
+
+	registry := &Registry{remote: &remote.Remote{}}
+	patches := gomonkey.ApplyMethod(reflect.TypeOf(&remote.Remote{}), "Push", func(_ *remote.Remote, _ context.Context, desc ocispec.Descriptor, byDigest bool, reader io.Reader) error {
+		require.True(t, byDigest)
+		require.Equal(t, utils.MediaTypeNydusBlob, desc.MediaType)
+		require.Equal(t, digest.Digest("sha256:205eed24cbec29ad9cb4593a73168ef1803402370a82f7d51ce25646fc2f943a"), desc.Digest)
+		content, err := io.ReadAll(reader)
+		require.NoError(t, err)
+		require.Equal(t, "blob-data", string(content))
+		return nil
+	})
+	defer patches.Reset()
+
+	desc, err := registry.Upload(context.Background(), "205eed24cbec29ad9cb4593a73168ef1803402370a82f7d51ce25646fc2f943a", tmp.Name(), 8, false)
+	require.NoError(t, err)
+	require.Equal(t, int64(8), desc.Size)
+	require.Equal(t, utils.MediaTypeNydusBlob, desc.MediaType)
+	assertAnnotations := map[string]string{
+		utils.LayerAnnotationUncompressed: "sha256:205eed24cbec29ad9cb4593a73168ef1803402370a82f7d51ce25646fc2f943a",
+		utils.LayerAnnotationNydusBlob:    "true",
+	}
+	require.Equal(t, assertAnnotations, desc.Annotations)
+}
+
+func TestRegistryUploadFailuresAndHelpers(t *testing.T) {
+	registry := &Registry{remote: &remote.Remote{}}
+
+	_, err := registry.Upload(context.Background(), "205eed24cbec29ad9cb4593a73168ef1803402370a82f7d51ce25646fc2f943a", "/non-existent", 1, false)
+	require.ErrorContains(t, err, "Open blob file")
+
+	tmp, err := os.CreateTemp(t.TempDir(), "blob-*")
+	require.NoError(t, err)
+	require.NoError(t, tmp.Close())
+
+	patches := gomonkey.ApplyMethod(reflect.TypeOf(&remote.Remote{}), "Push", func(_ *remote.Remote, _ context.Context, _ ocispec.Descriptor, _ bool, _ io.Reader) error {
+		return io.EOF
+	})
+	defer patches.Reset()
+
+	_, err = registry.Upload(context.Background(), "205eed24cbec29ad9cb4593a73168ef1803402370a82f7d51ce25646fc2f943a", tmp.Name(), 0, false)
+	require.ErrorContains(t, err, "Push blob layer")
+
+	require.NoError(t, registry.Finalize(false))
+	ok, err := registry.Check("ignored")
+	require.NoError(t, err)
+	require.True(t, ok)
+	require.Equal(t, RegistryBackend, registry.Type())
+}

contrib/nydusify/pkg/build/builder_test.go

@@ -0,0 +1,181 @@
+// Copyright 2026 Nydus Developers. All rights reserved.
+//
+// SPDX-License-Identifier: Apache-2.0
+
+package build
+
+import (
+	"bytes"
+	"os"
+	"path/filepath"
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func createFakeBinary(t *testing.T) (string, string, string) {
+	t.Helper()
+
+	tempDir := t.TempDir()
+	argsPath := filepath.Join(tempDir, "args.txt")
+	stdinPath := filepath.Join(tempDir, "stdin.txt")
+	binaryPath := filepath.Join(tempDir, "fake-nydus-image.sh")
+
+	script := "#!/bin/sh\n" +
+		"printf '%s\\n' \"$@\" > \"$NYDUS_ARGS_FILE\"\n" +
+		"cat > \"$NYDUS_STDIN_FILE\"\n"
+
+	require.NoError(t, os.WriteFile(binaryPath, []byte(script), 0o755))
+	t.Setenv("NYDUS_ARGS_FILE", argsPath)
+	t.Setenv("NYDUS_STDIN_FILE", stdinPath)
+
+	return binaryPath, argsPath, stdinPath
+}
+
+func readLines(t *testing.T, path string) []string {
+	t.Helper()
+
+	content, err := os.ReadFile(path)
+	require.NoError(t, err)
+	text := strings.TrimSpace(string(content))
+	if text == "" {
+		return nil
+	}
+
+	return strings.Split(text, "\n")
+}
+
+func TestNewBuilderUsesStdStreams(t *testing.T) {
+	builder := NewBuilder("/usr/bin/nydus-image")
+	require.Equal(t, "/usr/bin/nydus-image", builder.binaryPath)
+	require.Equal(t, os.Stdout, builder.stdout)
+	require.Equal(t, os.Stderr, builder.stderr)
+}
+
+func TestBuilderRunBuildCommand(t *testing.T) {
+	binaryPath, argsPath, stdinPath := createFakeBinary(t)
+	builder := NewBuilder(binaryPath)
+	builder.stdout = &bytes.Buffer{}
+	builder.stderr = &bytes.Buffer{}
+
+	err := builder.Run(BuilderOption{
+		ParentBootstrapPath: "/tmp/parent.boot",
+		ChunkDict:           "/tmp/chunk.dict",
+		BootstrapPath:       "/tmp/bootstrap.boot",
+		RootfsPath:          "/tmp/rootfs",
+		WhiteoutSpec:        "overlayfs",
+		OutputJSONPath:      "/tmp/output.json",
+		PrefetchPatterns:    "/etc\n/usr/bin",
+		BlobPath:            "/tmp/blob.data",
+		AlignedChunk:        true,
+		Compressor:          "zstd",
+		ChunkSize:           "0x200000",
+		FsVersion:           "6",
+	})
+	require.NoError(t, err)
+
+	require.Equal(t, []string{
+		"create",
+		"--parent-bootstrap",
+		"/tmp/parent.boot",
+		"--aligned-chunk",
+		"--chunk-dict",
+		"/tmp/chunk.dict",
+		"--bootstrap",
+		"/tmp/bootstrap.boot",
+		"--log-level",
+		"warn",
+		"--whiteout-spec",
+		"overlayfs",
+		"--output-json",
+		"/tmp/output.json",
+		"--blob",
+		"/tmp/blob.data",
+		"--fs-version",
+		"6",
+		"--compressor",
+		"zstd",
+		"--prefetch-policy",
+		"fs",
+		"--chunk-size",
+		"0x200000",
+		"/tmp/rootfs",
+	}, readLines(t, argsPath))
+
+	stdinContent, err := os.ReadFile(stdinPath)
+	require.NoError(t, err)
+	require.Equal(t, "/etc\n/usr/bin", string(stdinContent))
+}
+
+func TestBuilderCompactAndGenerateCommands(t *testing.T) {
+	binaryPath, argsPath, stdinPath := createFakeBinary(t)
+	builder := NewBuilder(binaryPath)
+	builder.stdout = &bytes.Buffer{}
+	builder.stderr = &bytes.Buffer{}
+
+	err := builder.Compact(CompactOption{
+		ChunkDict:           "/tmp/chunk.dict",
+		BootstrapPath:       "/tmp/bootstrap.boot",
+		OutputBootstrapPath: "/tmp/output.boot",
+		BackendType:         "oss",
+		BackendConfigPath:   "/tmp/backend.json",
+		OutputJSONPath:      "/tmp/output.json",
+		MinUsedRatio:        "30",
+		CompactBlobSize:     "1048576",
+		MaxCompactSize:      "2097152",
+		LayersToCompact:     "2",
+		BlobsDir:            "/tmp/blobs",
+	})
+	require.NoError(t, err)
+	require.Equal(t, []string{
+		"compact",
+		"--bootstrap",
+		"/tmp/bootstrap.boot",
+		"--blob-dir",
+		"/tmp/blobs",
+		"--min-used-ratio",
+		"30",
+		"--compact-blob-size",
+		"1048576",
+		"--max-compact-size",
+		"2097152",
+		"--layers-to-compact",
+		"2",
+		"--backend-type",
+		"oss",
+		"--backend-config-file",
+		"/tmp/backend.json",
+		"--log-level",
+		"info",
+		"--output-json",
+		"/tmp/output.json",
+		"--output-bootstrap",
+		"/tmp/output.boot",
+		"--chunk-dict",
+		"/tmp/chunk.dict",
+	}, readLines(t, argsPath))
+	require.Empty(t, readLines(t, stdinPath))
+
+	err = builder.Generate(GenerateOption{
+		BootstrapPaths:         []string{"/tmp/layer1.boot", "/tmp/layer2.boot"},
+		DatabasePath:           "/tmp/chunk.db",
+		ChunkdictBootstrapPath: "/tmp/chunkdict.boot",
+		OutputPath:             "/tmp/chunkdict.json",
+	})
+	require.NoError(t, err)
+	require.Equal(t, []string{
+		"chunkdict",
+		"generate",
+		"--log-level",
+		"warn",
+		"--bootstrap",
+		"/tmp/chunkdict.boot",
+		"--database",
+		"/tmp/chunk.db",
+		"--output-json",
+		"/tmp/chunkdict.json",
+		"/tmp/layer1.boot",
+		"/tmp/layer2.boot",
+	}, readLines(t, argsPath))
+}