From 76bb647d2437dbdea86ac1a0caf5d768ab924e18 Mon Sep 17 00:00:00 2001 From: ᴜɴᴋɴᴡᴏɴ Date: Sat, 11 Apr 2020 05:39:45 +0800 Subject: 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 --- internal/db/mocks.go | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'internal/db/mocks.go') 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 -- cgit v1.2.3