|
| 1 | +// Copyright 2023 The Gitea Authors. All rights reserved. |
| 2 | +// SPDX-License-Identifier: MIT |
| 3 | + |
| 4 | +package repository |
| 5 | + |
| 6 | +import ( |
| 7 | + "bytes" |
| 8 | + "context" |
| 9 | + "testing" |
| 10 | + "time" |
| 11 | + |
| 12 | + "code.gitea.io/gitea/models/db" |
| 13 | + git_model "code.gitea.io/gitea/models/git" |
| 14 | + repo_model "code.gitea.io/gitea/models/repo" |
| 15 | + "code.gitea.io/gitea/models/unittest" |
| 16 | + "code.gitea.io/gitea/modules/lfs" |
| 17 | + "code.gitea.io/gitea/modules/setting" |
| 18 | + "code.gitea.io/gitea/modules/storage" |
| 19 | + |
| 20 | + "github.com/stretchr/testify/assert" |
| 21 | +) |
| 22 | + |
| 23 | +func TestGarbageCollectLFSMetaObjects(t *testing.T) { |
| 24 | + assert.NoError(t, unittest.PrepareTestDatabase()) |
| 25 | + |
| 26 | + setting.LFS.StartServer = true |
| 27 | + err := storage.Init() |
| 28 | + assert.NoError(t, err) |
| 29 | + |
| 30 | + repo, err := repo_model.GetRepositoryByOwnerAndName(db.DefaultContext, "user2", "repo1") |
| 31 | + assert.NoError(t, err) |
| 32 | + |
| 33 | + // add lfs object |
| 34 | + lfsContent := []byte("gitea1") |
| 35 | + lfsOid := storeObjectInRepo(t, repo.ID, &lfsContent) |
| 36 | + |
| 37 | + // gc |
| 38 | + err = GarbageCollectLFSMetaObjects(context.Background(), GarbageCollectLFSMetaObjectsOptions{ |
| 39 | + AutoFix: true, |
| 40 | + OlderThan: time.Now().Add(7 * 24 * time.Hour).Add(5 * 24 * time.Hour), |
| 41 | + UpdatedLessRecentlyThan: time.Now().Add(7 * 24 * time.Hour).Add(3 * 24 * time.Hour), |
| 42 | + }) |
| 43 | + assert.NoError(t, err) |
| 44 | + |
| 45 | + // lfs meta has been deleted |
| 46 | + _, err = git_model.GetLFSMetaObjectByOid(db.DefaultContext, repo.ID, lfsOid) |
| 47 | + assert.ErrorIs(t, err, git_model.ErrLFSObjectNotExist) |
| 48 | +} |
| 49 | + |
| 50 | +func storeObjectInRepo(t *testing.T, repositoryID int64, content *[]byte) string { |
| 51 | + pointer, err := lfs.GeneratePointer(bytes.NewReader(*content)) |
| 52 | + assert.NoError(t, err) |
| 53 | + |
| 54 | + _, err = git_model.NewLFSMetaObject(db.DefaultContext, &git_model.LFSMetaObject{Pointer: pointer, RepositoryID: repositoryID}) |
| 55 | + assert.NoError(t, err) |
| 56 | + contentStore := lfs.NewContentStore() |
| 57 | + exist, err := contentStore.Exists(pointer) |
| 58 | + assert.NoError(t, err) |
| 59 | + if !exist { |
| 60 | + err := contentStore.Put(pointer, bytes.NewReader(*content)) |
| 61 | + assert.NoError(t, err) |
| 62 | + } |
| 63 | + return pointer.Oid |
| 64 | +} |
0 commit comments