go-vise

commit 2b636b9f00efb6f0556c0c19aa0116dfc61c5da3
parent c20d557a3dbbb71a54cc4e6a10e9096ef72af5cb
Author: lash <dev@holbrook.no>
Date:   Mon, 23 Sep 2024 18:36:20 +0100

Add tests for faulty persist across engine issue

Diffstat:
A
engine/persist_test.go
 | 
141
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

1 file changed, 141 insertions(+), 0 deletions(-)
diff --git a/engine/persist_test.go b/engine/persist_test.go
@@ -0,0 +1,141 @@
+package engine
+
+import (
+	"bytes"
+	"context"
+	"testing"
+
+	"git.defalsify.org/vise.git/cache"
+	"git.defalsify.org/vise.git/persist"
+	"git.defalsify.org/vise.git/state"
+	memdb "git.defalsify.org/vise.git/db/mem"
+)
+
+
+func TestPersistNewAcrossEngine(t *testing.T) {
+	var err error
+	var cfg Config
+	generateTestData(t)
+	st := state.NewState(1)
+	ca := cache.NewCache()
+	rs := newTestWrapper(dataDir, st)
+	ctx := context.Background()
+	store := memdb.NewMemDb()
+	store.Connect(ctx, "")
+	pe := persist.NewPersister(store)
+	en := NewEngine(cfg, rs)
+	en = en.WithState(st)
+	en = en.WithMemory(ca)
+	en = en.WithPersister(pe)
+	cont, err := en.Exec(ctx, []byte{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !cont {
+		t.Errorf("expected cont")
+	}
+
+	r := bytes.NewBuffer(nil)
+	_, err = en.Flush(ctx, r)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	cont, err = en.Exec(ctx, []byte("1"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !cont {
+		t.Errorf("expected cont")
+	}
+	r = bytes.NewBuffer(nil)
+	_, err = en.Flush(ctx, r)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = en.Finish()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	cfg.FlagCount = 1
+	pe = persist.NewPersister(store)
+	en = NewEngine(cfg, rs)
+	en = en.WithPersister(pe)
+	cont, err = en.Exec(ctx, []byte{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !cont {
+		t.Errorf("expected cont")
+	}
+	location, _ := st.Where()
+	if location != "foo" {
+		t.Errorf("expected location 'foo', got '%s", location)	
+	}
+}
+
+func TestPersistSameAcrossEngine(t *testing.T) {
+	var err error
+	var cfg Config
+	generateTestData(t)
+	st := state.NewState(1)
+	ca := cache.NewCache()
+	rs := newTestWrapper(dataDir, st)
+	ctx := context.Background()
+	store := memdb.NewMemDb()
+	store.Connect(ctx, "")
+	pe := persist.NewPersister(store)
+	pe = pe.WithFlush()
+	en := NewEngine(cfg, rs)
+	en = en.WithState(st)
+	en = en.WithMemory(ca)
+	en = en.WithPersister(pe)
+	cont, err := en.Exec(ctx, []byte{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !cont {
+		t.Errorf("expected cont")
+	}
+
+	r := bytes.NewBuffer(nil)
+	_, err = en.Flush(ctx, r)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	cont, err = en.Exec(ctx, []byte("1"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !cont {
+		t.Errorf("expected cont")
+	}
+	r = bytes.NewBuffer(nil)
+	_, err = en.Flush(ctx, r)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = en.Finish()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	cfg.FlagCount = 1
+	en = NewEngine(cfg, rs)
+	en = en.WithPersister(pe)
+	cont, err = en.Exec(ctx, []byte{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !cont {
+		t.Errorf("expected cont")
+	}
+	location, _ := st.Where()
+	if location != "foo" {
+		t.Errorf("expected location 'foo', got '%s", location)	
+	}
+}