Initial commit

2023-04-20 18:35:07 +00:00
parent 83a09f5789
commit 70f06033c2
8 changed files with 469 additions and 0 deletions
--- a/storebus_test.go
+++ b/storebus_test.go
@@ -0,0 +1,175 @@
+package storebus_test
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"testing"
+
+	"github.com/dchest/uniuri"
+	"github.com/gopatchy/metadata"
+	"github.com/gopatchy/storebus"
+	"github.com/stretchr/testify/require"
+)
+
+func TestStoreBus(t *testing.T) {
+	t.Parallel()
+
+	dir, err := os.MkdirTemp("", "")
+	require.NoError(t, err)
+
+	defer os.RemoveAll(dir)
+
+	ctx := context.Background()
+
+	dbname := fmt.Sprintf("file:%s?mode=memory&cache=shared", uniuri.New())
+
+	sb, err := storebus.NewStoreBus(dbname)
+	require.NoError(t, err)
+
+	defer sb.Close()
+
+	err = sb.Write(ctx, "storeBusTest", &storeBusTest{
+		Metadata: metadata.Metadata{
+			ID: "id1",
+		},
+		Opaque: "foo",
+	})
+	require.NoError(t, err)
+
+	c1, err := sb.ReadStream(ctx, "storeBusTest", "id1", newStoreBusTest)
+	require.NoError(t, err)
+
+	defer sb.CloseReadStream("storeBusTest", "id1", c1)
+
+	c2, err := sb.ListStream(ctx, "storeBusTest", newStoreBusTest)
+	require.NoError(t, err)
+
+	defer sb.CloseListStream("storeBusTest", c2)
+
+	out1 := (<-c1).(*storeBusTest)
+	require.Equal(t, "foo", out1.Opaque)
+	require.Equal(t, "etag:2c8edc6414452b8dee7826bd55e585f850ac47a0dcfc357dc1fcaaa3164cdfa2", out1.ETag)
+
+	l1 := <-c2
+	require.Len(t, l1, 1)
+	require.Equal(t, "foo", l1[0].(*storeBusTest).Opaque)
+	require.Equal(t, "etag:2c8edc6414452b8dee7826bd55e585f850ac47a0dcfc357dc1fcaaa3164cdfa2", l1[0].(*storeBusTest).ETag)
+
+	err = sb.Write(ctx, "storeBusTest", &storeBusTest{
+		Metadata: metadata.Metadata{
+			ID: "id1",
+		},
+		Opaque: "bar",
+	})
+	require.NoError(t, err)
+
+	out2 := (<-c1).(*storeBusTest)
+	require.Equal(t, "bar", out2.Opaque)
+	require.Equal(t, "etag:906fda69e9893280ca9294bd04eb276794da9a8904fc0b671c69175f08cc03c6", out2.ETag)
+
+	l2 := <-c2
+	require.Len(t, l2, 1)
+	require.Equal(t, "bar", l2[0].(*storeBusTest).Opaque)
+	require.Equal(t, "etag:906fda69e9893280ca9294bd04eb276794da9a8904fc0b671c69175f08cc03c6", l2[0].(*storeBusTest).ETag)
+
+	l2a, err := sb.List(ctx, "storeBusTest", newStoreBusTest)
+	require.NoError(t, err)
+	require.Len(t, l2a, 1)
+	require.Equal(t, "bar", l2a[0].(*storeBusTest).Opaque)
+	require.Equal(t, "etag:906fda69e9893280ca9294bd04eb276794da9a8904fc0b671c69175f08cc03c6", l2a[0].(*storeBusTest).ETag)
+
+	out2a, err := sb.Read(ctx, "storeBusTest", "id1", newStoreBusTest)
+	require.NoError(t, err)
+	require.Equal(t, "bar", out2a.(*storeBusTest).Opaque)
+	require.Equal(t, "etag:906fda69e9893280ca9294bd04eb276794da9a8904fc0b671c69175f08cc03c6", out2a.(*storeBusTest).ETag)
+
+	err = sb.Write(ctx, "storeBusTest", &storeBusTest{
+		Metadata: metadata.Metadata{
+			ID: "id2",
+		},
+		Opaque: "zig",
+	})
+	require.NoError(t, err)
+
+	l3 := <-c2
+	require.Len(t, l3, 2)
+	require.ElementsMatch(
+		t,
+		[]string{
+			l3[0].(*storeBusTest).Opaque,
+			l3[1].(*storeBusTest).Opaque,
+		},
+		[]string{
+			"bar",
+			"zig",
+		},
+	)
+
+	l3a, err := sb.List(ctx, "storeBusTest", newStoreBusTest)
+	require.NoError(t, err)
+	require.Len(t, l3a, 2)
+	require.ElementsMatch(
+		t,
+		[]string{
+			l3a[0].(*storeBusTest).Opaque,
+			l3a[1].(*storeBusTest).Opaque,
+		},
+		[]string{
+			"bar",
+			"zig",
+		},
+	)
+}
+
+func TestStoreBusDelete(t *testing.T) {
+	t.Parallel()
+
+	dir, err := os.MkdirTemp("", "")
+	require.NoError(t, err)
+
+	defer os.RemoveAll(dir)
+
+	ctx := context.Background()
+
+	dbname := fmt.Sprintf("file:%s?mode=memory&cache=shared", uniuri.New())
+
+	sb, err := storebus.NewStoreBus(dbname)
+	require.NoError(t, err)
+
+	defer sb.Close()
+
+	c1, err := sb.ReadStream(ctx, "storeBusTest", "id1", newStoreBusTest)
+	require.NoError(t, err)
+
+	defer sb.CloseReadStream("storeBusTest", "id1", c1)
+
+	preout := <-c1
+	require.Nil(t, preout)
+
+	err = sb.Write(ctx, "storeBusTest", &storeBusTest{
+		Metadata: metadata.Metadata{
+			ID: "id1",
+		},
+		Opaque: "foo",
+	})
+	require.NoError(t, err)
+
+	out := (<-c1).(*storeBusTest)
+	require.Equal(t, "foo", out.Opaque)
+
+	err = sb.Delete(ctx, "storeBusTest", "id1")
+	require.NoError(t, err)
+
+	_, ok := <-c1
+	require.False(t, ok)
+}
+
+type storeBusTest struct {
+	metadata.Metadata
+	Opaque string
+}
+
+func newStoreBusTest() any {
+	return &storeBusTest{}
+}