Function TestCheckReplicaMultiUser

pkg/ha/ha_tracker_test.go:530–570

Test that writes only happen every write timeout.

(t *testing.T)

// Test that writes only happen every write timeout.
func TestCheckReplicaMultiUser(t *testing.T) {
	t.Parallel()
	replica := "r1"
	replicaGroup := "c1"

	codec := GetReplicaDescCodec()
	kvStore, closer := consul.NewInMemoryClient(codec, log.NewNopLogger(), nil)
	t.Cleanup(func() { assert.NoError(t, closer.Close()) })

	mock := kv.PrefixClient(kvStore, "prefix")
	c, err := NewHATracker(HATrackerConfig{
		EnableHATracker:        true,
		KVStore:                kv.Config{Mock: mock},
		UpdateTimeout:          100 * time.Millisecond,
		UpdateTimeoutJitterMax: 0,
		FailoverTimeout:        time.Second,
	}, trackerLimits{maxReplicaGroups: 100}, haTrackerStatusConfig, nil, "test-ha-tracker", log.NewNopLogger())
	require.NoError(t, err)
	require.NoError(t, services.StartAndAwaitRunning(context.Background(), c))
	defer services.StopAndAwaitTerminated(context.Background(), c) //nolint:errcheck

	now := time.Now()

	// Write the first time for user 1.
	err = c.CheckReplica(context.Background(), "user1", replicaGroup, replica, now)
	assert.NoError(t, err)
	checkReplicaTimestamp(t, time.Second, c, "user1", replicaGroup, replica, now)

	// Write the first time for user 2.
	err = c.CheckReplica(context.Background(), "user2", replicaGroup, replica, now)
	assert.NoError(t, err)
	checkReplicaTimestamp(t, time.Second, c, "user2", replicaGroup, replica, now)

	// Now we've waited > 1s, so the timestamp should update.
	updated := now.Add(1100 * time.Millisecond)
	err = c.CheckReplica(context.Background(), "user1", replicaGroup, replica, updated)
	assert.NoError(t, err)
	checkReplicaTimestamp(t, time.Second, c, "user1", replicaGroup, replica, updated)
	// No update for user2.
	checkReplicaTimestamp(t, time.Second, c, "user2", replicaGroup, replica, now)
}

func TestCheckReplicaUpdateTimeoutJitter(t *testing.T) {
	t.Parallel()