chore: import upstream snapshot with attribution

luaplugin/api_store_test.go

@@ -0,0 +1,125 @@
+package luaplugin
+
+import (
+	"testing"
+
+	lua "github.com/yuin/gopher-lua"
+)
+
+// newStoreState returns an LState with cliamp.store registered for pluginName.
+func newStoreState(t *testing.T, pluginName string) *lua.LState {
+	t.Helper()
+	L := lua.NewState()
+	t.Cleanup(L.Close)
+	cliamp := L.NewTable()
+	registerStoreAPI(L, cliamp, pluginName)
+	L.SetGlobal("cliamp", cliamp)
+	return L
+}
+
+func TestStoreSetGetRoundTrip(t *testing.T) {
+	t.Setenv("HOME", t.TempDir())
+	L := newStoreState(t, "rt")
+
+	if err := L.DoString(`
+		cliamp.store.set("str", "hello")
+		cliamp.store.set("num", 42)
+		cliamp.store.set("flag", true)
+		cliamp.store.set("tbl", {a = 1, b = {2, 3}})
+		_G.s = cliamp.store.get("str")
+		_G.n = cliamp.store.get("num")
+		_G.f = cliamp.store.get("flag")
+		_G.t_a = cliamp.store.get("tbl").a
+		_G.t_b2 = cliamp.store.get("tbl").b[2]
+		_G.missing = cliamp.store.get("nope")
+	`); err != nil {
+		t.Fatal(err)
+	}
+
+	if got := L.GetGlobal("s").String(); got != "hello" {
+		t.Errorf("str = %q", got)
+	}
+	if got := float64(L.GetGlobal("n").(lua.LNumber)); got != 42 {
+		t.Errorf("num = %v", got)
+	}
+	if got := bool(L.GetGlobal("f").(lua.LBool)); !got {
+		t.Errorf("flag = %v", got)
+	}
+	if got := float64(L.GetGlobal("t_a").(lua.LNumber)); got != 1 {
+		t.Errorf("tbl.a = %v", got)
+	}
+	if got := float64(L.GetGlobal("t_b2").(lua.LNumber)); got != 3 {
+		t.Errorf("tbl.b[2] = %v", got)
+	}
+	if L.GetGlobal("missing") != lua.LNil {
+		t.Errorf("missing key should be nil, got %v", L.GetGlobal("missing"))
+	}
+}
+
+func TestStorePersistsAcrossInstances(t *testing.T) {
+	t.Setenv("HOME", t.TempDir())
+
+	L1 := newStoreState(t, "persist")
+	if err := L1.DoString(`cliamp.store.set("count", 7)`); err != nil {
+		t.Fatal(err)
+	}
+
+	// Fresh LState + fresh store object for the same plugin name reads from disk.
+	L2 := newStoreState(t, "persist")
+	if err := L2.DoString(`_G.v = cliamp.store.get("count")`); err != nil {
+		t.Fatal(err)
+	}
+	if got := float64(L2.GetGlobal("v").(lua.LNumber)); got != 7 {
+		t.Fatalf("persisted count = %v, want 7", got)
+	}
+}
+
+func TestStoreNamespaceIsolation(t *testing.T) {
+	t.Setenv("HOME", t.TempDir())
+
+	La := newStoreState(t, "plugin-a")
+	if err := La.DoString(`cliamp.store.set("secret", "a-only")`); err != nil {
+		t.Fatal(err)
+	}
+
+	Lb := newStoreState(t, "plugin-b")
+	if err := Lb.DoString(`_G.v = cliamp.store.get("secret")`); err != nil {
+		t.Fatal(err)
+	}
+	if Lb.GetGlobal("v") != lua.LNil {
+		t.Fatalf("plugin-b read plugin-a key: %v", Lb.GetGlobal("v"))
+	}
+}
+
+func TestStoreKeysAndClear(t *testing.T) {
+	t.Setenv("HOME", t.TempDir())
+	L := newStoreState(t, "kc")
+
+	if err := L.DoString(`
+		cliamp.store.set("b", 1)
+		cliamp.store.set("a", 2)
+		cliamp.store.set("c", 3)
+		_G.keys = cliamp.store.keys()
+		cliamp.store.delete("b")
+		_G.afterDelete = #cliamp.store.keys()
+		cliamp.store.clear()
+		_G.afterClear = #cliamp.store.keys()
+	`); err != nil {
+		t.Fatal(err)
+	}
+
+	keys := L.GetGlobal("keys").(*lua.LTable)
+	if keys.Len() != 3 {
+		t.Fatalf("keys len = %d, want 3", keys.Len())
+	}
+	// keys() is sorted.
+	if keys.RawGetInt(1).String() != "a" || keys.RawGetInt(3).String() != "c" {
+		t.Errorf("keys not sorted: %v, %v", keys.RawGetInt(1), keys.RawGetInt(3))
+	}
+	if got := float64(L.GetGlobal("afterDelete").(lua.LNumber)); got != 2 {
+		t.Errorf("after delete = %v, want 2", got)
+	}
+	if got := float64(L.GetGlobal("afterClear").(lua.LNumber)); got != 0 {
+		t.Errorf("after clear = %v, want 0", got)
+	}
+}