Learning/mRemoteNG
1
0
Fork 0
mirror of https://github.com/mRemoteNG/mRemoteNG.git synced 2026-02-17 22:11:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

updated some tests

Browse Source
This commit is contained in:
David Sparer
2017-04-14 16:34:28 -06:00
parent a2e0a0c6f7
commit 47a02afea6
5 changed files with 33 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
mRemoteNG.Specs/StepDefinitions/CredentialRepositorySteps.cs
View File

@@ -1,5 +1,6 @@
using System.Security;
using mRemoteNG.Credential;
using mRemoteNG.Security;
using mRemoteNG.Specs.Utilities;
using NUnit.Framework;
using TechTalk.SpecFlow;
@@ -10,15 +11,12 @@ namespace mRemoteNG.Specs.StepDefinitions
public class CredentialRepositorySteps
{
private ICredentialRepository _credentialRepository;
private readonly SecureString _key = "somePassword".ConvertToSecureString();
[Given(@"I have a credential repository")]
public void GivenIHaveACredentialRepository()
{
var utilityFactory = new XmlCredentialRepoBuilder();
var credentialXml = "<?xml version=\"1.0\" encoding=\"utf-8\"?>" +
"<Credentials EncryptionEngine=\"AES\" BlockCipherMode=\"GCM\" KdfIterations=\"1000\" SchemaVersion=\"1.0\">" +
"</Credentials>";
utilityFactory.XmlFileContent = credentialXml;
var utilityFactory = new XmlCredentialRepoBuilder {EncryptionKey = _key};
_credentialRepository = utilityFactory.BuildXmlCredentialRepo();
}
@@ -33,7 +31,7 @@ namespace mRemoteNG.Specs.StepDefinitions
[Given(@"The credential repository is loaded")]
public void GivenTheCredentialRepositoryIsLoaded()
{
_credentialRepository.LoadCredentials(new SecureString());
_credentialRepository.LoadCredentials(_key);
Assert.That(_credentialRepository.IsLoaded);
}
@@ -46,7 +44,7 @@ namespace mRemoteNG.Specs.StepDefinitions
[When(@"I click load")]
public void WhenIClickLoad()
{
_credentialRepository.LoadCredentials(new SecureString());
_credentialRepository.LoadCredentials(_key);
}
[Then(@"the credential repository is loaded")]

12
mRemoteNG.Specs/Utilities/CredRepoXmlFileBuilder.cs Normal file
View File

@@ -0,0 +1,12 @@
namespace mRemoteNG.Specs.Utilities
{
public class CredRepoXmlFileBuilder
{
public string Build(string authHeader)
{
return "<?xml version=\"1.0\" encoding=\"utf-8\"?>" +
$"<Credentials EncryptionEngine=\"AES\" BlockCipherMode=\"GCM\" KdfIterations=\"1000\" Auth=\"{authHeader}\" SchemaVersion=\"1.0\">" +
"</Credentials>";
}
}
}

8
mRemoteNG.Specs/Utilities/XmlCredentialRepoBuilder.cs
View File

@@ -1,5 +1,4 @@
using System.Collections.Generic;
using System.Security;
using System.Security;
using mRemoteNG.Config;
using mRemoteNG.Config.DataProviders;
using mRemoteNG.Config.Serializers.CredentialSerializer;
@@ -14,11 +13,12 @@ namespace mRemoteNG.Specs.Utilities
{
public SecureString EncryptionKey { get; set; } = "someKey1".ConvertToSecureString();
public ICryptographyProvider CryptographyProvider { get; set; } = new AeadCryptographyProvider();
public string XmlFileContent { get; set; } = "";
public ICredentialRepository BuildXmlCredentialRepo()
{
var dataProvider = new InMemoryStringDataProvider(XmlFileContent);
var xmlFileBuilder = new CredRepoXmlFileBuilder();
var xmlFileContent = xmlFileBuilder.Build(CryptographyProvider.Encrypt("someheaderdata", EncryptionKey));
var dataProvider = new InMemoryStringDataProvider(xmlFileContent);
var encryptor = new XmlCredentialPasswordEncryptorDecorator(
CryptographyProvider,
new XmlCredentialRecordSerializer()

1
mRemoteNG.Specs/mRemoteNG.Specs.csproj
View File

@@ -61,6 +61,7 @@
<Compile Include="Properties\AssemblyInfo.cs" />
<Compile Include="StepDefinitions\CredentialRepositoryListSteps.cs" />
<Compile Include="StepDefinitions\CredentialRepositorySteps.cs" />
<Compile Include="Utilities\CredRepoXmlFileBuilder.cs" />
<Compile Include="Utilities\XmlCredentialRepoBuilder.cs" />
</ItemGroup>
<ItemGroup>

14
mRemoteNGTests/Config/Serializers/CredentialSerializers/XmlCredentialPasswordDecryptorDecoratorTests.cs
View File

@@ -23,19 +23,27 @@ namespace mRemoteNGTests.Config.Serializers.CredentialSerializers
[Test]
public void OutputedCredentialHasDecryptedPassword()
{
var xml = GenerateXml();
var xml = GenerateCredentialXml();
var output = _sut.Deserialize(xml, _decryptionKey);
Assert.That(output.First().Password.ConvertToUnsecureString(), Is.EqualTo(_unencryptedPassword));
}
[Test]
public void DecryptionThrowsExceptionWhenAuthHeaderNotFound()
{
var xml = GenerateCredentialXml(false);
Assert.Throws<EncryptionException>(() => _sut.Deserialize(xml, _decryptionKey));
}
private string GenerateXml()
private string GenerateCredentialXml(bool includeAuthHeader = true)
{
var cryptoProvider = new AeadCryptographyProvider();
var authHeader = includeAuthHeader ? $"Auth=\"{cryptoProvider.Encrypt("someheader", _decryptionKey)}\"" : "";
var encryptedPassword = cryptoProvider.Encrypt(_unencryptedPassword, _decryptionKey);
return
"<?xml version=\"1.0\" encoding=\"utf-8\"?>" +
$"<Credentials EncryptionEngine=\"{cryptoProvider.CipherEngine}\" BlockCipherMode=\"{cryptoProvider.CipherMode}\" KdfIterations=\"{cryptoProvider.KeyDerivationIterations}\" SchemaVersion=\"1.0\">" +
$"<Credentials EncryptionEngine=\"{cryptoProvider.CipherEngine}\" BlockCipherMode=\"{cryptoProvider.CipherMode}\" KdfIterations=\"{cryptoProvider.KeyDerivationIterations}\" {authHeader} SchemaVersion=\"1.0\">" +
$"<Credential Id=\"ce6b0397-d476-4ffe-884b-dbe9347a88a8\" Title=\"New Credential\" Username=\"asdfasdf\" Domain=\"\" Password=\"{encryptedPassword}\" />" +
"</Credentials>";
}