diff options
Diffstat (limited to 'test/sandbox/assert.go')
| -rw-r--r-- | test/sandbox/assert.go | 31 |
1 files changed, 24 insertions, 7 deletions
diff --git a/test/sandbox/assert.go b/test/sandbox/assert.go index 2decde8f..4df2774a 100644 --- a/test/sandbox/assert.go +++ b/test/sandbox/assert.go @@ -2,6 +2,7 @@ package sandbox import ( "encoding/json" + "io/fs" "log" "os" ) @@ -15,6 +16,16 @@ var ( func printf(format string, v ...any) { printfFunc(format, v...) } func fatalf(format string, v ...any) { fatalfFunc(format, v...) } +func mustDecode(wantFile string, v any) { + if f, err := os.Open(wantFile); err != nil { + fatalf("cannot open %q: %v", wantFile, err) + } else if err = json.NewDecoder(f).Decode(v); err != nil { + fatalf("cannot decode %q: %v", wantFile, err) + } else if err = f.Close(); err != nil { + fatalf("cannot close %q: %v", wantFile, err) + } +} + func MustAssertMounts(name, hostMountsFile, wantFile string) { hostMounts := make([]*Mntent, 0, 128) if err := IterMounts(hostMountsFile, func(e *Mntent) { @@ -24,13 +35,7 @@ func MustAssertMounts(name, hostMountsFile, wantFile string) { } var want []Mntent - if f, err := os.Open(wantFile); err != nil { - fatalf("cannot open %q: %v", wantFile, err) - } else if err = json.NewDecoder(f).Decode(&want); err != nil { - fatalf("cannot decode %q: %v", wantFile, err) - } else if err = f.Close(); err != nil { - fatalf("cannot close %q: %v", wantFile, err) - } + mustDecode(wantFile, &want) for i := range want { if want[i].Opts == "host_passthrough" { @@ -61,3 +66,15 @@ func MustAssertMounts(name, hostMountsFile, wantFile string) { fatalf("cannot iterate mounts: %v", err) } } + +func MustAssertFS(e fs.FS, wantFile string) { + var want *FS + mustDecode(wantFile, &want) + if want == nil { + fatalf("invalid payload") + } + + if err := want.Compare(".", e); err != nil { + fatalf("%v", err) + } +} |