Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/service/backup_archive_test.go

@@ -0,0 +1,62 @@
+//go:build unit
+
+package service
+
+import (
+	"bytes"
+	"crypto/sha256"
+	"fmt"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestSplitBackupFile_ReassemblesExactBytes(t *testing.T) {
+	src := writeBackupArchiveFixture(t, []byte("0123456789abcdefg"))
+	parts, err := splitBackupFile(src, 5)
+	require.NoError(t, err)
+	require.Len(t, parts, 4)
+
+	var got bytes.Buffer
+	for i, part := range parts {
+		require.Equal(t, i+1, part.Index)
+		require.LessOrEqual(t, part.SizeBytes, int64(5))
+		data, readErr := os.ReadFile(part.Path)
+		require.NoError(t, readErr)
+		require.Equal(t, fmt.Sprintf("%x", sha256.Sum256(data)), part.SHA256)
+		got.Write(data)
+	}
+	require.Equal(t, []byte("0123456789abcdefg"), got.Bytes())
+}
+
+func TestSplitBackupFile_RejectsInvalidInput(t *testing.T) {
+	src := writeBackupArchiveFixture(t, []byte("data"))
+
+	_, err := splitBackupFile(src, 0)
+	require.Error(t, err)
+
+	empty := writeBackupArchiveFixture(t, nil)
+	_, err = splitBackupFile(empty, 5)
+	require.Error(t, err)
+
+	_, err = splitBackupFile(filepathForMissingBackupArchive(t), 5)
+	require.Error(t, err)
+}
+
+func writeBackupArchiveFixture(t *testing.T, content []byte) string {
+	t.Helper()
+	path := filepathForBackupArchive(t)
+	require.NoError(t, os.WriteFile(path, content, 0o600))
+	return path
+}
+
+func filepathForBackupArchive(t *testing.T) string {
+	t.Helper()
+	return t.TempDir() + "/archive.gz"
+}
+
+func filepathForMissingBackupArchive(t *testing.T) string {
+	t.Helper()
+	return t.TempDir() + "/missing.gz"
+}