Function TestPartitionRingDesc_Merge_RemoveOwner

ring/partition_ring_model_test.go:836–1046

(t *testing.T)

}

func TestPartitionRingDesc_Merge_RemoveOwner(t *testing.T) {
	now := time.Unix(10000, 0)

	tests := map[string]struct {
		localCAS             bool
		local                *PartitionRingDesc
		incoming             *PartitionRingDesc
		expectedUpdatedLocal memberlist.Mergeable
		expectedChange       memberlist.Mergeable
	}{
		"local change: owner removed and local owner state is not deleted yet": {
			localCAS: true,
			local: &PartitionRingDesc{
				Partitions: map[int32]PartitionDesc{
					1: {Id: 1, Tokens: []uint32{1, 2, 3}, State: PartitionActive, StateTimestamp: 10},
					2: {Id: 2, Tokens: []uint32{4, 5, 6}, State: PartitionActive, StateTimestamp: 20},
				},
				Owners: map[string]OwnerDesc{
					"ingester-zone-a-0": {OwnedPartition: 1, State: OwnerActive, UpdatedTimestamp: 10},
					"ingester-zone-a-1": {OwnedPartition: 2, State: OwnerActive, UpdatedTimestamp: 20},
				},
			},
			incoming: &PartitionRingDesc{
				Partitions: map[int32]PartitionDesc{
					1: {Id: 1, Tokens: []uint32{1, 2, 3}, State: PartitionActive, StateTimestamp: 10},
					2: {Id: 2, Tokens: []uint32{4, 5, 6}, State: PartitionActive, StateTimestamp: 20},
				},
				Owners: map[string]OwnerDesc{
					"ingester-zone-a-0": {OwnedPartition: 1, State: OwnerActive, UpdatedTimestamp: 10},
					// Owner ingester-zone-a-1 removed.
				},
			},
			expectedUpdatedLocal: &PartitionRingDesc{
				Partitions: map[int32]PartitionDesc{
					1: {Id: 1, Tokens: []uint32{1, 2, 3}, State: PartitionActive, StateTimestamp: 10},
					2: {Id: 2, Tokens: []uint32{4, 5, 6}, State: PartitionActive, StateTimestamp: 20},
				},
				Owners: map[string]OwnerDesc{
					"ingester-zone-a-0": {OwnedPartition: 1, State: OwnerActive, UpdatedTimestamp: 10},
					"ingester-zone-a-1": {OwnedPartition: 2, State: OwnerDeleted, UpdatedTimestamp: now.Unix()},
				},
			},
			expectedChange: &PartitionRingDesc{
				Partitions: map[int32]PartitionDesc{},
				Owners: map[string]OwnerDesc{
					"ingester-zone-a-1": {OwnedPartition: 2, State: OwnerDeleted, UpdatedTimestamp: now.Unix()},
				},
			},
		},
		"local change: partition removed and local partition state is already deleted": {
			localCAS: true,
			local: &PartitionRingDesc{
				Partitions: map[int32]PartitionDesc{
					1: {Id: 1, Tokens: []uint32{1, 2, 3}, State: PartitionActive, StateTimestamp: 10},
					2: {Id: 2, Tokens: []uint32{4, 5, 6}, State: PartitionActive, StateTimestamp: 20},
				},
				Owners: map[string]OwnerDesc{
					"ingester-zone-a-0": {OwnedPartition: 1, State: OwnerActive, UpdatedTimestamp: 10},