fix: rsync double SSH spec in RunRemote, restore dropped flags in Run
- Fix Bug A: RunRemote was re-prefixing destination with user@host: when engine.go:135 already pre-prefixed it for remote-to-remote, causing "admin@host:admin@host:/path" to be passed to rsync - Fix Bug B: Run used args[1:] when inserting -e ssh, silently dropping the first rsync flag (e.g. -aP became flag-less) - Fix Bug C: RunRemote extracted source/dest in wrong order for pull direction (buildArgs reverses them but RunRemote assumed push order) - Add rsync_runner_test.go covering buildArgs, RunRemote prefixing, and flag preservation
This commit is contained in:
@@ -0,0 +1,259 @@
|
||||
package syncengine
|
||||
|
||||
import (
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
type remoteCmdTest struct {
|
||||
srcHost string
|
||||
srcUser string
|
||||
srcPath string
|
||||
dstHost string
|
||||
dstUser string
|
||||
dstPath string
|
||||
direction string
|
||||
}
|
||||
|
||||
func (tc remoteCmdTest) build() (srcArg, dstArg string) {
|
||||
srcUserHost := tc.srcUser + "@" + tc.srcHost
|
||||
dstUserHost := tc.dstUser + "@" + tc.dstHost
|
||||
|
||||
runner := NewRsyncRunner("/tmp/ssh", "")
|
||||
args := runner.buildArgs(&SyncPairConfig{
|
||||
Source: tc.srcPath,
|
||||
Dest: tc.dstPath,
|
||||
Direction: tc.direction,
|
||||
})
|
||||
|
||||
var sourcePath, destPath string
|
||||
if tc.direction == "pull" {
|
||||
sourcePath = args[len(args)-1]
|
||||
destPath = args[len(args)-2]
|
||||
} else {
|
||||
sourcePath = args[len(args)-2]
|
||||
destPath = args[len(args)-1]
|
||||
}
|
||||
|
||||
var remoteSrc, remoteDst string
|
||||
if tc.direction == "pull" {
|
||||
remoteSrc = sourcePath
|
||||
remoteDst = destPath
|
||||
if !strings.Contains(destPath, "@") {
|
||||
remoteDst = dstUserHost + ":" + destPath
|
||||
}
|
||||
} else {
|
||||
remoteDst = destPath
|
||||
if !strings.Contains(destPath, "@") {
|
||||
remoteDst = dstUserHost + ":" + destPath
|
||||
}
|
||||
remoteSrc = srcUserHost + ":" + sourcePath
|
||||
}
|
||||
return remoteSrc, remoteDst
|
||||
}
|
||||
|
||||
func TestBuildArgs_Push(t *testing.T) {
|
||||
runner := NewRsyncRunner("/tmp/ssh", "")
|
||||
pair := &SyncPairConfig{
|
||||
Source: "/local/src",
|
||||
Dest: "admin@10.5.0.144:/remote/dst",
|
||||
Direction: "push",
|
||||
RsyncFlags: "-aP",
|
||||
ExcludePatterns: []string{},
|
||||
}
|
||||
args := runner.buildArgs(pair)
|
||||
if args[0] != "-aP" {
|
||||
t.Errorf("first flag = %q, want %q", args[0], "-aP")
|
||||
}
|
||||
if args[len(args)-2] != "/local/src" {
|
||||
t.Errorf("source = %q, want %q", args[len(args)-2], "/local/src")
|
||||
}
|
||||
if args[len(args)-1] != "admin@10.5.0.144:/remote/dst" {
|
||||
t.Errorf("dest = %q, want %q", args[len(args)-1], "admin@10.5.0.144:/remote/dst")
|
||||
}
|
||||
}
|
||||
|
||||
func TestBuildArgs_Pull(t *testing.T) {
|
||||
runner := NewRsyncRunner("/tmp/ssh", "")
|
||||
pair := &SyncPairConfig{
|
||||
Source: "admin@10.5.0.144:/remote/src",
|
||||
Dest: "/local/dst",
|
||||
Direction: "pull",
|
||||
RsyncFlags: "-aP",
|
||||
ExcludePatterns: []string{},
|
||||
}
|
||||
args := runner.buildArgs(pair)
|
||||
if args[len(args)-2] != "/local/dst" {
|
||||
t.Errorf("pull: second-to-last (dest) = %q, want %q", args[len(args)-2], "/local/dst")
|
||||
}
|
||||
if args[len(args)-1] != "admin@10.5.0.144:/remote/src" {
|
||||
t.Errorf("pull: last (source) = %q, want %q", args[len(args)-1], "admin@10.5.0.144:/remote/src")
|
||||
}
|
||||
}
|
||||
|
||||
func TestBuildArgs_Mirror(t *testing.T) {
|
||||
runner := NewRsyncRunner("/tmp/ssh", "")
|
||||
pair := &SyncPairConfig{
|
||||
Source: "/local/src",
|
||||
Dest: "admin@10.5.0.144:/remote/dst",
|
||||
Direction: "mirror",
|
||||
RsyncFlags: "-aP",
|
||||
ExcludePatterns: []string{},
|
||||
}
|
||||
args := runner.buildArgs(pair)
|
||||
foundDelete := false
|
||||
for _, a := range args {
|
||||
if a == "--delete" {
|
||||
foundDelete = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !foundDelete {
|
||||
t.Errorf("mirror args = %v, want --delete present", args)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBuildArgs_MultiTokenFlags(t *testing.T) {
|
||||
runner := NewRsyncRunner("/tmp/ssh", "")
|
||||
pair := &SyncPairConfig{
|
||||
Source: "/local/src",
|
||||
Dest: "admin@10.5.0.144:/remote/dst",
|
||||
Direction: "push",
|
||||
RsyncFlags: "-aP --partial",
|
||||
ExcludePatterns: []string{},
|
||||
}
|
||||
args := runner.buildArgs(pair)
|
||||
if args[0] != "-aP" {
|
||||
t.Errorf("first flag = %q, want %q", args[0], "-aP")
|
||||
}
|
||||
if args[1] != "--partial" {
|
||||
t.Errorf("second flag = %q, want %q", args[1], "--partial")
|
||||
}
|
||||
}
|
||||
|
||||
func TestBuildArgs_ExcludePatterns(t *testing.T) {
|
||||
runner := NewRsyncRunner("/tmp/ssh", "")
|
||||
pair := &SyncPairConfig{
|
||||
Source: "/local/src",
|
||||
Dest: "admin@10.5.0.144:/remote/dst",
|
||||
Direction: "push",
|
||||
RsyncFlags: "-aP",
|
||||
ExcludePatterns: []string{"*.tmp", ".DS_Store"},
|
||||
}
|
||||
args := runner.buildArgs(pair)
|
||||
var excludes []string
|
||||
for _, a := range args {
|
||||
if strings.HasPrefix(a, "--exclude=") {
|
||||
excludes = append(excludes, a)
|
||||
}
|
||||
}
|
||||
if len(excludes) != 2 {
|
||||
t.Errorf("excludes = %v, want 2 exclude entries", excludes)
|
||||
}
|
||||
if excludes[0] != "--exclude=*.tmp" {
|
||||
t.Errorf("exclude[0] = %q, want %q", excludes[0], "--exclude=*.tmp")
|
||||
}
|
||||
}
|
||||
|
||||
func TestRunRemote_PushDestNoDoublePrefix(t *testing.T) {
|
||||
tc := remoteCmdTest{
|
||||
srcHost: "10.5.1.10",
|
||||
srcUser: "root",
|
||||
srcPath: "/share/homes/admin/media",
|
||||
dstHost: "10.5.0.144",
|
||||
dstUser: "admin",
|
||||
dstPath: "admin@10.5.0.144:/share/media/peliculas",
|
||||
direction: "push",
|
||||
}
|
||||
_, dstArg := tc.build()
|
||||
|
||||
if countOccurrences(dstArg, "@") > 1 {
|
||||
t.Errorf("push dest %q has double SSH spec", dstArg)
|
||||
}
|
||||
if dstArg != "admin@10.5.0.144:/share/media/peliculas" {
|
||||
t.Errorf("push dest = %q, want 'admin@10.5.0.144:/share/media/peliculas'", dstArg)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRunRemote_PushSrcCorrectlyPrefixed(t *testing.T) {
|
||||
tc := remoteCmdTest{
|
||||
srcHost: "10.5.1.10",
|
||||
srcUser: "root",
|
||||
srcPath: "/share/homes/admin/media",
|
||||
dstHost: "10.5.0.144",
|
||||
dstUser: "admin",
|
||||
dstPath: "/share/media/peliculas",
|
||||
direction: "push",
|
||||
}
|
||||
srcArg, dstArg := tc.build()
|
||||
|
||||
if !strings.HasPrefix(srcArg, "root@10.5.1.10:/") {
|
||||
t.Errorf("push src = %q, want prefix 'root@10.5.1.10:/'", srcArg)
|
||||
}
|
||||
if !strings.HasPrefix(dstArg, "admin@10.5.0.144:/") {
|
||||
t.Errorf("push dst = %q, want prefix 'admin@10.5.0.144:/'", dstArg)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRunRemote_PullSrcNotPrefixed(t *testing.T) {
|
||||
tc := remoteCmdTest{
|
||||
srcHost: "10.5.0.144",
|
||||
srcUser: "admin",
|
||||
srcPath: "admin@10.5.0.144:/share/media/peliculas",
|
||||
dstHost: "10.5.1.10",
|
||||
dstUser: "root",
|
||||
dstPath: "/share/data",
|
||||
direction: "pull",
|
||||
}
|
||||
srcArg, _ := tc.build()
|
||||
|
||||
if countOccurrences(srcArg, "@") > 1 {
|
||||
t.Errorf("pull src %q has double SSH spec", srcArg)
|
||||
}
|
||||
if srcArg != "admin@10.5.0.144:/share/media/peliculas" {
|
||||
t.Errorf("pull src = %q, want 'admin@10.5.0.144:/share/media/peliculas'", srcArg)
|
||||
}
|
||||
}
|
||||
|
||||
func countOccurrences(s, substr string) int {
|
||||
return strings.Count(s, substr)
|
||||
}
|
||||
|
||||
func TestRun_FlagsPreservedWithPrivKey(t *testing.T) {
|
||||
runner := NewRsyncRunner("/tmp/ssh", "/tmp/ssh/id_ed25519")
|
||||
pair := &SyncPairConfig{
|
||||
Source: "/local/src",
|
||||
Dest: "admin@10.5.0.144:/remote/dst",
|
||||
Direction: "push",
|
||||
RsyncFlags: "-aP --partial",
|
||||
ExcludePatterns: []string{},
|
||||
}
|
||||
|
||||
cmd := runner.buildRsyncCmd(pair)
|
||||
|
||||
if cmd.Args[0] != "rsync" {
|
||||
t.Errorf("cmd.Args[0] = %q, want 'rsync'", cmd.Args[0])
|
||||
}
|
||||
if cmd.Args[1] != "-e" {
|
||||
t.Errorf("cmd.Args[1] = %q, want '-e' (the -e flag for ssh)", cmd.Args[1])
|
||||
}
|
||||
if !strings.Contains(cmd.Args[2], "ssh -i") {
|
||||
t.Errorf("cmd.Args[2] = %q, want ssh -i ...", cmd.Args[2])
|
||||
}
|
||||
hasAP := false
|
||||
hasPartial := false
|
||||
for i, a := range cmd.Args {
|
||||
if a == "-aP" && i > 2 {
|
||||
hasAP = true
|
||||
}
|
||||
if a == "--partial" && i > 2 {
|
||||
hasPartial = true
|
||||
}
|
||||
}
|
||||
if !hasAP {
|
||||
t.Errorf("cmd.Args = %v, want -aP flag preserved (not dropped)", cmd.Args)
|
||||
}
|
||||
if !hasPartial {
|
||||
t.Errorf("cmd.Args = %v, want --partial flag present", cmd.Args)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user