2017-01-08 06:10:53 +03: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 models
import (
"testing"
2017-01-25 05:49:51 +03:00
2021-09-19 14:49:59 +03:00
"code.gitea.io/gitea/models/db"
2021-11-28 14:58:28 +03:00
"code.gitea.io/gitea/models/perm"
2021-11-12 17:36:47 +03:00
"code.gitea.io/gitea/models/unittest"
2021-11-24 12:49:20 +03:00
user_model "code.gitea.io/gitea/models/user"
2021-11-17 15:34:35 +03:00
2017-01-08 06:10:53 +03:00
"github.com/stretchr/testify/assert"
)
func TestAccessLevel ( t * testing . T ) {
2021-11-12 17:36:47 +03:00
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2017-01-08 06:10:53 +03:00
2021-11-24 12:49:20 +03:00
user2 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 2 } ) . ( * user_model . User )
user5 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 5 } ) . ( * user_model . User )
user29 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 29 } ) . ( * user_model . User )
2018-06-21 19:00:13 +03:00
// A public repository owned by User 2
2021-11-16 11:53:21 +03:00
repo1 := unittest . AssertExistsAndLoadBean ( t , & Repository { ID : 1 } ) . ( * Repository )
2018-06-21 19:00:13 +03:00
assert . False ( t , repo1 . IsPrivate )
// A private repository owned by Org 3
2021-11-16 11:53:21 +03:00
repo3 := unittest . AssertExistsAndLoadBean ( t , & Repository { ID : 3 } ) . ( * Repository )
2018-11-28 14:26:14 +03:00
assert . True ( t , repo3 . IsPrivate )
2017-01-08 06:10:53 +03:00
2020-01-13 20:33:46 +03:00
// Another public repository
2021-11-16 11:53:21 +03:00
repo4 := unittest . AssertExistsAndLoadBean ( t , & Repository { ID : 4 } ) . ( * Repository )
2020-01-13 20:33:46 +03:00
assert . False ( t , repo4 . IsPrivate )
// org. owned private repo
2021-11-16 11:53:21 +03:00
repo24 := unittest . AssertExistsAndLoadBean ( t , & Repository { ID : 24 } ) . ( * Repository )
2020-01-13 20:33:46 +03:00
2018-11-28 14:26:14 +03:00
level , err := AccessLevel ( user2 , repo1 )
2017-01-08 06:10:53 +03:00
assert . NoError ( t , err )
2021-11-28 14:58:28 +03:00
assert . Equal ( t , perm . AccessModeOwner , level )
2017-01-08 06:10:53 +03:00
2018-11-28 14:26:14 +03:00
level , err = AccessLevel ( user2 , repo3 )
2017-01-08 06:10:53 +03:00
assert . NoError ( t , err )
2021-11-28 14:58:28 +03:00
assert . Equal ( t , perm . AccessModeOwner , level )
2017-01-08 06:10:53 +03:00
2018-11-28 14:26:14 +03:00
level , err = AccessLevel ( user5 , repo1 )
2017-01-08 06:10:53 +03:00
assert . NoError ( t , err )
2021-11-28 14:58:28 +03:00
assert . Equal ( t , perm . AccessModeRead , level )
2017-01-08 06:10:53 +03:00
2018-11-28 14:26:14 +03:00
level , err = AccessLevel ( user5 , repo3 )
2017-01-08 06:10:53 +03:00
assert . NoError ( t , err )
2021-11-28 14:58:28 +03:00
assert . Equal ( t , perm . AccessModeNone , level )
2020-01-13 20:33:46 +03:00
// restricted user has no access to a public repo
level , err = AccessLevel ( user29 , repo1 )
assert . NoError ( t , err )
2021-11-28 14:58:28 +03:00
assert . Equal ( t , perm . AccessModeNone , level )
2020-01-13 20:33:46 +03:00
// ... unless he's a collaborator
level , err = AccessLevel ( user29 , repo4 )
assert . NoError ( t , err )
2021-11-28 14:58:28 +03:00
assert . Equal ( t , perm . AccessModeWrite , level )
2020-01-13 20:33:46 +03:00
// ... or a team member
level , err = AccessLevel ( user29 , repo24 )
assert . NoError ( t , err )
2021-11-28 14:58:28 +03:00
assert . Equal ( t , perm . AccessModeRead , level )
2017-01-08 06:10:53 +03:00
}
func TestHasAccess ( t * testing . T ) {
2021-11-12 17:36:47 +03:00
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2017-01-08 06:10:53 +03:00
2021-11-24 12:49:20 +03:00
user1 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 2 } ) . ( * user_model . User )
user2 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 5 } ) . ( * user_model . User )
2018-06-21 19:00:13 +03:00
// A public repository owned by User 2
2021-11-16 11:53:21 +03:00
repo1 := unittest . AssertExistsAndLoadBean ( t , & Repository { ID : 1 } ) . ( * Repository )
2018-06-21 19:00:13 +03:00
assert . False ( t , repo1 . IsPrivate )
// A private repository owned by Org 3
2021-11-16 11:53:21 +03:00
repo2 := unittest . AssertExistsAndLoadBean ( t , & Repository { ID : 3 } ) . ( * Repository )
2018-06-21 19:00:13 +03:00
assert . True ( t , repo2 . IsPrivate )
2017-01-08 06:10:53 +03:00
2018-11-28 14:26:14 +03:00
has , err := HasAccess ( user1 . ID , repo1 )
assert . NoError ( t , err )
assert . True ( t , has )
2017-01-08 06:10:53 +03:00
2019-07-22 10:11:16 +03:00
_ , err = HasAccess ( user1 . ID , repo2 )
2018-11-28 14:26:14 +03:00
assert . NoError ( t , err )
2017-01-08 06:10:53 +03:00
2019-07-22 10:11:16 +03:00
_ , err = HasAccess ( user2 . ID , repo1 )
2018-11-28 14:26:14 +03:00
assert . NoError ( t , err )
2017-01-08 06:10:53 +03:00
2019-07-22 10:11:16 +03:00
_ , err = HasAccess ( user2 . ID , repo2 )
2018-11-28 14:26:14 +03:00
assert . NoError ( t , err )
2017-01-08 06:10:53 +03:00
}
func TestRepository_RecalculateAccesses ( t * testing . T ) {
// test with organization repo
2021-11-12 17:36:47 +03:00
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2021-11-16 11:53:21 +03:00
repo1 := unittest . AssertExistsAndLoadBean ( t , & Repository { ID : 3 } ) . ( * Repository )
2017-01-08 06:10:53 +03:00
assert . NoError ( t , repo1 . GetOwner ( ) )
2021-09-23 18:45:36 +03:00
_ , err := db . GetEngine ( db . DefaultContext ) . Delete ( & Collaboration { UserID : 2 , RepoID : 3 } )
2017-01-08 06:10:53 +03:00
assert . NoError ( t , err )
assert . NoError ( t , repo1 . RecalculateAccesses ( ) )
access := & Access { UserID : 2 , RepoID : 3 }
2021-09-23 18:45:36 +03:00
has , err := db . GetEngine ( db . DefaultContext ) . Get ( access )
2017-01-08 06:10:53 +03:00
assert . NoError ( t , err )
assert . True ( t , has )
2021-11-28 14:58:28 +03:00
assert . Equal ( t , perm . AccessModeOwner , access . Mode )
2017-01-08 06:10:53 +03:00
}
func TestRepository_RecalculateAccesses2 ( t * testing . T ) {
// test with non-organization repo
2021-11-12 17:36:47 +03:00
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2021-11-16 11:53:21 +03:00
repo1 := unittest . AssertExistsAndLoadBean ( t , & Repository { ID : 4 } ) . ( * Repository )
2017-01-08 06:10:53 +03:00
assert . NoError ( t , repo1 . GetOwner ( ) )
2021-09-23 18:45:36 +03:00
_ , err := db . GetEngine ( db . DefaultContext ) . Delete ( & Collaboration { UserID : 4 , RepoID : 4 } )
2017-01-08 06:10:53 +03:00
assert . NoError ( t , err )
assert . NoError ( t , repo1 . RecalculateAccesses ( ) )
2021-09-23 18:45:36 +03:00
has , err := db . GetEngine ( db . DefaultContext ) . Get ( & Access { UserID : 4 , RepoID : 4 } )
2017-01-08 06:10:53 +03:00
assert . NoError ( t , err )
assert . False ( t , has )
}
2020-01-13 20:33:46 +03:00
func TestRepository_RecalculateAccesses3 ( t * testing . T ) {
2021-11-12 17:36:47 +03:00
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2021-11-16 11:53:21 +03:00
team5 := unittest . AssertExistsAndLoadBean ( t , & Team { ID : 5 } ) . ( * Team )
2021-11-24 12:49:20 +03:00
user29 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 29 } ) . ( * user_model . User )
2020-01-13 20:33:46 +03:00
2021-09-23 18:45:36 +03:00
has , err := db . GetEngine ( db . DefaultContext ) . Get ( & Access { UserID : 29 , RepoID : 23 } )
2020-01-13 20:33:46 +03:00
assert . NoError ( t , err )
assert . False ( t , has )
// adding user29 to team5 should add an explicit access row for repo 23
// even though repo 23 is public
assert . NoError ( t , AddTeamMember ( team5 , user29 . ID ) )
2021-09-23 18:45:36 +03:00
has , err = db . GetEngine ( db . DefaultContext ) . Get ( & Access { UserID : 29 , RepoID : 23 } )
2020-01-13 20:33:46 +03:00
assert . NoError ( t , err )
assert . True ( t , has )
}