diff options
| author | Ophestra <cat@gensokyo.uk> | 2025-10-30 18:43:55 +0900 |
|---|---|---|
| committer | Ophestra <cat@gensokyo.uk> | 2025-10-30 18:43:55 +0900 |
| commit | ebdcff1049fb979405677a823a47410df7ee7c4f (patch) | |
| tree | a5a14d409e4bc9f3b4b2e2c15f6c23de5ad0755f /internal/state/segment_test.go | |
| parent | 46c5ce493638ea2d034bbbdb9449e93b66d0330f (diff) | |
internal/store: rename from state
This reduces collision with local variable names, and generally makes sense for the new store package, since it no longer specifies the state struct.
Signed-off-by: Ophestra <cat@gensokyo.uk>
Diffstat (limited to 'internal/state/segment_test.go')
| -rw-r--r-- | internal/state/segment_test.go | 259 |
1 files changed, 0 insertions, 259 deletions
diff --git a/internal/state/segment_test.go b/internal/state/segment_test.go deleted file mode 100644 index 834cb99a..00000000 --- a/internal/state/segment_test.go +++ /dev/null @@ -1,259 +0,0 @@ -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()) - } - } -} |
