mirror of https://github.com/go-gitea/gitea.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
2.7 KiB
82 lines
2.7 KiB
// 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) |
|
}
|
|
|