Function TestPartitionRingDesc_AddOrUpdateOwner

ring/partition_ring_model_test.go:128–184

(t *testing.T)

}

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

	t.Run("should add a new owner", func(t *testing.T) {
		desc := NewPartitionRingDesc()
		require.True(t, desc.AddOrUpdateOwner("instance-1", OwnerActive, 1, now))

		assert.Equal(t, &PartitionRingDesc{
			Partitions: map[int32]PartitionDesc{},
			Owners: map[string]OwnerDesc{
				"instance-1": {
					UpdatedTimestamp: now.Unix(),
					State:            OwnerActive,
					OwnedPartition:   1,
				},
			},
		}, desc)
	})

	t.Run("should update an existing owner", func(t *testing.T) {
		desc := NewPartitionRingDesc()
		require.True(t, desc.AddOrUpdateOwner("instance-1", OwnerActive, 1, now))

		// Update the owner.
		require.True(t, desc.AddOrUpdateOwner("instance-1", OwnerActive, 2, now.Add(time.Second)))

		assert.Equal(t, &PartitionRingDesc{
			Partitions: map[int32]PartitionDesc{},
			Owners: map[string]OwnerDesc{
				"instance-1": {
					UpdatedTimestamp: now.Add(time.Second).Unix(),
					State:            OwnerActive,
					OwnedPartition:   2,
				},
			},
		}, desc)
	})

	t.Run("should be a no-op if the owner already exist", func(t *testing.T) {
		desc := NewPartitionRingDesc()
		desc.AddOrUpdateOwner("instance-1", OwnerActive, 1, now)

		// Update the owner.
		require.False(t, desc.AddOrUpdateOwner("instance-1", OwnerActive, 1, now.Add(time.Second)))

		assert.Equal(t, &PartitionRingDesc{
			Partitions: map[int32]PartitionDesc{},
			Owners: map[string]OwnerDesc{
				"instance-1": {
					UpdatedTimestamp: now.Unix(), // Timestamp should not be updated.
					State:            OwnerActive,
					OwnedPartition:   1,
				},
			},
		}, desc)
	})
}