query_consistency_test.go

package goriak

import (
	"testing"
)

func TestSetRawWithConsistancyOptions(t *testing.T) {
	_, err := Bucket("raw", "default").
		SetRaw([]byte("Hello Raw")).
		WithPw(4).Run(con())

	expectedError := "ClientError|[Cluster] all retries exhausted and/or no nodes available to execute command|InnerError|RiakError|0|{n_val_violation,3}"

	if err.Error() != expectedError {
		t.Error("Unexpected error PW:", err)
	}

	_, err = Bucket("raw", "default").
		SetRaw([]byte("Hello Raw")).
		WithDw(4).Run(con())

	if err.Error() != expectedError {
		t.Error("Unexpected error DW:", err)
	}

	_, err = Bucket("raw", "default").
		SetRaw([]byte("Hello Raw")).
		WithW(4).Run(con())

	if err.Error() != expectedError {
		t.Error("Unexpected error W:", err)
	}
}

func TestGetWithConsistancyOptions(t *testing.T) {
	input, err := Bucket("raw", "default").
		SetRaw([]byte("Hello Raw")).
		Run(con())

	_, err = Bucket("raw", "default").
		GetRaw(input.Key, &[]byte{}).
		WithPr(4).Run(con())

	expectedError := "ClientError|[Cluster] all retries exhausted and/or no nodes available to execute command|InnerError|RiakError|0|{n_val_violation,3}"

	if err.Error() != expectedError {
		t.Error("Unexpected error PW:", err)
	}

	_, err = Bucket("raw", "default").
		GetRaw(input.Key, &[]byte{}).
		WithR(4).Run(con())

	if err.Error() != expectedError {
		t.Error("Unexpected error PW:", err)
	}
}

func TestSetWithConsistancyOptions(t *testing.T) {
	val := struct {
		A string
	}{
		A: "Hello A",
	}

	_, err := Bucket("testsuitemap", "maps").
		Set(val).
		WithPw(4).Run(con())

	expectedError := "ClientError|[Cluster] all retries exhausted and/or no nodes available to execute command|InnerError|RiakError|0|{n_val_violation,3}"

	if err.Error() != expectedError {
		t.Error("Unexpected error PW:", err)
	}

	_, err = Bucket("testsuitemap", "maps").
		Set(val).
		WithDw(4).Run(con())

	if err.Error() != expectedError {
		t.Error("Unexpected error DW:", err)
	}

	_, err = Bucket("testsuitemap", "maps").
		Set(val).
		WithW(4).Run(con())

	if err.Error() != expectedError {
		t.Error("Unexpected error W:", err)
	}
}

func TestDeleteWithConsistancyOptions(t *testing.T) {
	c := con()

	insert, err := Bucket("raw", "default").SetRaw([]byte("Hello Delete")).Run(c)

	if err != nil {
		t.Error(err)
	}

	expectedError := "ClientError|[Cluster] all retries exhausted and/or no nodes available to execute command|InnerError|RiakError|0|{n_val_violation,3}"

	_, err = Bucket("raw", "default").Delete(insert.Key).WithDw(4).Run(c)

	if err.Error() != expectedError {
		t.Error(err)
	}

	_, err = Bucket("raw", "default").Delete(insert.Key).WithPw(4).Run(c)

	if err.Error() != expectedError {
		t.Error(err)
	}

	_, err = Bucket("raw", "default").Delete(insert.Key).WithPr(4).Run(c)

	if err.Error() != expectedError {
		t.Error(err)
	}

	_, err = Bucket("raw", "default").Delete(insert.Key).WithPw(4).Run(c)

	if err.Error() != expectedError {
		t.Error(err)
	}

	_, err = Bucket("raw", "default").Delete(insert.Key).WithR(4).Run(c)

	if err.Error() != expectedError {
		t.Error(err)
	}

	_, err = Bucket("raw", "default").Delete(insert.Key).WithW(4).Run(c)

	if err.Error() != expectedError {
		t.Error(err)
	}
}

func TestUpdateHllWithConsistancyOptions(t *testing.T) {
	_, err := Bucket("raw", "hlls").UpdateHyperLogLog().
		Add([]byte{1}).
		WithDw(5).
		WithPw(5).
		WithW(5).
		Run(con())

	if err == nil {
		t.Error("no error")
	}
}

func TestGetHllWithConsistancyOptions(t *testing.T) {
	_, err := Bucket("raw", "hlls").GetHyperLogLog("123").
		WithPr(5).
		WithR(5).
		Run(con())

	if err == nil {
		t.Error("no error")
	}
}