Function TestSet_SPop

internal/data/set_test.go:295–336

(t *testing.T)

}

func TestSet_SPop(t *testing.T) {
	set := NewSet()

	key := "set"

	expectRecords := testutils.GenerateRecords(2)
	values := make([][]byte, 2)
	for i := range expectRecords {
		values[i] = expectRecords[i].Value
	}
	m := map[*core.Record]struct{}{}
	for _, expectRecord := range expectRecords {
		m[expectRecord] = struct{}{}
	}

	require.NoError(t, set.SAdd(key, values, expectRecords))

	tests := []struct {
		name string
		key  string
		set  *Set
		ok   bool
	}{
		{"normal set SPop", key, set, true},
		{"normal set SPop", key, set, true},
		{"normal set SPop", key, set, false},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			record := tt.set.SPop(tt.key)
			_, ok := m[record]
			require.Equal(t, tt.ok, ok)
		})
	}

	t.Run("set SPop key not found", func(t *testing.T) {
		s := NewSet()
		rec := s.SPop("not_found_key")
		assert.Nil(t, rec)
	})
}

func TestSet_SIsMember(t *testing.T) {
	set := NewSet()