mirror of
https://github.com/tailscale/tailscale.git
synced 2024-12-12 03:04:40 +00:00
d603d18956
Previously, the test simply relied on: defer close() to cleanup file handles. This works fine on Unix-based systems, but not on Windows, which dislikes deleting files where an open file handle continues to exist. Fix the test by explicitly closing the file handle after we are done with the resource. Updates tailscale/corp#14772 Signed-off-by: Joe Tsai <joetsai@digital-static.net>
75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
// Copyright (c) Tailscale Inc & AUTHORS
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package taildrop
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"math/rand"
|
|
"os"
|
|
"testing"
|
|
"testing/iotest"
|
|
|
|
"tailscale.com/util/must"
|
|
)
|
|
|
|
func TestResume(t *testing.T) {
|
|
oldBlockSize := blockSize
|
|
defer func() { blockSize = oldBlockSize }()
|
|
blockSize = 256
|
|
|
|
m := ManagerOptions{Logf: t.Logf, Dir: t.TempDir()}.New()
|
|
defer m.Shutdown()
|
|
|
|
rn := rand.New(rand.NewSource(0))
|
|
want := make([]byte, 12345)
|
|
must.Get(io.ReadFull(rn, want))
|
|
|
|
t.Run("resume-noexist", func(t *testing.T) {
|
|
r := io.Reader(bytes.NewReader(want))
|
|
|
|
next, close, err := m.HashPartialFile("", "foo")
|
|
must.Do(err)
|
|
defer close()
|
|
offset, r, err := ResumeReader(r, next)
|
|
must.Do(err)
|
|
must.Do(close()) // Windows wants the file handle to be closed to rename it.
|
|
|
|
must.Get(m.PutFile("", "foo", r, offset, -1))
|
|
got := must.Get(os.ReadFile(must.Get(joinDir(m.opts.Dir, "foo"))))
|
|
if !bytes.Equal(got, want) {
|
|
t.Errorf("content mismatches")
|
|
}
|
|
})
|
|
|
|
t.Run("resume-retry", func(t *testing.T) {
|
|
rn := rand.New(rand.NewSource(0))
|
|
for i := 0; true; i++ {
|
|
r := io.Reader(bytes.NewReader(want))
|
|
|
|
next, close, err := m.HashPartialFile("", "bar")
|
|
must.Do(err)
|
|
defer close()
|
|
offset, r, err := ResumeReader(r, next)
|
|
must.Do(err)
|
|
must.Do(close()) // Windows wants the file handle to be closed to rename it.
|
|
|
|
numWant := rn.Int63n(min(int64(len(want))-offset, 1000) + 1)
|
|
if offset < int64(len(want)) {
|
|
r = io.MultiReader(io.LimitReader(r, numWant), iotest.ErrReader(io.ErrClosedPipe))
|
|
}
|
|
if _, err := m.PutFile("", "bar", r, offset, -1); err == nil {
|
|
break
|
|
}
|
|
if i > 1000 {
|
|
t.Fatalf("too many iterations to complete the test")
|
|
}
|
|
}
|
|
got := must.Get(os.ReadFile(must.Get(joinDir(m.opts.Dir, "bar"))))
|
|
if !bytes.Equal(got, want) {
|
|
t.Errorf("content mismatches")
|
|
}
|
|
})
|
|
}
|