diff options
Diffstat (limited to 'internal/db/mock_gen.go')
| -rw-r--r-- | internal/db/mock_gen.go | 214 |
1 files changed, 214 insertions, 0 deletions
diff --git a/internal/db/mock_gen.go b/internal/db/mock_gen.go new file mode 100644 index 00000000..564dcc57 --- /dev/null +++ b/internal/db/mock_gen.go @@ -0,0 +1,214 @@ +// Copyright 2020 The Gogs 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 db + +import ( + "testing" + + "gogs.io/gogs/internal/lfsutil" +) + +//go:generate go-mockgen -f gogs.io/gogs/internal/db -i AccessTokensStore -o mocks.go + +func SetMockAccessTokensStore(t *testing.T, mock AccessTokensStore) { + before := AccessTokens + AccessTokens = mock + t.Cleanup(func() { + AccessTokens = before + }) +} + +var _ LFSStore = (*MockLFSStore)(nil) + +type MockLFSStore struct { + MockCreateObject func(repoID int64, oid lfsutil.OID, size int64, storage lfsutil.Storage) error + MockGetObjectByOID func(repoID int64, oid lfsutil.OID) (*LFSObject, error) + MockGetObjectsByOIDs func(repoID int64, oids ...lfsutil.OID) ([]*LFSObject, error) +} + +func (m *MockLFSStore) CreateObject(repoID int64, oid lfsutil.OID, size int64, storage lfsutil.Storage) error { + return m.MockCreateObject(repoID, oid, size, storage) +} + +func (m *MockLFSStore) GetObjectByOID(repoID int64, oid lfsutil.OID) (*LFSObject, error) { + return m.MockGetObjectByOID(repoID, oid) +} + +func (m *MockLFSStore) GetObjectsByOIDs(repoID int64, oids ...lfsutil.OID) ([]*LFSObject, error) { + return m.MockGetObjectsByOIDs(repoID, oids...) +} + +func SetMockLFSStore(t *testing.T, mock LFSStore) { + before := LFS + LFS = mock + t.Cleanup(func() { + LFS = before + }) +} + +var _ loginSourceFilesStore = (*mockLoginSourceFilesStore)(nil) + +type mockLoginSourceFilesStore struct { + MockGetByID func(id int64) (*LoginSource, error) + MockLen func() int + MockList func(opts ListLoginSourceOpts) []*LoginSource + MockUpdate func(source *LoginSource) +} + +func (m *mockLoginSourceFilesStore) GetByID(id int64) (*LoginSource, error) { + return m.MockGetByID(id) +} + +func (m *mockLoginSourceFilesStore) Len() int { + return m.MockLen() +} + +func (m *mockLoginSourceFilesStore) List(opts ListLoginSourceOpts) []*LoginSource { + return m.MockList(opts) +} + +func (m *mockLoginSourceFilesStore) Update(source *LoginSource) { + m.MockUpdate(source) +} + +func setMockLoginSourceFilesStore(t *testing.T, db *loginSources, mock loginSourceFilesStore) { + before := db.files + db.files = mock + t.Cleanup(func() { + db.files = before + }) +} + +var _ loginSourceFileStore = (*mockLoginSourceFileStore)(nil) + +type mockLoginSourceFileStore struct { + MockSetGeneral func(name, value string) + MockSetConfig func(cfg interface{}) error + MockSave func() error +} + +func (m *mockLoginSourceFileStore) SetGeneral(name, value string) { + m.MockSetGeneral(name, value) +} + +func (m *mockLoginSourceFileStore) SetConfig(cfg interface{}) error { + return m.MockSetConfig(cfg) +} + +func (m *mockLoginSourceFileStore) Save() error { + return m.MockSave() +} + +var _ PermsStore = (*MockPermsStore)(nil) + +type MockPermsStore struct { + MockAccessMode func(userID, repoID int64, opts AccessModeOptions) AccessMode + MockAuthorize func(userID, repoID int64, desired AccessMode, opts AccessModeOptions) bool + MockSetRepoPerms func(repoID int64, accessMap map[int64]AccessMode) error +} + +func (m *MockPermsStore) AccessMode(userID, repoID int64, opts AccessModeOptions) AccessMode { + return m.MockAccessMode(userID, repoID, opts) +} + +func (m *MockPermsStore) Authorize(userID, repoID int64, desired AccessMode, opts AccessModeOptions) bool { + return m.MockAuthorize(userID, repoID, desired, opts) +} + +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 + t.Cleanup(func() { + Perms = before + }) +} + +var _ ReposStore = (*MockReposStore)(nil) + +type MockReposStore struct { + MockGetByName func(ownerID int64, name string) (*Repository, error) +} + +func (m *MockReposStore) GetByName(ownerID int64, name string) (*Repository, error) { + return m.MockGetByName(ownerID, name) +} + +func SetMockReposStore(t *testing.T, mock ReposStore) { + before := Repos + Repos = mock + t.Cleanup(func() { + Repos = before + }) +} + +var _ TwoFactorsStore = (*MockTwoFactorsStore)(nil) + +type MockTwoFactorsStore struct { + MockCreate func(userID int64, key, secret string) error + MockGetByUserID func(userID int64) (*TwoFactor, error) + MockIsUserEnabled func(userID int64) bool +} + +func (m *MockTwoFactorsStore) Create(userID int64, key, secret string) error { + return m.MockCreate(userID, key, secret) +} + +func (m *MockTwoFactorsStore) GetByUserID(userID int64) (*TwoFactor, error) { + return m.MockGetByUserID(userID) +} + +func (m *MockTwoFactorsStore) IsUserEnabled(userID int64) bool { + return m.MockIsUserEnabled(userID) +} + +func SetMockTwoFactorsStore(t *testing.T, mock TwoFactorsStore) { + before := TwoFactors + TwoFactors = mock + t.Cleanup(func() { + TwoFactors = before + }) +} + +var _ UsersStore = (*MockUsersStore)(nil) + +type MockUsersStore struct { + MockAuthenticate func(username, password string, loginSourceID int64) (*User, error) + MockCreate func(username, email string, opts CreateUserOpts) (*User, error) + MockGetByEmail func(email string) (*User, error) + MockGetByID func(id int64) (*User, error) + MockGetByUsername func(username string) (*User, error) +} + +func (m *MockUsersStore) Authenticate(username, password string, loginSourceID int64) (*User, error) { + return m.MockAuthenticate(username, password, loginSourceID) +} + +func (m *MockUsersStore) Create(username, email string, opts CreateUserOpts) (*User, error) { + return m.MockCreate(username, email, opts) +} + +func (m *MockUsersStore) GetByEmail(email string) (*User, error) { + return m.MockGetByEmail(email) +} + +func (m *MockUsersStore) GetByID(id int64) (*User, error) { + return m.MockGetByID(id) +} + +func (m *MockUsersStore) GetByUsername(username string) (*User, error) { + return m.MockGetByUsername(username) +} + +func SetMockUsersStore(t *testing.T, mock UsersStore) { + before := Users + Users = mock + t.Cleanup(func() { + Users = before + }) +} |