mirror of https://github.com/gogs/gogs.git
68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
// 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"
|
|
)
|
|
|
|
//go:generate go-mockgen -f gogs.io/gogs/internal/db -i AccessTokensStore -i LFSStore -i LoginSourcesStore -i LoginSourceFilesStore -i loginSourceFileStore -i PermsStore -i ReposStore -i TwoFactorsStore -i UsersStore -o mocks.go
|
|
|
|
func SetMockAccessTokensStore(t *testing.T, mock AccessTokensStore) {
|
|
before := AccessTokens
|
|
AccessTokens = mock
|
|
t.Cleanup(func() {
|
|
AccessTokens = before
|
|
})
|
|
}
|
|
|
|
func SetMockLFSStore(t *testing.T, mock LFSStore) {
|
|
before := LFS
|
|
LFS = mock
|
|
t.Cleanup(func() {
|
|
LFS = before
|
|
})
|
|
}
|
|
|
|
func setMockLoginSourceFilesStore(t *testing.T, db *loginSources, mock loginSourceFilesStore) {
|
|
before := db.files
|
|
db.files = mock
|
|
t.Cleanup(func() {
|
|
db.files = before
|
|
})
|
|
}
|
|
|
|
func SetMockPermsStore(t *testing.T, mock PermsStore) {
|
|
before := Perms
|
|
Perms = mock
|
|
t.Cleanup(func() {
|
|
Perms = before
|
|
})
|
|
}
|
|
|
|
func SetMockReposStore(t *testing.T, mock ReposStore) {
|
|
before := Repos
|
|
Repos = mock
|
|
t.Cleanup(func() {
|
|
Repos = before
|
|
})
|
|
}
|
|
|
|
func SetMockTwoFactorsStore(t *testing.T, mock TwoFactorsStore) {
|
|
before := TwoFactors
|
|
TwoFactors = mock
|
|
t.Cleanup(func() {
|
|
TwoFactors = before
|
|
})
|
|
}
|
|
|
|
func SetMockUsersStore(t *testing.T, mock UsersStore) {
|
|
before := Users
|
|
Users = mock
|
|
t.Cleanup(func() {
|
|
Users = before
|
|
})
|
|
}
|