aboutsummaryrefslogtreecommitdiff
path: root/internal/db/mocks.go
diff options
context:
space:
mode:
authorᴜɴᴋɴᴡᴏɴ <u@gogs.io>2020-04-11 05:39:45 +0800
committerGitHub <noreply@github.com>2020-04-11 05:39:45 +0800
commit76bb647d2437dbdea86ac1a0caf5d768ab924e18 (patch)
tree94a59406f014521d17e662ba4dcf2545b6d3a5ff /internal/db/mocks.go
parente077ecdd9d95ecb76d91105b3858ee48d52c0cc2 (diff)
db: add tests for permissions (#6088)
* Add flag to print SQLs * Add tests for perms * Make results stable * codecov: only show diff * Once again, stable find results
Diffstat (limited to 'internal/db/mocks.go')
1 files changed, 7 insertions, 2 deletions
diff --git a/internal/db/mocks.go b/internal/db/mocks.go
index 12622a7c..7e6ddced 100644
--- a/internal/db/mocks.go
+++ b/internal/db/mocks.go
@@ -81,8 +81,9 @@ func SetMockLFSStore(t *testing.T, mock LFSStore) {
var _ PermsStore = (*MockPermsStore)(nil)
type MockPermsStore struct {
- MockAccessMode func(userID int64, repo *Repository) AccessMode
- MockAuthorize func(userID int64, repo *Repository, desired AccessMode) bool
+ MockAccessMode func(userID int64, repo *Repository) AccessMode
+ MockAuthorize func(userID int64, repo *Repository, desired AccessMode) bool
+ MockSetRepoPerms func(repoID int64, accessMap map[int64]AccessMode) error
}
func (m *MockPermsStore) AccessMode(userID int64, repo *Repository) AccessMode {
@@ -93,6 +94,10 @@ func (m *MockPermsStore) Authorize(userID int64, repo *Repository, desired Acces
return m.MockAuthorize(userID, repo, desired)
}
+func (m *MockPermsStore) SetRepoPerms(repoID int64, accessMap map[int64]AccessMode) error {
+ return m.MockSetRepoPerms(repoID, accessMap)
+}
+
func SetMockPermsStore(t *testing.T, mock PermsStore) {
before := Perms
Perms = mock