mirror of
https://github.com/PowerShell/openssh-portable.git
synced 2025-07-31 01:35:11 +02:00
SFTP pester testcases (#66)
PowerShell/Win32-OpenSSH#499 PowerShell/Win32-OpenSSH#501
This commit is contained in:
bagajjal
Manoj Ampalam
parent
073eadfc98
commit
4132f8cebd
267
regress/pesterTests/SFTP.Tests.ps1
Normal file
267
regress/pesterTests/SFTP.Tests.ps1
Normal file
@ -0,0 +1,267 @@
|
|||||||
|
using module .\PlatformAbstractLayer.psm1
|
||||||
|
|
||||||
|
Describe "SFTP Testcases" -Tags "CI" {
|
||||||
|
BeforeAll {
|
||||||
|
$rootDirectory = $TestDrive
|
||||||
|
|
||||||
|
$outputFileName = "output.txt"
|
||||||
|
$batchFileName = "sftp-batchcmds.txt"
|
||||||
|
$outputFilePath = Join-Path $rootDirectory $outputFileName
|
||||||
|
$batchFilePath = Join-Path $rootDirectory $batchFileName
|
||||||
|
|
||||||
|
$tempFileName = "tempFile.txt"
|
||||||
|
$tempFilePath = Join-Path $rootDirectory $tempFileName
|
||||||
|
|
||||||
|
$tempUnicodeFileName = "tempFile_язык.txt"
|
||||||
|
$tempUnicodeFilePath = Join-Path $rootDirectory $tempUnicodeFileName
|
||||||
|
|
||||||
|
$clientDirectory = Join-Path $rootDirectory 'client_dir'
|
||||||
|
$serverDirectory = Join-Path $rootDirectory 'server_dir'
|
||||||
|
|
||||||
|
$null = New-Item $clientDirectory -ItemType directory -Force
|
||||||
|
$null = New-Item $serverDirectory -ItemType directory -Force
|
||||||
|
$null = New-Item $batchFilePath -ItemType file -Force
|
||||||
|
$null = New-Item $outputFilePath -ItemType file -Force
|
||||||
|
$null = New-Item $tempFilePath -ItemType file -Force -value "temp file data"
|
||||||
|
$null = New-Item $tempUnicodeFilePath -ItemType file -Force -value "temp file data"
|
||||||
|
|
||||||
|
$expectedOutputDelimiter = "#DL$"
|
||||||
|
|
||||||
|
[Machine] $client = [Machine]::new([MachineRole]::Client)
|
||||||
|
[Machine] $server = [Machine]::new([MachineRole]::Server)
|
||||||
|
$client.SetupClient($server)
|
||||||
|
$server.SetupServer($client)
|
||||||
|
|
||||||
|
$testData1 = @(
|
||||||
|
@{
|
||||||
|
title = "put, ls for non-unicode file names"
|
||||||
|
logonstr = "$($server.localadminusername)@$($server.machinename)"
|
||||||
|
options = '-i $identifyfile'
|
||||||
|
commands = "put $tempFilePath $serverDirectory
|
||||||
|
ls $serverDirectory"
|
||||||
|
expectedoutput = (join-path $serverdirectory $tempFileName).replace("\", "/")
|
||||||
|
},
|
||||||
|
@{
|
||||||
|
title = "get, ls for non-unicode file names"
|
||||||
|
logonstr = "$($server.localadminusername)@$($server.machinename)"
|
||||||
|
options = '-i $identifyfile'
|
||||||
|
commands = "get $tempFilePath $clientDirectory
|
||||||
|
ls $clientDirectory"
|
||||||
|
expectedoutput = (join-path $clientDirectory $tempFileName).replace("\", "/")
|
||||||
|
},
|
||||||
|
@{
|
||||||
|
title = "mput, ls for non-unicode file names"
|
||||||
|
logonstr = "$($server.localadminusername)@$($server.machinename)"
|
||||||
|
options = '-i $identifyfile'
|
||||||
|
commands = "mput $tempFilePath $serverDirectory
|
||||||
|
ls $serverDirectory"
|
||||||
|
expectedoutput = (join-path $serverdirectory $tempFileName).replace("\", "/")
|
||||||
|
},
|
||||||
|
@{
|
||||||
|
title = "mget, ls for non-unicode file names"
|
||||||
|
logonstr = "$($server.localadminusername)@$($server.machinename)"
|
||||||
|
options = '-i $identifyfile'
|
||||||
|
commands = "mget $tempFilePath $clientDirectory
|
||||||
|
ls $clientDirectory"
|
||||||
|
expectedoutput = (join-path $clientDirectory $tempFileName).replace("\", "/")
|
||||||
|
},
|
||||||
|
@{
|
||||||
|
title = "mkdir, cd, pwd for non-unicode directory names"
|
||||||
|
logonstr = "$($server.localadminusername)@$($server.machinename)"
|
||||||
|
options = '-i $identifyfile'
|
||||||
|
commands = "cd $serverdirectory
|
||||||
|
mkdir server_test_dir
|
||||||
|
cd server_test_dir
|
||||||
|
pwd"
|
||||||
|
expectedoutput = (join-path $serverdirectory "server_test_dir").replace("\", "/")
|
||||||
|
},
|
||||||
|
@{
|
||||||
|
Title = "lmkdir, lcd, lpwd for non-unicode directory names"
|
||||||
|
LogonStr = "$($server.localAdminUserName)@$($server.MachineName)"
|
||||||
|
Options = '-i $identifyFile'
|
||||||
|
Commands = "lcd $clientDirectory
|
||||||
|
lmkdir client_test_dir
|
||||||
|
lcd client_test_dir
|
||||||
|
lpwd"
|
||||||
|
ExpectedOutput = (Join-Path $clientDirectory "client_test_dir")
|
||||||
|
},
|
||||||
|
@{
|
||||||
|
title = "put, ls for unicode file names"
|
||||||
|
logonstr = "$($server.localadminusername)@$($server.machinename)"
|
||||||
|
options = '-i $identifyfile'
|
||||||
|
commands = "put $tempUnicodeFilePath $serverDirectory
|
||||||
|
ls $serverDirectory"
|
||||||
|
expectedoutput = (join-path $serverdirectory $tempUnicodeFileName).replace("\", "/")
|
||||||
|
},
|
||||||
|
@{
|
||||||
|
title = "get, ls for unicode file names"
|
||||||
|
logonstr = "$($server.localadminusername)@$($server.machinename)"
|
||||||
|
options = '-i $identifyfile'
|
||||||
|
commands = "get $tempUnicodeFilePath $clientDirectory
|
||||||
|
ls $clientDirectory"
|
||||||
|
expectedoutput = (join-path $clientDirectory $tempUnicodeFileName).replace("\", "/")
|
||||||
|
},
|
||||||
|
@{
|
||||||
|
title = "mput, ls for unicode file names"
|
||||||
|
logonstr = "$($server.localadminusername)@$($server.machinename)"
|
||||||
|
options = '-i $identifyfile'
|
||||||
|
commands = "mput $tempUnicodeFilePath $serverDirectory
|
||||||
|
ls $serverDirectory"
|
||||||
|
expectedoutput = (join-path $serverdirectory $tempUnicodeFileName).replace("\", "/")
|
||||||
|
},
|
||||||
|
@{
|
||||||
|
title = "mget, ls for unicode file names"
|
||||||
|
logonstr = "$($server.localadminusername)@$($server.machinename)"
|
||||||
|
options = '-i $identifyfile'
|
||||||
|
commands = "mget $tempUnicodeFilePath $clientDirectory
|
||||||
|
ls $clientDirectory"
|
||||||
|
expectedoutput = (join-path $clientDirectory $tempUnicodeFileName).replace("\", "/")
|
||||||
|
},
|
||||||
|
@{
|
||||||
|
title = "mkdir, cd, pwd for unicode directory names"
|
||||||
|
logonstr = "$($server.localadminusername)@$($server.machinename)"
|
||||||
|
options = '-i $identifyfile'
|
||||||
|
commands = "cd $serverdirectory
|
||||||
|
mkdir server_test_dir_язык
|
||||||
|
cd server_test_dir_язык
|
||||||
|
pwd"
|
||||||
|
expectedoutput = (join-path $serverdirectory "server_test_dir_язык").replace("\", "/")
|
||||||
|
},
|
||||||
|
@{
|
||||||
|
Title = "lmkdir, lcd, lpwd for unicode directory names"
|
||||||
|
LogonStr = "$($server.localAdminUserName)@$($server.MachineName)"
|
||||||
|
Options = '-i $identifyFile'
|
||||||
|
Commands = "lcd $clientDirectory
|
||||||
|
lmkdir client_test_dir_язык
|
||||||
|
lcd client_test_dir_язык
|
||||||
|
lpwd
|
||||||
|
lls $clientDirectory"
|
||||||
|
ExpectedOutput = (Join-Path $clientDirectory "client_test_dir_язык")
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
$testData2 = @(
|
||||||
|
@{
|
||||||
|
title = "rm, rmdir, rename for unicode file, directory"
|
||||||
|
logonstr = "$($server.localadminusername)@$($server.machinename)"
|
||||||
|
options = '-i $identifyfile -b $batchFilePath'
|
||||||
|
|
||||||
|
tmpFileName1 = $tempUnicodeFileName
|
||||||
|
tmpFilePath1 = $tempUnicodeFilePath
|
||||||
|
tmpFileName2 = "tempfile_язык_2.txt"
|
||||||
|
tmpFilePath2 = (join-path $serverDirectory "tempfile_язык_2.txt")
|
||||||
|
|
||||||
|
tmpDirectoryName1 = "test_dir_язык_1"
|
||||||
|
tmpDirectoryPath1 = (join-path $serverDirectory "test_dir_язык_1")
|
||||||
|
tmpDirectoryName2 = "test_dir_язык_2"
|
||||||
|
tmpDirectoryPath2 = (join-path $serverDirectory "test_dir_язык_2")
|
||||||
|
},
|
||||||
|
@{
|
||||||
|
title = "rm, rmdir, rename for non-unicode file, directory"
|
||||||
|
logonstr = "$($server.localadminusername)@$($server.machinename)"
|
||||||
|
options = '-i $identifyfile -b $batchFilePath'
|
||||||
|
|
||||||
|
tmpFileName1 = $tempFileName
|
||||||
|
tmpFilePath1 = $tempFilePath
|
||||||
|
tmpFileName2 = "tempfile_2.txt"
|
||||||
|
tmpFilePath2 = (join-path $serverDirectory "tempfile_2.txt")
|
||||||
|
|
||||||
|
tmpDirectoryName1 = "test_dir_1"
|
||||||
|
tmpDirectoryPath1 = (join-path $serverDirectory "test_dir_1")
|
||||||
|
tmpDirectoryName2 = "test_dir_2"
|
||||||
|
tmpDirectoryPath2 = (join-path $serverDirectory "test_dir_2")
|
||||||
|
}
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
AfterAll {
|
||||||
|
$client.CleanupClient()
|
||||||
|
$server.CleanupServer()
|
||||||
|
}
|
||||||
|
|
||||||
|
Context "Single signon" {
|
||||||
|
BeforeAll {
|
||||||
|
$Server.SecureHostKeys($server.PrivateHostKeyPaths)
|
||||||
|
$identifyFile = $client.clientPrivateKeyPaths[0]
|
||||||
|
.\ssh-add.exe $identifyFile #setup single signon
|
||||||
|
}
|
||||||
|
AfterAll {
|
||||||
|
$Server.CleanupHostKeys()
|
||||||
|
.\ssh-add.exe -D #cleanup single signon
|
||||||
|
|
||||||
|
Get-Item $rootDirectory | Remove-Item -Recurse -Force -ErrorAction SilentlyContinue
|
||||||
|
}
|
||||||
|
|
||||||
|
BeforeEach {
|
||||||
|
Get-ChildItem $serverDirectory | Remove-Item -Recurse -Force -ErrorAction SilentlyContinue
|
||||||
|
Get-ChildItem $clientDirectory | Remove-Item -Recurse -Force -ErrorAction SilentlyContinue
|
||||||
|
Remove-Item $batchFilePath
|
||||||
|
Remove-Item $outputFilePath
|
||||||
|
}
|
||||||
|
|
||||||
|
It '<Title>' -TestCases:$testData1 {
|
||||||
|
param([string]$Title, $LogonStr, $Options, $Commands, $ExpectedOutput, $SkipVerification = $false)
|
||||||
|
|
||||||
|
Set-Content $batchFilePath -Encoding UTF8 -value $Commands
|
||||||
|
$str = $ExecutionContext.InvokeCommand.ExpandString(".\sftp $($Options) -b $batchFilePath $($LogonStr) > $outputFilePath")
|
||||||
|
$client.RunCmd($str)
|
||||||
|
|
||||||
|
#validate file content.
|
||||||
|
$($ExpectedOutput).split($expectedOutputDelimiter) | foreach {
|
||||||
|
$outputFilePath | Should Contain ([RegEx]::Escape($_))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
It '<Title>' -TestCases:$testData2 {
|
||||||
|
param([string]$Title, $LogonStr, $Options, $tmpFileName1, $tmpFilePath1, $tmpFileName2, $tmpFilePath2, $tmpDirectoryName1, $tmpDirectoryPath1, $tmpDirectoryName2, $tmpDirectoryPath2, $SkipVerification = $false)
|
||||||
|
|
||||||
|
#rm (remove file)
|
||||||
|
$commands = "mkdir $tmpDirectoryPath1
|
||||||
|
put $tmpFilePath1 $tmpDirectoryPath1
|
||||||
|
ls $tmpDirectoryPath1"
|
||||||
|
Set-Content $batchFilePath -Encoding UTF8 -value $commands
|
||||||
|
$str = $ExecutionContext.InvokeCommand.ExpandString(".\sftp $($Options) $($LogonStr) > $outputFilePath")
|
||||||
|
$client.RunCmd($str)
|
||||||
|
$outputFilePath | Should Contain ([RegEx]::Escape((join-path $tmpDirectoryPath1 $tmpFileName1).replace("\", "/")))
|
||||||
|
|
||||||
|
$commands = "rm $tmpDirectoryPath1\*
|
||||||
|
ls $tmpDirectoryPath1
|
||||||
|
pwd
|
||||||
|
"
|
||||||
|
Set-Content $batchFilePath -Encoding UTF8 -value $commands
|
||||||
|
$str = $ExecutionContext.InvokeCommand.ExpandString(".\sftp $($Options) $($LogonStr) > $outputFilePath")
|
||||||
|
$client.RunCmd($str)
|
||||||
|
$outputFilePath | Should Not Contain ([RegEx]::Escape((join-path $tmpDirectoryPath1 $tmpFileName1).replace("\", "/")))
|
||||||
|
|
||||||
|
#rename file
|
||||||
|
Remove-Item $outputFilePath
|
||||||
|
Copy-Item $tmpFilePath1 -destination $tmpDirectoryPath1
|
||||||
|
$commands = "rename $tmpDirectoryPath1\$tmpFileName1 $tmpDirectoryPath1\$tmpFileName2
|
||||||
|
ls $tmpDirectoryPath1
|
||||||
|
pwd"
|
||||||
|
Set-Content $batchFilePath -Encoding UTF8 -value $commands
|
||||||
|
$str = $ExecutionContext.InvokeCommand.ExpandString(".\sftp $($Options) $($LogonStr) > $outputFilePath")
|
||||||
|
$client.RunCmd($str)
|
||||||
|
$outputFilePath | Should Contain ([RegEx]::Escape((join-path $tmpDirectoryPath1 $tmpFileName2).replace("\", "/")))
|
||||||
|
|
||||||
|
#rename directory
|
||||||
|
Remove-Item $outputFilePath
|
||||||
|
$commands = "rm $tmpDirectoryPath1\*
|
||||||
|
rename $tmpDirectoryPath1 $tmpDirectoryPath2
|
||||||
|
ls $serverDirectory"
|
||||||
|
Set-Content $batchFilePath -Encoding UTF8 -value $commands
|
||||||
|
$str = $ExecutionContext.InvokeCommand.ExpandString(".\sftp $($Options) $($LogonStr) > $outputFilePath")
|
||||||
|
$client.RunCmd($str)
|
||||||
|
$outputFilePath | Should Contain ([RegEx]::Escape($tmpDirectoryPath2.replace("\", "/")))
|
||||||
|
|
||||||
|
#rmdir (remove directory)
|
||||||
|
Remove-Item $outputFilePath
|
||||||
|
$commands = "rmdir $tmpDirectoryPath2
|
||||||
|
ls $serverDirectory"
|
||||||
|
Set-Content $batchFilePath -Encoding UTF8 -value $commands
|
||||||
|
$str = $ExecutionContext.InvokeCommand.ExpandString(".\sftp $($Options) $($LogonStr) > $outputFilePath")
|
||||||
|
$client.RunCmd($str)
|
||||||
|
$outputFilePath | Should Not Contain ([RegEx]::Escape($tmpDirectoryPath2).replace("\", "/"))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
11
sftp.c
11
sftp.c
@ -2208,17 +2208,22 @@ interactive_loop(struct sftp_conn *conn, char *file1, char *file2)
|
|||||||
free(dir);
|
free(dir);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
interactive = !batchmode && isatty(STDIN_FILENO);
|
||||||
|
err = 0;
|
||||||
|
|
||||||
#ifdef WINDOWS
|
#ifdef WINDOWS
|
||||||
/* Min buffer size allowed in Windows is 2*/
|
/* Min buffer size allowed in Windows is 2*/
|
||||||
setvbuf(stdout, NULL, _IOLBF, 2);
|
setvbuf(stdout, NULL, _IOLBF, 2);
|
||||||
setvbuf(infile, NULL, _IOLBF, 2);
|
|
||||||
|
/* We do this only in interactive mode as we are unable to read files with UTF8 BOM */
|
||||||
|
if(interactive)
|
||||||
|
setvbuf(infile, NULL, _IOLBF, 2);
|
||||||
#else /* !WINDOWS */
|
#else /* !WINDOWS */
|
||||||
setvbuf(stdout, NULL, _IOLBF, 0);
|
setvbuf(stdout, NULL, _IOLBF, 0);
|
||||||
setvbuf(infile, NULL, _IOLBF, 0);
|
setvbuf(infile, NULL, _IOLBF, 0);
|
||||||
#endif /* !WINDOWS */
|
#endif /* !WINDOWS */
|
||||||
|
|
||||||
interactive = !batchmode && isatty(STDIN_FILENO);
|
|
||||||
err = 0;
|
|
||||||
for (;;) {
|
for (;;) {
|
||||||
char *cp;
|
char *cp;
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user