2017-11-28 13:58:37 -07:00
// Copyright 2017 The Gitea Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package integrations
import (
"fmt"
"net/http"
"testing"
"time"
2021-12-09 18:27:50 -07:00
repo_model "code.gitea.io/gitea/models/repo"
2021-11-16 01:53:21 -07:00
"code.gitea.io/gitea/models/unittest"
2021-11-24 02:49:20 -07:00
user_model "code.gitea.io/gitea/models/user"
2021-06-05 17:59:27 -06:00
"code.gitea.io/gitea/modules/lfs"
2017-11-28 13:58:37 -07:00
"code.gitea.io/gitea/modules/setting"
2019-05-11 04:21:34 -06:00
api "code.gitea.io/gitea/modules/structs"
2017-11-28 13:58:37 -07:00
"github.com/stretchr/testify/assert"
)
func TestAPILFSLocksNotStarted ( t * testing . T ) {
2019-11-25 16:21:37 -07:00
defer prepareTestEnv ( t ) ( )
2017-11-28 13:58:37 -07:00
setting . LFS . StartServer = false
2021-11-24 02:49:20 -07:00
user := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 2 } ) . ( * user_model . User )
2021-12-09 18:27:50 -07:00
repo := unittest . AssertExistsAndLoadBean ( t , & repo_model . Repository { ID : 1 } ) . ( * repo_model . Repository )
2017-11-28 13:58:37 -07:00
2017-11-28 16:35:23 -07:00
req := NewRequestf ( t , "GET" , "/%s/%s.git/info/lfs/locks" , user . Name , repo . Name )
2017-11-28 13:58:37 -07:00
MakeRequest ( t , req , http . StatusNotFound )
2017-11-28 16:35:23 -07:00
req = NewRequestf ( t , "POST" , "/%s/%s.git/info/lfs/locks" , user . Name , repo . Name )
2017-11-28 13:58:37 -07:00
MakeRequest ( t , req , http . StatusNotFound )
2017-11-28 16:35:23 -07:00
req = NewRequestf ( t , "GET" , "/%s/%s.git/info/lfs/locks/verify" , user . Name , repo . Name )
2017-11-28 13:58:37 -07:00
MakeRequest ( t , req , http . StatusNotFound )
2017-11-28 16:35:23 -07:00
req = NewRequestf ( t , "GET" , "/%s/%s.git/info/lfs/locks/10/unlock" , user . Name , repo . Name )
2017-11-28 13:58:37 -07:00
MakeRequest ( t , req , http . StatusNotFound )
}
func TestAPILFSLocksNotLogin ( t * testing . T ) {
2019-11-25 16:21:37 -07:00
defer prepareTestEnv ( t ) ( )
2017-11-28 13:58:37 -07:00
setting . LFS . StartServer = true
2021-11-24 02:49:20 -07:00
user := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 2 } ) . ( * user_model . User )
2021-12-09 18:27:50 -07:00
repo := unittest . AssertExistsAndLoadBean ( t , & repo_model . Repository { ID : 1 } ) . ( * repo_model . Repository )
2017-11-28 13:58:37 -07:00
2017-11-28 16:35:23 -07:00
req := NewRequestf ( t , "GET" , "/%s/%s.git/info/lfs/locks" , user . Name , repo . Name )
2021-06-05 17:59:27 -06:00
req . Header . Set ( "Accept" , lfs . MediaType )
2018-01-27 09:48:15 -07:00
resp := MakeRequest ( t , req , http . StatusUnauthorized )
2017-11-28 13:58:37 -07:00
var lfsLockError api . LFSLockError
DecodeJSON ( t , resp , & lfsLockError )
2021-05-15 09:32:09 -06:00
assert . Equal ( t , "You must have pull access to list locks" , lfsLockError . Message )
2017-11-28 13:58:37 -07:00
}
func TestAPILFSLocksLogged ( t * testing . T ) {
2019-11-25 16:21:37 -07:00
defer prepareTestEnv ( t ) ( )
2017-11-28 13:58:37 -07:00
setting . LFS . StartServer = true
2022-01-20 10:46:10 -07:00
user2 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 2 } ) . ( * user_model . User ) // in org 3
user4 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 4 } ) . ( * user_model . User ) // in org 3
2017-11-28 13:58:37 -07:00
2021-12-09 18:27:50 -07:00
repo1 := unittest . AssertExistsAndLoadBean ( t , & repo_model . Repository { ID : 1 } ) . ( * repo_model . Repository )
repo3 := unittest . AssertExistsAndLoadBean ( t , & repo_model . Repository { ID : 3 } ) . ( * repo_model . Repository ) // own by org 3
2017-11-28 13:58:37 -07:00
tests := [ ] struct {
2021-11-24 02:49:20 -07:00
user * user_model . User
2021-12-09 18:27:50 -07:00
repo * repo_model . Repository
2017-11-28 13:58:37 -07:00
path string
httpResult int
addTime [ ] int
} {
{ user : user2 , repo : repo1 , path : "foo/bar.zip" , httpResult : http . StatusCreated , addTime : [ ] int { 0 } } ,
{ user : user2 , repo : repo1 , path : "path/test" , httpResult : http . StatusCreated , addTime : [ ] int { 0 } } ,
{ user : user2 , repo : repo1 , path : "path/test" , httpResult : http . StatusConflict } ,
{ user : user2 , repo : repo1 , path : "Foo/BaR.zip" , httpResult : http . StatusConflict } ,
{ user : user2 , repo : repo1 , path : "Foo/Test/../subFOlder/../Relative/../BaR.zip" , httpResult : http . StatusConflict } ,
2018-01-27 09:48:15 -07:00
{ user : user4 , repo : repo1 , path : "FoO/BaR.zip" , httpResult : http . StatusUnauthorized } ,
{ user : user4 , repo : repo1 , path : "path/test-user4" , httpResult : http . StatusUnauthorized } ,
2017-11-28 13:58:37 -07:00
{ user : user2 , repo : repo1 , path : "patH/Test-user4" , httpResult : http . StatusCreated , addTime : [ ] int { 0 } } ,
{ user : user2 , repo : repo1 , path : "some/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/long/path" , httpResult : http . StatusCreated , addTime : [ ] int { 0 } } ,
{ user : user2 , repo : repo3 , path : "test/foo/bar.zip" , httpResult : http . StatusCreated , addTime : [ ] int { 1 , 2 } } ,
{ user : user4 , repo : repo3 , path : "test/foo/bar.zip" , httpResult : http . StatusConflict } ,
{ user : user4 , repo : repo3 , path : "test/foo/bar.bin" , httpResult : http . StatusCreated , addTime : [ ] int { 1 , 2 } } ,
}
resultsTests := [ ] struct {
2021-11-24 02:49:20 -07:00
user * user_model . User
2021-12-09 18:27:50 -07:00
repo * repo_model . Repository
2017-11-28 13:58:37 -07:00
totalCount int
oursCount int
theirsCount int
2021-11-24 02:49:20 -07:00
locksOwners [ ] * user_model . User
2017-11-28 13:58:37 -07:00
locksTimes [ ] time . Time
} {
2021-11-24 02:49:20 -07:00
{ user : user2 , repo : repo1 , totalCount : 4 , oursCount : 4 , theirsCount : 0 , locksOwners : [ ] * user_model . User { user2 , user2 , user2 , user2 } , locksTimes : [ ] time . Time { } } ,
{ user : user2 , repo : repo3 , totalCount : 2 , oursCount : 1 , theirsCount : 1 , locksOwners : [ ] * user_model . User { user2 , user4 } , locksTimes : [ ] time . Time { } } ,
{ user : user4 , repo : repo3 , totalCount : 2 , oursCount : 1 , theirsCount : 1 , locksOwners : [ ] * user_model . User { user2 , user4 } , locksTimes : [ ] time . Time { } } ,
2017-11-28 13:58:37 -07:00
}
deleteTests := [ ] struct {
2021-11-24 02:49:20 -07:00
user * user_model . User
2021-12-09 18:27:50 -07:00
repo * repo_model . Repository
2017-11-28 13:58:37 -07:00
lockID string
} { }
2022-01-20 10:46:10 -07:00
// create locks
2017-11-28 13:58:37 -07:00
for _ , test := range tests {
session := loginUser ( t , test . user . Name )
2017-11-28 16:35:23 -07:00
req := NewRequestWithJSON ( t , "POST" , fmt . Sprintf ( "/%s.git/info/lfs/locks" , test . repo . FullName ( ) ) , map [ string ] string { "path" : test . path } )
2021-06-05 17:59:27 -06:00
req . Header . Set ( "Accept" , lfs . MediaType )
req . Header . Set ( "Content-Type" , lfs . M ediaType )
2019-08-15 04:53:28 -06:00
resp := session . MakeRequest ( t , req , test . httpResult )
2017-11-28 13:58:37 -07:00
if len ( test . addTime ) > 0 {
2019-08-15 04:53:28 -06:00
var lfsLock api . LFSLockResponse
DecodeJSON ( t , resp , & lfsLock )
2022-01-20 10:46:10 -07:00
assert . EqualValues ( t , lfsLock . Lock . LockedAt . Format ( time . RFC3339 ) , lfsLock . Lock . LockedAt . Format ( time . RFC3339Nano ) ) // locked at should be rounded to second
2017-11-28 13:58:37 -07:00
for _ , id := range test . addTime {
resultsTests [ id ] . locksTimes = append ( resultsTests [ id ] . locksTimes , time . Now ( ) )
}
}
}
2022-01-20 10:46:10 -07:00
// check creation
2017-11-28 13:58:37 -07:00
for _ , test := range resultsTests {
session := loginUser ( t , test . user . Name )
2017-11-28 16:35:23 -07:00
req := NewRequestf ( t , "GET" , "/%s.git/info/lfs/locks" , test . repo . FullName ( ) )
2021-06-05 17:59:27 -06:00
req . Header . Set ( "Accept" , lfs . MediaType )
2017-11-28 13:58:37 -07:00
resp := session . MakeRequest ( t , req , http . StatusOK )
var lfsLocks api . LFSLockList
DecodeJSON ( t , resp , & lfsLocks )
assert . Len ( t , lfsLocks . Locks , test . totalCount )
for i , lock := range lfsLocks . Locks {
assert . EqualValues ( t , test . locksOwners [ i ] . DisplayName ( ) , lock . Owner . Name )
2019-10-03 01:56:26 -06:00
assert . WithinDuration ( t , test . locksTimes [ i ] , lock . LockedAt , 10 * time . Second )
2022-01-20 10:46:10 -07:00
assert . EqualValues ( t , lock . LockedAt . Format ( time . RFC3339 ) , lock . LockedAt . Format ( time . RFC3339Nano ) ) // locked at should be rounded to second
2017-11-28 13:58:37 -07:00
}
2017-11-28 16:35:23 -07:00
req = NewRequestWithJSON ( t , "POST" , fmt . Sprintf ( "/%s.git/info/lfs/locks/verify" , test . repo . FullName ( ) ) , map [ string ] string { } )
2021-06-05 17:59:27 -06:00
req . Header . Set ( "Accept" , lfs . MediaType )
req . Header . Set ( "Content-Type" , lfs . MediaType )
2017-11-28 13:58:37 -07:00
resp = session . MakeRequest ( t , req , http . StatusOK )
var lfsLocksVerify api . LFSLockListVerify
DecodeJSON ( t , resp , & lfsLocksVerify )
assert . Len ( t , lfsLocksVerify . Ours , test . oursCount )
assert . Len ( t , lfsLocksVerify . Theirs , test . theirsCount )
for _ , lock := range lfsLocksVerify . Ours {
assert . EqualValues ( t , test . user . DisplayName ( ) , lock . Owner . Name )
deleteTests = append ( deleteTests , struct {
2021-11-24 02:49:20 -07:00
user * user_model . User
2021-12-09 18:27:50 -07:00
repo * repo_model . Repository
2017-11-28 13:58:37 -07:00
lockID string
} { test . user , test . repo , lock . ID } )
}
for _ , lock := range lfsLocksVerify . Theirs {
assert . NotEqual ( t , test . user . DisplayName ( ) , lock . Owner . Name )
}
}
2022-01-20 10:46:10 -07:00
// remove all locks
2017-11-28 13:58:37 -07:00
for _ , test := range deleteTests {
session := loginUser ( t , test . user . Name )
2017-11-28 16:35:23 -07:00
req := NewRequestWithJSON ( t , "POST" , fmt . Sprintf ( "/%s.git/info/lfs/locks/%s/unlock" , test . repo . FullName ( ) , test . lockID ) , map [ string ] string { } )
2021-06-05 17:59:27 -06:00
req . Header . Set ( "Accept" , lfs . MediaType )
req . Header . Set ( "Content-Type" , lfs . MediaType )
2017-11-28 13:58:37 -07:00
resp := session . MakeRequest ( t , req , http . StatusOK )
var lfsLockRep api . LFSLockResponse
DecodeJSON ( t , resp , & lfsLockRep )
assert . Equal ( t , test . lockID , lfsLockRep . Lock . ID )
assert . Equal ( t , test . user . DisplayName ( ) , lfsLockRep . Lock . Owner . Name )
}
// check that we don't have any lock
for _ , test := range resultsTests {
session := loginUser ( t , test . user . Name )
2017-11-28 16:35:23 -07:00
req := NewRequestf ( t , "GET" , "/%s.git/info/lfs/locks" , test . repo . FullName ( ) )
2021-06-05 17:59:27 -06:00
req . Header . Set ( "Accept" , lfs . MediaType )
2017-11-28 13:58:37 -07:00
resp := session . MakeRequest ( t , req , http . StatusOK )
var lfsLocks api . LFSLockList
DecodeJSON ( t , resp , & lfsLocks )
assert . Len ( t , lfsLocks . Locks , 0 )
}
}