mirror of
https://git.vectorsigma.ru/public/AdGuardHome.git
synced 2026-07-28 21:39:11 +00:00
5e2b4405fd
Merge in DNS/adguard-home from upd-all to master Squashed commit of the following: commit 72a41e9e2d9b9a5fbf24fa69322f9c4dcf3f5fb7 Author: Eugene Burkov <E.Burkov@AdGuard.COM> Date: Thu Aug 14 19:42:11 2025 +0300 specs: export tests commit aa729009306e492fd8a046b5e84f49be87e2a57f Author: Eugene Burkov <E.Burkov@AdGuard.COM> Date: Thu Aug 14 19:19:54 2025 +0300 all: upd golibs commit 526ce744cfdf167d1e5b763422092e6f450993a6 Author: Eugene Burkov <E.Burkov@AdGuard.COM> Date: Thu Aug 14 17:21:57 2025 +0300 all: upd scripts commit ecc4312764b31e8b84e462321e3690a9b8eebbf6 Author: Eugene Burkov <E.Burkov@AdGuard.COM> Date: Thu Aug 14 17:02:53 2025 +0300 all: upd go & tools
101 lines
2.1 KiB
Go
101 lines
2.1 KiB
Go
package aghrenameio_test
|
|
|
|
import (
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/aghrenameio"
|
|
"github.com/AdguardTeam/golibs/errors"
|
|
"github.com/AdguardTeam/golibs/testutil"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// testPerm is the common permission mode for tests.
|
|
const testPerm fs.FileMode = 0o644
|
|
|
|
// Common file data for tests.
|
|
var (
|
|
initialData = []byte("initial data\n")
|
|
newData = []byte("new data\n")
|
|
)
|
|
|
|
func TestPendingFile(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
targetPath := newInitialFile(t)
|
|
f, err := aghrenameio.NewPendingFile(targetPath, testPerm)
|
|
require.NoError(t, err)
|
|
|
|
_, err = f.Write(newData)
|
|
require.NoError(t, err)
|
|
|
|
err = f.CloseReplace()
|
|
require.NoError(t, err)
|
|
|
|
gotData, err := os.ReadFile(targetPath)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, newData, gotData)
|
|
}
|
|
|
|
// newInitialFile is a test helper that returns the path to the file containing
|
|
// [initialData].
|
|
func newInitialFile(tb testing.TB) (targetPath string) {
|
|
tb.Helper()
|
|
|
|
dir := tb.TempDir()
|
|
targetPath = filepath.Join(dir, "target")
|
|
|
|
err := os.WriteFile(targetPath, initialData, 0o644)
|
|
require.NoError(tb, err)
|
|
|
|
return targetPath
|
|
}
|
|
|
|
func TestWithDeferredCleanup(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
const testError errors.Error = "test error"
|
|
|
|
testCases := []struct {
|
|
error error
|
|
name string
|
|
wantErrMsg string
|
|
wantData []byte
|
|
}{{
|
|
name: "success",
|
|
error: nil,
|
|
wantErrMsg: "",
|
|
wantData: newData,
|
|
}, {
|
|
name: "error",
|
|
error: testError,
|
|
wantErrMsg: testError.Error(),
|
|
wantData: initialData,
|
|
}}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
targetPath := newInitialFile(t)
|
|
f, err := aghrenameio.NewPendingFile(targetPath, testPerm)
|
|
require.NoError(t, err)
|
|
|
|
_, err = f.Write(newData)
|
|
require.NoError(t, err)
|
|
|
|
err = aghrenameio.WithDeferredCleanup(tc.error, f)
|
|
testutil.AssertErrorMsg(t, tc.wantErrMsg, err)
|
|
|
|
gotData, err := os.ReadFile(targetPath)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, tc.wantData, gotData)
|
|
})
|
|
}
|
|
}
|