aboutsummaryrefslogtreecommitdiffhomepage
path: root/internal/state/segment_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/state/segment_test.go')
-rw-r--r--internal/state/segment_test.go259
1 files changed, 259 insertions, 0 deletions
diff --git a/internal/state/segment_test.go b/internal/state/segment_test.go
new file mode 100644
index 00000000..834cb99a
--- /dev/null
+++ b/internal/state/segment_test.go
@@ -0,0 +1,259 @@
+package state
+
+import (
+ "errors"
+ "iter"
+ "os"
+ "reflect"
+ "slices"
+ "strings"
+ "syscall"
+ "testing"
+
+ "hakurei.app/container/check"
+ "hakurei.app/container/stub"
+ "hakurei.app/hst"
+ "hakurei.app/internal/lockedfile"
+)
+
+func TestStateEntryHandle(t *testing.T) {
+ t.Parallel()
+
+ t.Run("lockout", func(t *testing.T) {
+ t.Parallel()
+ wantErr := func() error { return stub.UniqueError(0) }
+ eh := stateEntryHandle{decodeErr: wantErr(), pathname: check.MustAbs("/proc/nonexistent")}
+
+ if _, err := eh.open(-1, 0); !reflect.DeepEqual(err, wantErr()) {
+ t.Errorf("open: error = %v, want %v", err, wantErr())
+ }
+ if err := eh.destroy(); !reflect.DeepEqual(err, wantErr()) {
+ t.Errorf("destroy: error = %v, want %v", err, wantErr())
+ }
+ if err := eh.save(nil); !reflect.DeepEqual(err, wantErr()) {
+ t.Errorf("save: error = %v, want %v", err, wantErr())
+ }
+ if _, err := eh.load(nil); !reflect.DeepEqual(err, wantErr()) {
+ t.Errorf("load: error = %v, want %v", err, wantErr())
+ }
+ })
+
+ t.Run("od", func(t *testing.T) {
+ t.Parallel()
+
+ {
+ eh := stateEntryHandle{pathname: check.MustAbs(t.TempDir()).Append("entry")}
+ if f, err := eh.open(os.O_CREATE|syscall.O_EXCL, 0); err != nil {
+ t.Fatalf("open: error = %v", err)
+ } else if err = f.Close(); err != nil {
+ t.Errorf("Close: error = %v", err)
+ }
+ if err := eh.destroy(); err != nil {
+ t.Fatalf("destroy: error = %v", err)
+ }
+ }
+
+ t.Run("nonexistent", func(t *testing.T) {
+ t.Parallel()
+ eh := stateEntryHandle{pathname: check.MustAbs("/proc/nonexistent")}
+
+ wantErrOpen := &hst.AppError{Step: "open state entry",
+ Err: &os.PathError{Op: "open", Path: "/proc/nonexistent", Err: syscall.ENOENT}}
+ if _, err := eh.open(os.O_CREATE|syscall.O_EXCL, 0); !reflect.DeepEqual(err, wantErrOpen) {
+ t.Errorf("open: error = %#v, want %#v", err, wantErrOpen)
+ }
+
+ wantErrDestroy := &hst.AppError{Step: "destroy state entry",
+ Err: &os.PathError{Op: "remove", Path: "/proc/nonexistent", Err: syscall.ENOENT}}
+ if err := eh.destroy(); !reflect.DeepEqual(err, wantErrDestroy) {
+ t.Errorf("destroy: error = %#v, want %#v", err, wantErrDestroy)
+ }
+ })
+ })
+
+ t.Run("saveload", func(t *testing.T) {
+ t.Parallel()
+ eh := stateEntryHandle{pathname: check.MustAbs(t.TempDir()).Append("entry"),
+ ID: newTemplateState().ID}
+
+ if err := eh.save(newTemplateState()); err != nil {
+ t.Fatalf("save: error = %v", err)
+ }
+
+ t.Run("validate", func(t *testing.T) {
+ t.Parallel()
+
+ t.Run("internal", func(t *testing.T) {
+ t.Parallel()
+
+ var got hst.State
+ if f, err := os.Open(eh.pathname.String()); err != nil {
+ t.Fatal(err.Error())
+ } else if _, err = entryDecode(f, &got); err != nil {
+ t.Fatalf("entryDecode: error = %v", err)
+ } else if err = f.Close(); err != nil {
+ t.Fatal(f.Close())
+ }
+
+ if want := newTemplateState(); !reflect.DeepEqual(&got, want) {
+ t.Errorf("entryDecode: %#v, want %#v", &got, want)
+ }
+ })
+
+ t.Run("load header only", func(t *testing.T) {
+ t.Parallel()
+
+ if et, err := eh.load(nil); err != nil {
+ t.Fatalf("load: error = %v", err)
+ } else if want := newTemplateState().Enablements.Unwrap(); et != want {
+ t.Errorf("load: et = %x, want %x", et, want)
+ }
+ })
+
+ t.Run("load", func(t *testing.T) {
+ t.Parallel()
+
+ var got hst.State
+ if _, err := eh.load(&got); err != nil {
+ t.Fatalf("load: error = %v", err)
+ } else if want := newTemplateState(); !reflect.DeepEqual(&got, want) {
+ t.Errorf("load: %#v, want %#v", &got, want)
+ }
+ })
+
+ t.Run("load inconsistent", func(t *testing.T) {
+ t.Parallel()
+ wantErr := &hst.AppError{Step: "validate state identifier", Err: os.ErrInvalid,
+ Msg: "state entry 00000000000000000000000000000000 has unexpected id aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"}
+
+ ehi := stateEntryHandle{pathname: eh.pathname}
+ if _, err := ehi.load(new(hst.State)); !reflect.DeepEqual(err, wantErr) {
+ t.Errorf("load: error = %#v, want %#v", err, wantErr)
+ }
+ })
+ })
+ })
+}
+
+func TestStoreHandle(t *testing.T) {
+ t.Parallel()
+
+ testCases := []struct {
+ name string
+ ents [2][]string
+ want func(newEh func(err error, name string) *stateEntryHandle) []*stateEntryHandle
+ ext func(t *testing.T, entries iter.Seq[*stateEntryHandle], n int)
+ }{
+ {"errors", [2][]string{{
+ "e81eb203b4190ac5c3842ef44d429945",
+ "lock",
+ "f0-invalid",
+ }, {
+ "f1-directory",
+ }}, func(newEh func(err error, name string) *stateEntryHandle) []*stateEntryHandle {
+ return []*stateEntryHandle{
+ newEh(nil, "e81eb203b4190ac5c3842ef44d429945"),
+ newEh(&hst.AppError{Step: "decode store segment entry",
+ Err: hst.IdentifierDecodeError{Err: hst.ErrIdentifierLength}}, "f0-invalid"),
+ newEh(&hst.AppError{Step: "read store segment entries",
+ Err: errors.New(`unexpected directory "f1-directory" in store`)}, "f1-directory"),
+ }
+ }, nil},
+
+ {"success", [2][]string{{
+ "e81eb203b4190ac5c3842ef44d429945",
+ "7958cfbb9272d9cf9cfd61c85afa13f1",
+ "d0b5f7446dd5bd3424ff2f7ac9cace1e",
+ "c8c8e2c4aea5c32fe47240ff8caa874e",
+ "fa0d30b249d80f155a1f80ceddcc32f2",
+ "lock",
+ }}, func(newEh func(err error, name string) *stateEntryHandle) []*stateEntryHandle {
+ return []*stateEntryHandle{
+ newEh(nil, "7958cfbb9272d9cf9cfd61c85afa13f1"),
+ newEh(nil, "c8c8e2c4aea5c32fe47240ff8caa874e"),
+ newEh(nil, "d0b5f7446dd5bd3424ff2f7ac9cace1e"),
+ newEh(nil, "e81eb203b4190ac5c3842ef44d429945"),
+ newEh(nil, "fa0d30b249d80f155a1f80ceddcc32f2"),
+ }
+ }, func(t *testing.T, entries iter.Seq[*stateEntryHandle], n int) {
+ if n != 5 {
+ t.Fatalf("entries: n = %d", n)
+ }
+
+ // check partial drain
+ for range entries {
+ break
+ }
+ }},
+ }
+ for _, tc := range testCases {
+ t.Run(tc.name, func(t *testing.T) {
+ t.Parallel()
+
+ p := check.MustAbs(t.TempDir()).Append("segment")
+ if err := os.Mkdir(p.String(), 0700); err != nil {
+ t.Fatal(err.Error())
+ }
+ createEntries(t, p, tc.ents)
+
+ var got []*stateEntryHandle
+ if entries, n, err := (&storeHandle{
+ identity: -0xbad,
+ path: p,
+ fileMu: lockedfile.MutexAt(p.Append("lock").String()),
+ }).entries(); err != nil {
+ t.Fatalf("entries: error = %v", err)
+ } else {
+ got = slices.AppendSeq(make([]*stateEntryHandle, 0, n), entries)
+ if tc.ext != nil {
+ tc.ext(t, entries, n)
+ }
+ }
+
+ slices.SortFunc(got, func(a, b *stateEntryHandle) int { return strings.Compare(a.pathname.String(), b.pathname.String()) })
+ want := tc.want(func(err error, name string) *stateEntryHandle {
+ eh := stateEntryHandle{decodeErr: err, pathname: p.Append(name)}
+ if err == nil {
+ if err = eh.UnmarshalText([]byte(name)); err != nil {
+ t.Fatalf("UnmarshalText: error = %v", err)
+ }
+ }
+ return &eh
+ })
+
+ if !reflect.DeepEqual(got, want) {
+ t.Errorf("entries: %q, want %q", got, want)
+ }
+ })
+ }
+
+ t.Run("nonexistent", func(t *testing.T) {
+ var wantErr = &hst.AppError{Step: "read store segment entries", Err: &os.PathError{
+ Op: "open",
+ Path: "/proc/nonexistent",
+ Err: syscall.ENOENT,
+ }}
+ if _, _, err := (&storeHandle{
+ identity: -0xbad,
+ path: check.MustAbs("/proc/nonexistent"),
+ }).entries(); !reflect.DeepEqual(err, wantErr) {
+ t.Fatalf("entries: error = %#v, want %#v", err, wantErr)
+ }
+ })
+}
+
+// createEntries creates file and directory entries in the specified prefix.
+func createEntries(t *testing.T, prefix *check.Absolute, ents [2][]string) {
+ for _, s := range ents[0] {
+ if f, err := os.OpenFile(prefix.Append(s).String(), os.O_CREATE|os.O_EXCL, 0600); err != nil {
+ t.Fatal(err.Error())
+ } else if err = f.Close(); err != nil {
+ t.Fatal(err.Error())
+ }
+ }
+ for _, s := range ents[1] {
+ if err := os.Mkdir(prefix.Append(s).String(), 0700); err != nil {
+ t.Fatal(err.Error())
+ }
+ }
+}
generated by cgit v1.3.1 (git 2.54.0) at 2026-08-30 00:12:39 +0000