tx_set_test.go

package flashdb

import (
	"os"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestFlashDB_SCard(t *testing.T) {
	db := getTestDB()
	defer db.Close()
	defer os.RemoveAll(tmpDir)

	if err := db.Update(func(tx *Tx) error {
		tx.SAdd(testKey, "foo", "bar", "baz")
		return nil
	}); err != nil {
		t.Fatal(err)
	}

	db.View(func(tx *Tx) error {
		cnt := tx.SCard(testKey)
		assert.Equal(t, 3, cnt)
		return nil
	})
}

func TestFlashDB_SIsMember(t *testing.T) {
	db := getTestDB()
	defer db.Close()
	defer os.RemoveAll(tmpDir)

	if err := db.Update(func(tx *Tx) error {
		tx.SAdd(testKey, "foo", "bar", "baz")
		return nil
	}); err != nil {
		t.Fatal(err)
	}

	db.View(func(tx *Tx) error {
		assert.True(t, tx.SIsMember(testKey, "foo"))
		assert.True(t, tx.SIsMember(testKey, "bar"))
		assert.True(t, tx.SIsMember(testKey, "baz"))
		return nil
	})
}

func TestFlashDB_SRem(t *testing.T) {
	db := getTestDB()
	defer db.Close()
	defer os.RemoveAll(tmpDir)

	if err := db.Update(func(tx *Tx) error {
		tx.SAdd(testKey, "foo", "bar", "baz")
		tx.SRem(testKey, "foo")
		return nil
	}); err != nil {
		t.Fatal(err)
	}

	if err := db.View(func(tx *Tx) error {
		assert.False(t, tx.SIsMember(testKey, "foo"))
		return nil
	}); err != nil {
		t.Fatal(err)
	}

}

func TestFlashDB_SClear(t *testing.T) {
	db := getTestDB()
	defer db.Close()
	defer os.RemoveAll(tmpDir)

	if err := db.Update(func(tx *Tx) error {
		tx.SAdd(testKey, "foo", "bar", "baz")
		return nil
	}); err != nil {
		t.Fatal(err)
	}

	if err := db.Update(func(tx *Tx) error {
		resp := tx.SClear(testKey)
		assert.NoError(t, resp)
		return nil
	}); err != nil {
		t.Fatal(err)
	}

	if err := db.View(func(tx *Tx) error {
		assert.False(t, tx.SKeyExists(testKey))
		return nil
	}); err != nil {
		t.Fatal(err)
	}
}

func TestFlashDB_SDiff(t *testing.T) {
	db := getTestDB()
	defer db.Close()
	defer os.RemoveAll(tmpDir)

	if err := db.Update(func(tx *Tx) error {
		err := tx.SAdd("set1", "foo", "bar", "baz")
		assert.NoError(t, err)
		err = tx.SAdd("set2", "foo", "bar")
		assert.NoError(t, err)
		return nil
	}); err != nil {
		t.Fatal(err)
	}

	if err := db.View(func(tx *Tx) error {
		assert.Equal(t, 3, len(tx.SMembers("set1")))
		assert.Equal(t, 2, len(tx.SMembers("set2")))
		res := tx.SDiff("set1", "set2")
		assert.Equal(t, 1, len(res))
		assert.Equal(t, "baz", res[0])
		return nil
	}); err != nil {
		t.Fatal(err)
	}

}