gitea/models/git/lfs_lock_test.go at eecf538bd6bf9a94a59ff7e9a35bdde794a884ed · GiteaBot/gitea · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
// Copyright 2026 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package git

import (
	"fmt"
	"testing"
	"time"

	repo_model "code.gitea.io/gitea/models/repo"
	"code.gitea.io/gitea/models/unittest"
	user_model "code.gitea.io/gitea/models/user"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func createTestLock(t *testing.T, repo *repo_model.Repository, owner *user_model.User) *LFSLock {
	t.Helper()

	path := fmt.Sprintf("%s-%d-%d", t.Name(), repo.ID, time.Now().UnixNano())
	lock, err := CreateLFSLock(t.Context(), repo, &LFSLock{
		OwnerID: owner.ID,
		Path:    path,
	})
	require.NoError(t, err)
	return lock
}

func TestGetLFSLockByIDAndRepo(t *testing.T) {
	require.NoError(t, unittest.PrepareTestDatabase())

	repo1 := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1})
	repo3 := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 3})
	user2 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
	user4 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 4})

	lockRepo1 := createTestLock(t, repo1, user2)
	lockRepo3 := createTestLock(t, repo3, user4)

	fetched, err := GetLFSLockByIDAndRepo(t.Context(), lockRepo1.ID, repo1.ID)
	require.NoError(t, err)
	assert.Equal(t, lockRepo1.ID, fetched.ID)
	assert.Equal(t, repo1.ID, fetched.RepoID)

	_, err = GetLFSLockByIDAndRepo(t.Context(), lockRepo1.ID, repo3.ID)
	assert.Error(t, err)
	assert.True(t, IsErrLFSLockNotExist(err))

	_, err = GetLFSLockByIDAndRepo(t.Context(), lockRepo3.ID, repo1.ID)
	assert.Error(t, err)
	assert.True(t, IsErrLFSLockNotExist(err))
}

func TestDeleteLFSLockByIDRequiresRepoMatch(t *testing.T) {
	require.NoError(t, unittest.PrepareTestDatabase())

	repo1 := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1})
	repo3 := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 3})
	user2 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
	user4 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 4})

	lockRepo1 := createTestLock(t, repo1, user2)
	lockRepo3 := createTestLock(t, repo3, user4)

	_, err := DeleteLFSLockByID(t.Context(), lockRepo3.ID, repo1, user2, true)
	assert.Error(t, err)
	assert.True(t, IsErrLFSLockNotExist(err))

	existing, err := GetLFSLockByIDAndRepo(t.Context(), lockRepo3.ID, repo3.ID)
	require.NoError(t, err)
	assert.Equal(t, lockRepo3.ID, existing.ID)

	deleted, err := DeleteLFSLockByID(t.Context(), lockRepo3.ID, repo3, user4, true)
	require.NoError(t, err)
	assert.Equal(t, lockRepo3.ID, deleted.ID)

	deleted, err = DeleteLFSLockByID(t.Context(), lockRepo1.ID, repo1, user2, false)
	require.NoError(t, err)
	assert.Equal(t, lockRepo1.ID, deleted.ID)
}