Function TestLabelSetTracker

pkg/util/labelset/tracker_test.go:12–71

(t *testing.T)

)

func TestLabelSetTracker(t *testing.T) {
	tracker := NewLabelSetTracker()
	userID := "1"
	userID2 := "2"
	userID3 := "3"

	tracker.Track(userID, 0, labels.FromStrings("foo", "bar"))
	require.True(t, tracker.labelSetExists(userID, 0, labels.FromStrings("foo", "bar")))
	tracker.Track(userID, 1, labels.FromStrings("foo", "baz"))
	require.True(t, tracker.labelSetExists(userID, 1, labels.FromStrings("foo", "baz")))
	tracker.Track(userID, 3, labels.EmptyLabels())
	require.True(t, tracker.labelSetExists(userID, 3, labels.EmptyLabels()))
	tracker.Track(userID2, 0, labels.FromStrings("foo", "bar"))
	require.True(t, tracker.labelSetExists(userID2, 0, labels.FromStrings("foo", "bar")))
	tracker.Track(userID2, 2, labels.FromStrings("cluster", "us-west-2"))
	require.True(t, tracker.labelSetExists(userID2, 2, labels.FromStrings("cluster", "us-west-2")))

	// Increment metrics and add a new user.
	tracker.Track(userID, 3, labels.EmptyLabels())
	require.True(t, tracker.labelSetExists(userID, 3, labels.EmptyLabels()))
	tracker.Track(userID2, 2, labels.FromStrings("cluster", "us-west-2"))
	require.True(t, tracker.labelSetExists(userID2, 2, labels.FromStrings("cluster", "us-west-2")))
	tracker.Track(userID2, 4, labels.FromStrings("cluster", "us-west-2"))
	require.True(t, tracker.labelSetExists(userID2, 4, labels.FromStrings("cluster", "us-west-2")))
	tracker.Track(userID3, 4, labels.FromStrings("cluster", "us-east-1"))
	require.True(t, tracker.labelSetExists(userID3, 4, labels.FromStrings("cluster", "us-east-1")))

	// Remove user 2.
	userSet := map[string]map[uint64]struct {
	}{
		userID:  {0: struct{}{}, 1: struct{}{}, 2: struct{}{}, 3: struct{}{}},
		userID3: {0: struct{}{}, 1: struct{}{}, 2: struct{}{}, 3: struct{}{}, 4: struct{}{}},
	}
	tracker.UpdateMetrics(userSet, noopDeleteMetrics)
	// user 2 removed.
	require.False(t, tracker.userExists(userID2))
	// user 1 and 3 remain unchanged.
	require.True(t, tracker.labelSetExists(userID, 0, labels.FromStrings("foo", "bar")))
	require.True(t, tracker.labelSetExists(userID, 1, labels.FromStrings("foo", "baz")))
	require.True(t, tracker.labelSetExists(userID, 3, labels.EmptyLabels()))
	require.True(t, tracker.labelSetExists(userID3, 4, labels.FromStrings("cluster", "us-east-1")))

	// Simulate existing limits removed for each user.
	userSet = map[string]map[uint64]struct {
	}{
		userID:  {0: struct{}{}},
		userID2: {},
		userID3: {},
	}
	tracker.UpdateMetrics(userSet, noopDeleteMetrics)
	// User 2 and 3 removed. User 1 exists.
	require.True(t, tracker.userExists(userID))
	require.False(t, tracker.userExists(userID2))
	require.False(t, tracker.userExists(userID3))

	require.True(t, tracker.labelSetExists(userID, 0, labels.FromStrings("foo", "bar")))
	require.False(t, tracker.labelSetExists(userID, 1, labels.FromStrings("foo", "baz")))
	require.False(t, tracker.labelSetExists(userID, 3, labels.EmptyLabels()))