server: Use zaptest in bucket tests and move backendMock to separate file

server/auth/store_mock_test.go

@@ -0,0 +1,127 @@
+// 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) 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 {
+	users := []*authpb.User{}
+	for _, u := range t.be.users {
+		users = append(users, u)
+	}
+	return users
+}
+
+func (t txMock) UnsafeGetAllRoles() []*authpb.Role {
+	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)
+}