// Copyright 2022 The Gitea Authors. All rights reserved.// SPDX-License-Identifier: MIT package cmd import (	"io"	"os"	"strings"	"testing" 	"forgejo.org/models/actions"	"forgejo.org/models/db"	"forgejo.org/models/packages"	"forgejo.org/models/unittest"	user_model "forgejo.org/models/user"	packages_module "forgejo.org/modules/packages"	"forgejo.org/modules/setting"	"forgejo.org/modules/storage"	"forgejo.org/modules/test"	packages_service "forgejo.org/services/packages" 	"github.com/stretchr/testify/assert"	"github.com/stretchr/testify/require") func createLocalStorage(t *testing.T) (storage.ObjectStorage, string) {	t.Helper() 	p := t.TempDir() 	storage, err := storage.NewLocalStorage(		t.Context(),		&setting.Storage{			Path: p,		})	require.NoError(t, err) 	return storage, p} func TestMigratePackages(t *testing.T) {	require.NoError(t, unittest.PrepareTestDatabase()) 	creator := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}) 	content := "package main\n\nfunc main() {\nfmt.Println(\"hi\")\n}\n"	buf, err := packages_module.CreateHashedBufferFromReaderWithSize(strings.NewReader(content), 1024)	require.NoError(t, err)	defer buf.Close() 	v, f, err := packages_service.CreatePackageAndAddFile(db.DefaultContext, &packages_service.PackageCreationInfo{		PackageInfo: packages_service.PackageInfo{			Owner:       creator,			PackageType: packages.TypeGeneric,			Name:        "test",			Version:     "1.0.0",		},		Creator:           creator,		SemverCompatible:  true,		VersionProperties: map[string]string{},	}, &packages_service.PackageFileCreationInfo{		PackageFileInfo: packages_service.PackageFileInfo{			Filename: "a.go",		},		Creator: creator,		Data:    buf,		IsLead:  true,	})	require.NoError(t, err)	assert.NotNil(t, v)	assert.NotNil(t, f) 	ctx := t.Context() 	dstStorage, p := createLocalStorage(t) 	err = migratePackages(ctx, dstStorage)	require.NoError(t, err) 	entries, err := os.ReadDir(p)	require.NoError(t, err)	assert.Len(t, entries, 2)	assert.Equal(t, "01", entries[0].Name())	assert.Equal(t, "tmp", entries[1].Name())} func TestMigrateActionsArtifacts(t *testing.T) {	require.NoError(t, unittest.PrepareTestDatabase()) 	srcStorage, _ := createLocalStorage(t)	defer test.MockVariableValue(&storage.ActionsArtifacts, srcStorage)()	id := int64(42) 	addArtifact := func(storagePath string, status actions.ArtifactStatus) {		id++		artifact := &actions.ActionArtifact{			ID:           id,			ArtifactName: storagePath,			StoragePath:  storagePath,			Status:       int64(status),		}		_, err := db.GetEngine(db.DefaultContext).Insert(artifact)		require.NoError(t, err)		srcStorage.Save(storagePath, strings.NewReader(storagePath), -1)	} 	exists := "/exists"	addArtifact(exists, actions.ArtifactStatusUploadConfirmed) 	expired := "/expired"	addArtifact(expired, actions.ArtifactStatusExpired) 	notFound := "/notfound"	addArtifact(notFound, actions.ArtifactStatusUploadConfirmed)	srcStorage.Delete(notFound) 	dstStorage, _ := createLocalStorage(t) 	require.NoError(t, migrateActionsArtifacts(db.DefaultContext, dstStorage)) 	object, err := dstStorage.Open(exists)	require.NoError(t, err)	buf, err := io.ReadAll(object)	require.NoError(t, err)	assert.Equal(t, exists, string(buf)) 	_, err = dstStorage.Stat(expired)	require.Error(t, err) 	_, err = dstStorage.Stat(notFound)	require.Error(t, err)}