Function TestSet_SUnion

internal/data/set_test.go:401–455

(t *testing.T)

}

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

	key1 := "set1"
	key2 := "set2"
	key3 := "set3"
	key4 := "set4"

	expectRecords := testutils.GenerateRecords(10)

	values := make([][]byte, 10)
	for i := range expectRecords {
		values[i] = expectRecords[i].Value
	}

	require.NoError(t, set.SAdd(key1, values[:5], expectRecords[:5]))
	require.NoError(t, set.SAdd(key2, values[2:6], expectRecords[2:6]))
	require.NoError(t, set.SAdd(key3, values[2:7], expectRecords[2:7]))
	require.NoError(t, set.SAdd(key4, values[7:], expectRecords[7:]))

	type args struct {
		key1 string
		key2 string
	}

	tests := []struct {
		name    string
		args    args
		set     *Set
		want    []*core.Record
		wantErr bool
	}{
		{"normal set Union1", args{key1, key4}, set,
			[]*core.Record{expectRecords[0], expectRecords[1], expectRecords[2], expectRecords[3], expectRecords[4], expectRecords[7], expectRecords[8], expectRecords[9]},
			false},
		{
			"normal set Union2", args{key2, key3}, set,
			[]*core.Record{expectRecords[2], expectRecords[3], expectRecords[4], expectRecords[5], expectRecords[6]},
			false,
		},
		{"first fake set", args{"fake_key1", key2}, set, nil, true},
		{"second fake set", args{key1, "fake_key2"}, set, nil, true},
		{"two fake set", args{"fake_key1", "fake_key2"}, set, nil, true},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := tt.set.SUnion(tt.args.key1, tt.args.key2)
			if (err != nil) != tt.wantErr {
				require.Errorf(t, err, "Get() error = %v, wantErr %v", tt.wantErr)
			}
			require.ElementsMatchf(t, tt.want, got, "Get() got = %v, want %v", got, tt.want)
		})
	}
}