Function Test_LeaderElection

pkg/usagestats/reporter_test.go:23–71

(t *testing.T)

)

func Test_LeaderElection(t *testing.T) {
	stabilityCheckInterval = 100 * time.Millisecond

	result := make(chan *ClusterSeed, 10)

	objectClient, err := local.NewBackend(&local.Config{
		Path: t.TempDir(),
	})
	require.NoError(t, err)

	for i := 0; i < 3; i++ {
		go func() {
			r, leaderErr := NewReporter(Config{Leader: true, Enabled: true}, kv.Config{
				Store: "inmemory",
			}, objectClient, objectClient, log.NewNopLogger(), nil)
			require.NoError(t, leaderErr)
			r.init(context.Background())
			result <- r.cluster
		}()
	}
	for i := 0; i < 7; i++ {
		go func() {
			r, nonLeaderError := NewReporter(Config{Leader: false, Enabled: true}, kv.Config{
				Store: "inmemory",
			}, objectClient, objectClient, log.NewNopLogger(), nil)
			require.NoError(t, nonLeaderError)
			r.init(context.Background())
			result <- r.cluster
		}()
	}

	var UID []string
	for i := 0; i < 10; i++ {
		cluster := <-result
		require.NotNil(t, cluster)
		UID = append(UID, cluster.UID)
	}
	first := UID[0]
	for _, uid := range UID {
		require.Equal(t, first, uid)
	}
	kvClient, err := kv.NewClient(kv.Config{Store: "inmemory"}, JSONCodec, nil, log.NewNopLogger())
	require.NoError(t, err)
	// verify that the ID found is also correctly stored in the kv store and not overridden by another leader.
	data, err := kvClient.Get(context.Background(), seedKey)
	require.NoError(t, err)
	t.Logf("data: %+v", data.(*ClusterSeed))
	require.Equal(t, data.(*ClusterSeed).UID, first)
}

func Test_LeaderElectionWithBrokenSeedFile(t *testing.T) {
	stabilityCheckInterval = 100 * time.Millisecond