mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
72cf0cc04a
declare an empty slice to var s []int replace s :=[]int{}, https://github.com/golang/go/wiki/CodeReviewComments#declaring-empty-slices
Signed-off-by: demoManito <1430482733@qq.com>
132 lines
2.7 KiB
Go
132 lines
2.7 KiB
Go
// Copyright 2021 The etcd Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package auth
|
|
|
|
import "go.etcd.io/etcd/api/v3/authpb"
|
|
|
|
type backendMock struct {
|
|
users map[string]*authpb.User
|
|
roles map[string]*authpb.Role
|
|
enabled bool
|
|
revision uint64
|
|
}
|
|
|
|
func newBackendMock() *backendMock {
|
|
return &backendMock{
|
|
users: make(map[string]*authpb.User),
|
|
roles: make(map[string]*authpb.Role),
|
|
}
|
|
}
|
|
|
|
func (b *backendMock) CreateAuthBuckets() {
|
|
}
|
|
|
|
func (b *backendMock) ForceCommit() {
|
|
}
|
|
|
|
func (b *backendMock) ReadTx() AuthReadTx {
|
|
return &txMock{be: b}
|
|
}
|
|
|
|
func (b *backendMock) BatchTx() AuthBatchTx {
|
|
return &txMock{be: b}
|
|
}
|
|
|
|
func (b *backendMock) GetUser(s string) *authpb.User {
|
|
return b.users[s]
|
|
}
|
|
|
|
func (b *backendMock) GetAllUsers() []*authpb.User {
|
|
return b.BatchTx().UnsafeGetAllUsers()
|
|
}
|
|
|
|
func (b *backendMock) GetRole(s string) *authpb.Role {
|
|
return b.roles[s]
|
|
}
|
|
|
|
func (b *backendMock) GetAllRoles() []*authpb.Role {
|
|
return b.BatchTx().UnsafeGetAllRoles()
|
|
}
|
|
|
|
var _ AuthBackend = (*backendMock)(nil)
|
|
|
|
type txMock struct {
|
|
be *backendMock
|
|
}
|
|
|
|
var _ AuthBatchTx = (*txMock)(nil)
|
|
|
|
func (t txMock) UnsafeReadAuthEnabled() bool {
|
|
return t.be.enabled
|
|
}
|
|
|
|
func (t txMock) UnsafeReadAuthRevision() uint64 {
|
|
return t.be.revision
|
|
}
|
|
|
|
func (t txMock) UnsafeGetUser(s string) *authpb.User {
|
|
return t.be.users[s]
|
|
}
|
|
|
|
func (t txMock) UnsafeGetRole(s string) *authpb.Role {
|
|
return t.be.roles[s]
|
|
}
|
|
|
|
func (t txMock) UnsafeGetAllUsers() []*authpb.User {
|
|
var users []*authpb.User
|
|
for _, u := range t.be.users {
|
|
users = append(users, u)
|
|
}
|
|
return users
|
|
}
|
|
|
|
func (t txMock) UnsafeGetAllRoles() []*authpb.Role {
|
|
var roles []*authpb.Role
|
|
for _, r := range t.be.roles {
|
|
roles = append(roles, r)
|
|
}
|
|
return roles
|
|
}
|
|
|
|
func (t txMock) Lock() {
|
|
}
|
|
|
|
func (t txMock) Unlock() {
|
|
}
|
|
|
|
func (t txMock) UnsafeSaveAuthEnabled(enabled bool) {
|
|
t.be.enabled = enabled
|
|
}
|
|
|
|
func (t txMock) UnsafeSaveAuthRevision(rev uint64) {
|
|
t.be.revision = rev
|
|
}
|
|
|
|
func (t txMock) UnsafePutUser(user *authpb.User) {
|
|
t.be.users[string(user.Name)] = user
|
|
}
|
|
|
|
func (t txMock) UnsafeDeleteUser(s string) {
|
|
delete(t.be.users, s)
|
|
}
|
|
|
|
func (t txMock) UnsafePutRole(role *authpb.Role) {
|
|
t.be.roles[string(role.Name)] = role
|
|
}
|
|
|
|
func (t txMock) UnsafeDeleteRole(s string) {
|
|
delete(t.be.roles, s)
|
|
}
|