diff options
Diffstat (limited to 'internal/db/mocks.go')
| -rw-r--r-- | internal/db/mocks.go | 9 |
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 |