mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
98 lines
3.0 KiB
Go
98 lines
3.0 KiB
Go
// Copyright 2022 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 membership
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/coreos/go-semver/semver"
|
|
"github.com/stretchr/testify/assert"
|
|
"go.uber.org/zap"
|
|
|
|
"go.etcd.io/etcd/client/pkg/v3/types"
|
|
serverversion "go.etcd.io/etcd/server/v3/etcdserver/version"
|
|
)
|
|
|
|
func TestAddRemoveMember(t *testing.T) {
|
|
c := newTestCluster(t, nil)
|
|
be := newMembershipBackend()
|
|
c.SetBackend(be)
|
|
c.AddMember(newTestMemberAsLearner(17, nil, "node17", nil), true)
|
|
c.RemoveMember(17, true)
|
|
c.AddMember(newTestMember(18, nil, "node18", nil), true)
|
|
c.RemoveMember(18, true)
|
|
|
|
// Skipping removal of already removed member
|
|
c.RemoveMember(17, true)
|
|
c.RemoveMember(18, true)
|
|
|
|
c.AddMember(newTestMember(19, nil, "node19", nil), true)
|
|
|
|
// Recover from backend
|
|
c2 := newTestCluster(t, nil)
|
|
c2.SetBackend(be)
|
|
c2.Recover(func(*zap.Logger, *semver.Version) {})
|
|
assert.Equal(t, []*Member{{ID: types.ID(19),
|
|
Attributes: Attributes{Name: "node19"}}}, c2.Members())
|
|
assert.Equal(t, true, c2.IsIDRemoved(17))
|
|
assert.Equal(t, true, c2.IsIDRemoved(18))
|
|
assert.Equal(t, false, c2.IsIDRemoved(19))
|
|
}
|
|
|
|
type backendMock struct {
|
|
members map[types.ID]*Member
|
|
removed map[types.ID]bool
|
|
version *semver.Version
|
|
downgradeInfo *serverversion.DowngradeInfo
|
|
}
|
|
|
|
var _ MembershipBackend = (*backendMock)(nil)
|
|
|
|
func newMembershipBackend() MembershipBackend {
|
|
return &backendMock{
|
|
members: make(map[types.ID]*Member),
|
|
removed: make(map[types.ID]bool),
|
|
downgradeInfo: &serverversion.DowngradeInfo{Enabled: false},
|
|
}
|
|
}
|
|
|
|
func (b *backendMock) MustCreateBackendBuckets() {}
|
|
|
|
func (b *backendMock) ClusterVersionFromBackend() *semver.Version { return b.version }
|
|
func (b *backendMock) MustSaveClusterVersionToBackend(version *semver.Version) {
|
|
b.version = version
|
|
}
|
|
|
|
func (b *backendMock) MustReadMembersFromBackend() (x map[types.ID]*Member, y map[types.ID]bool) {
|
|
return b.members, b.removed
|
|
}
|
|
func (b *backendMock) MustSaveMemberToBackend(m *Member) {
|
|
b.members[m.ID] = m
|
|
}
|
|
func (b *backendMock) TrimMembershipFromBackend() error {
|
|
b.members = make(map[types.ID]*Member)
|
|
b.removed = make(map[types.ID]bool)
|
|
return nil
|
|
}
|
|
func (b *backendMock) MustDeleteMemberFromBackend(id types.ID) {
|
|
delete(b.members, id)
|
|
b.removed[id] = true
|
|
}
|
|
|
|
func (b *backendMock) MustSaveDowngradeToBackend(downgradeInfo *serverversion.DowngradeInfo) {
|
|
b.downgradeInfo = downgradeInfo
|
|
}
|
|
func (b *backendMock) DowngradeInfoFromBackend() *serverversion.DowngradeInfo { return b.downgradeInfo }
|