From 49600a6f46d99ff35d883381670548de1f01846e Mon Sep 17 00:00:00 2001 From: Ophestra Date: Sun, 31 Aug 2025 23:11:25 +0900 Subject: container/stub: export stub helpers These are very useful in many packages containing relatively large amount of code making calls to difficult or impossible to stub functions. Signed-off-by: Ophestra --- container/initsymlink_test.go | 30 ++++++++++++++++-------------- 1 file changed, 16 insertions(+), 14 deletions(-) (limited to 'container/initsymlink_test.go') diff --git a/container/initsymlink_test.go b/container/initsymlink_test.go index db1c1a5e..8e94ead5 100644 --- a/container/initsymlink_test.go +++ b/container/initsymlink_test.go @@ -3,6 +3,8 @@ package container import ( "os" "testing" + + "hakurei.app/container/stub" ) func TestSymlinkOp(t *testing.T) { @@ -10,9 +12,9 @@ func TestSymlinkOp(t *testing.T) { {"mkdir", &Params{ParentPerm: 0700}, &SymlinkOp{ Target: MustAbs("/etc/nixos"), LinkName: "/etc/static/nixos", - }, nil, nil, []kexpect{ - {"mkdirAll", expectArgs{"/sysroot/etc", os.FileMode(0700)}, nil, errUnique}, - }, errUnique}, + }, nil, nil, []stub.Call{ + {"mkdirAll", stub.ExpectArgs{"/sysroot/etc", os.FileMode(0700)}, nil, stub.UniqueError(1)}, + }, stub.UniqueError(1)}, {"abs", &Params{ParentPerm: 0755}, &SymlinkOp{ Target: MustAbs("/etc/mtab"), @@ -24,27 +26,27 @@ func TestSymlinkOp(t *testing.T) { Target: MustAbs("/etc/mtab"), LinkName: "/etc/mtab", Dereference: true, - }, []kexpect{ - {"readlink", expectArgs{"/etc/mtab"}, "/proc/mounts", errUnique}, - }, errUnique, nil, nil}, + }, []stub.Call{ + {"readlink", stub.ExpectArgs{"/etc/mtab"}, "/proc/mounts", stub.UniqueError(0)}, + }, stub.UniqueError(0), nil, nil}, {"success noderef", &Params{ParentPerm: 0700}, &SymlinkOp{ Target: MustAbs("/etc/nixos"), LinkName: "/etc/static/nixos", - }, nil, nil, []kexpect{ - {"mkdirAll", expectArgs{"/sysroot/etc", os.FileMode(0700)}, nil, nil}, - {"symlink", expectArgs{"/etc/static/nixos", "/sysroot/etc/nixos"}, nil, nil}, + }, nil, nil, []stub.Call{ + {"mkdirAll", stub.ExpectArgs{"/sysroot/etc", os.FileMode(0700)}, nil, nil}, + {"symlink", stub.ExpectArgs{"/etc/static/nixos", "/sysroot/etc/nixos"}, nil, nil}, }, nil}, {"success", &Params{ParentPerm: 0755}, &SymlinkOp{ Target: MustAbs("/etc/mtab"), LinkName: "/etc/mtab", Dereference: true, - }, []kexpect{ - {"readlink", expectArgs{"/etc/mtab"}, "/proc/mounts", nil}, - }, nil, []kexpect{ - {"mkdirAll", expectArgs{"/sysroot/etc", os.FileMode(0755)}, nil, nil}, - {"symlink", expectArgs{"/proc/mounts", "/sysroot/etc/mtab"}, nil, nil}, + }, []stub.Call{ + {"readlink", stub.ExpectArgs{"/etc/mtab"}, "/proc/mounts", nil}, + }, nil, []stub.Call{ + {"mkdirAll", stub.ExpectArgs{"/sysroot/etc", os.FileMode(0755)}, nil, nil}, + {"symlink", stub.ExpectArgs{"/proc/mounts", "/sysroot/etc/mtab"}, nil, nil}, }, nil}, }) -- cgit v1.3.1