Function TestPartitionRingDesc_RemoveTombstones

ring/partition_ring_model_test.go:1096–1140

(t *testing.T)

}

func TestPartitionRingDesc_RemoveTombstones(t *testing.T) {
	now := time.Now()

	createTestRing := func() *PartitionRingDesc {
		desc := NewPartitionRingDesc()
		desc.AddPartition(1, PartitionActive, now.Add(1*time.Second))
		desc.AddPartition(2, PartitionInactive, now.Add(2*time.Second))
		desc.AddPartition(3, PartitionDeleted, now.Add(3*time.Second))
		desc.AddOrUpdateOwner("owner-1", OwnerActive, 1, now.Add(4*time.Second))
		desc.AddOrUpdateOwner("owner-2", OwnerActive, 2, now.Add(4*time.Second))
		desc.AddOrUpdateOwner("owner-3", OwnerDeleted, 3, now.Add(4*time.Second))
		return desc
	}

	t.Run("should remove all tombstones when limit is zero value", func(t *testing.T) {
		desc := createTestRing()
		total, removed := desc.RemoveTombstones(time.Time{})
		assert.Equal(t, 0, total)
		assert.Equal(t, 2, removed)
		assert.False(t, desc.HasPartition(3))
		assert.False(t, desc.HasOwner("owner-3"))
	})

	t.Run("should remove tombstones older or equal to the limit when specified", func(t *testing.T) {
		desc := createTestRing()

		total, removed := desc.RemoveTombstones(now)
		assert.Equal(t, 2, total)
		assert.Equal(t, 0, removed)
		assert.True(t, desc.HasPartition(3))
		assert.True(t, desc.HasOwner("owner-3"))

		total, removed = desc.RemoveTombstones(now.Add(3 * time.Second))
		assert.Equal(t, 1, total)
		assert.Equal(t, 1, removed)
		assert.False(t, desc.HasPartition(3))
		assert.True(t, desc.HasOwner("owner-3"))

		total, removed = desc.RemoveTombstones(now.Add(4 * time.Second))
		assert.Equal(t, 0, total)
		assert.Equal(t, 1, removed)
		assert.False(t, desc.HasPartition(3))
		assert.False(t, desc.HasOwner("owner-3"))
	})
}

func TestPartitionRingDesc_UpdatePartitionState(t *testing.T) {
	now := time.Now()