SFTP pester testcases (#66)
PowerShell/Win32-OpenSSH#499 PowerShell/Win32-OpenSSH#501
This commit is contained in:
bagajjal
Manoj Ampalam
parent
073eadfc98
commit
4132f8cebd
|
|
@ -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);
|
||||
}
|
||||
|
||||
interactive = !batchmode && isatty(STDIN_FILENO);
|
||||
err = 0;
|
||||
|
||||
#ifdef WINDOWS
|
||||
/* Min buffer size allowed in Windows is 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 */
|
||||
setvbuf(stdout, NULL, _IOLBF, 0);
|
||||
setvbuf(infile, NULL, _IOLBF, 0);
|
||||
#endif /* !WINDOWS */
|
||||
|
||||
interactive = !batchmode && isatty(STDIN_FILENO);
|
||||
err = 0;
|
||||
|
||||
for (;;) {
|
||||
char *cp;
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue