Function TestDeleteMultipleClients

kv/memberlist/memberlist_client_test.go:751–808

(t *testing.T)

}

func TestDeleteMultipleClients(t *testing.T) {

	const deleteTime = 1 * time.Second

	var cfg KVConfig
	flagext.DefaultValues(&cfg)
	cfg.TCPTransport = TCPTransportConfig{
		BindAddrs: getLocalhostAddrs(),
		BindPort:  0, // randomize
	}

	cfg.Codecs = []codec.Codec{
		dataCodec{},
	}

	cfg.GossipNodes = 1
	cfg.GossipInterval = 100 * time.Millisecond
	cfg.PushPullInterval = deleteTime
	cfg.ObsoleteEntriesTimeout = deleteTime

	mkv1 := NewKV(cfg, log.NewNopLogger(), &staticDNSProviderMock{}, prometheus.NewPedanticRegistry())
	require.NoError(t, services.StartAndAwaitRunning(context.Background(), mkv1))
	defer services.StopAndAwaitTerminated(context.Background(), mkv1) //nolint:errcheck

	kv1, err := NewClient(mkv1, dataCodec{})
	require.NoError(t, err)

	const memberKey = "entry"

	// Calling updateFn once creates single entry, in JOINING state.
	err = cas(kv1, key, updateFn(memberKey))
	require.NoError(t, err)

	// We will read values from second KV, which will join the first one
	cfg.JoinMembers = []string{net.JoinHostPort("127.0.0.1", strconv.Itoa(mkv1.GetListeningPort()))}

	mkv2 := NewKV(cfg, log.NewNopLogger(), &staticDNSProviderMock{}, prometheus.NewPedanticRegistry())
	require.NoError(t, services.StartAndAwaitRunning(context.Background(), mkv2))
	defer services.StopAndAwaitTerminated(context.Background(), mkv2) //nolint:errcheck

	// While waiting for mkv2 to start, we can already create a client for it.
	// Any client operations will block until mkv2 transitioned to Running state.
	kv2, err := NewClient(mkv2, dataCodec{})
	require.NoError(t, err)

	val, err := kv2.Get(context.Background(), key)
	require.NoError(t, err)
	require.NotNil(t, val)

	err = kv1.Delete(context.Background(), key)
	if err != nil {
		t.Fatalf("Failed to delete key %s: %v", key, err)
	}

	// wait for the obselete entries to be removed.
	checkMemberlistEntry(t, kv1, key, 10*deleteTime)
	checkMemberlistEntry(t, kv2, key, 10*deleteTime)
}