mirror of
https://github.com/tailscale/tailscale.git
synced 2025-08-13 22:47:30 +00:00
d897d809d6
In Android, we are prompting the user to select a Taildrop directory when they first receive a Taildrop: we block writes on Taildrop dir selection. This means that we cannot use Dir inside managerOptions, since the http request would not get the new Taildrop extension. This PR removes, in the Android case, the reliance on m.opts.Dir, and instead has FileOps hold the correct directory. This expands FileOps to be the Taildrop interface for all file system operations. Updates tailscale/corp#29211 Signed-off-by: kari-ts <kari@tailscale.com> restore tstest
70 lines
1.5 KiB
Go
70 lines
1.5 KiB
Go
// Copyright (c) Tailscale Inc & AUTHORS
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package taildrop
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"tailscale.com/tstime"
|
|
"tailscale.com/util/must"
|
|
)
|
|
|
|
func TestPutFile(t *testing.T) {
|
|
const content = "hello, world"
|
|
|
|
tests := []struct {
|
|
name string
|
|
directFileMode bool
|
|
}{
|
|
{"DirectFileMode", true},
|
|
{"NonDirectFileMode", false},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
dir := t.TempDir()
|
|
mgr := managerOptions{
|
|
Logf: t.Logf,
|
|
Clock: tstime.DefaultClock{},
|
|
State: nil,
|
|
fileOps: must.Get(newFileOps(dir)),
|
|
DirectFileMode: tt.directFileMode,
|
|
SendFileNotify: func() {},
|
|
}.New()
|
|
|
|
id := clientID("0")
|
|
n, err := mgr.PutFile(id, "file.txt", strings.NewReader(content), 0, int64(len(content)))
|
|
if err != nil {
|
|
t.Fatalf("PutFile error: %v", err)
|
|
}
|
|
if n != int64(len(content)) {
|
|
t.Errorf("wrote %d bytes; want %d", n, len(content))
|
|
}
|
|
|
|
path := filepath.Join(dir, "file.txt")
|
|
|
|
got, err := os.ReadFile(path)
|
|
if err != nil {
|
|
t.Fatalf("ReadFile %q: %v", path, err)
|
|
}
|
|
if string(got) != content {
|
|
t.Errorf("file contents = %q; want %q", string(got), content)
|
|
}
|
|
|
|
entries, err := os.ReadDir(dir)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, entry := range entries {
|
|
if strings.Contains(entry.Name(), ".partial") {
|
|
t.Errorf("unexpected partial file left behind: %s", entry.Name())
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|