First implementation of CachingFileStorage
+ some tests to test loading and saving to cached storage
This commit is contained in:
@@ -66,6 +66,7 @@
|
||||
<Compile Include="TestKp2aApp.cs" />
|
||||
<Compile Include="TestLoadDb.cs" />
|
||||
<Compile Include="TestLoadDbCredentials.cs" />
|
||||
<Compile Include="TestCachingFileStorage.cs" />
|
||||
<Compile Include="TestSaveDb.cs" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
|
||||
@@ -19,8 +19,9 @@ namespace Kp2aUnitTests
|
||||
TestRunner runner = new TestRunner();
|
||||
// Run all tests from this assembly
|
||||
//runner.AddTests(Assembly.GetExecutingAssembly());
|
||||
//runner.AddTests(new List<Type> { typeof(TestSaveDb)});
|
||||
runner.AddTests(typeof(TestSaveDb).GetMethod("TestLoadEditSaveWithSyncKdbp"));
|
||||
runner.AddTests(new List<Type> { typeof(TestCachingFileStorage) });
|
||||
//runner.AddTests(typeof(TestCachingFileStorage).GetMethod("TestSaveToRemote"));
|
||||
//runner.AddTests(typeof(TestLoadDb).GetMethod("TestLoadKdbpWithPasswordOnly"));
|
||||
//runner.AddTests(typeof(TestSaveDb).GetMethod("TestLoadKdbxAndSaveKdbp_TestIdenticalFiles"));
|
||||
return runner;
|
||||
}
|
||||
|
||||
358
src/Kp2aUnitTests/TestCachingFileStorage.cs
Normal file
358
src/Kp2aUnitTests/TestCachingFileStorage.cs
Normal file
@@ -0,0 +1,358 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.IO;
|
||||
using System.Linq;
|
||||
using System.Text;
|
||||
using Android.App;
|
||||
using KeePassLib.Serialization;
|
||||
using KeePassLib.Utility;
|
||||
using Microsoft.VisualStudio.TestTools.UnitTesting;
|
||||
using keepass2android.Io;
|
||||
|
||||
namespace Kp2aUnitTests
|
||||
{
|
||||
[TestClass]
|
||||
class TestCachingFileStorage: TestBase
|
||||
{
|
||||
private TestFileStorage _testFileStorage;
|
||||
private CachingFileStorage _fileStorage;
|
||||
private static readonly string CachingTestFile = DefaultDirectory + "cachingTestFile.txt";
|
||||
private string _defaultCacheFileContents = "default contents";
|
||||
private TestCacheSupervisor _testCacheSupervisor;
|
||||
|
||||
class TestCacheSupervisor: ICacheSupervisor
|
||||
{
|
||||
public bool CouldntOpenFromRemoteCalled { get; set; }
|
||||
public bool CouldntSaveToRemoteCalled { get; set; }
|
||||
public bool NotifyOpenFromLocalDueToConflictCalled { get; set; }
|
||||
|
||||
|
||||
public void CouldntSaveToRemote(IOConnectionInfo ioc, Exception e)
|
||||
{
|
||||
CouldntSaveToRemoteCalled = true;
|
||||
}
|
||||
|
||||
public void CouldntOpenFromRemote(IOConnectionInfo ioc, Exception ex)
|
||||
{
|
||||
CouldntOpenFromRemoteCalled = true;
|
||||
}
|
||||
|
||||
public void NotifyOpenFromLocalDueToConflict(IOConnectionInfo ioc)
|
||||
{
|
||||
NotifyOpenFromLocalDueToConflictCalled = true;
|
||||
}
|
||||
}
|
||||
|
||||
public class TestFileStorage: IFileStorage
|
||||
{
|
||||
private BuiltInFileStorage _builtIn = new BuiltInFileStorage();
|
||||
|
||||
public bool Offline { get; set; }
|
||||
|
||||
|
||||
public void DeleteFile(IOConnectionInfo ioc)
|
||||
{
|
||||
if (Offline)
|
||||
throw new IOException("offline");
|
||||
_builtIn.DeleteFile(ioc);
|
||||
}
|
||||
|
||||
public bool CheckForFileChangeFast(IOConnectionInfo ioc, string previousFileVersion)
|
||||
{
|
||||
if (Offline)
|
||||
return false;
|
||||
return _builtIn.CheckForFileChangeFast(ioc, previousFileVersion);
|
||||
}
|
||||
|
||||
public string GetCurrentFileVersionFast(IOConnectionInfo ioc)
|
||||
{
|
||||
if (Offline)
|
||||
throw new IOException("offline");
|
||||
return _builtIn.GetCurrentFileVersionFast(ioc);
|
||||
}
|
||||
|
||||
public Stream OpenFileForRead(IOConnectionInfo ioc)
|
||||
{
|
||||
if (Offline)
|
||||
throw new IOException("offline");
|
||||
return _builtIn.OpenFileForRead(ioc);
|
||||
}
|
||||
|
||||
public IWriteTransaction OpenWriteTransaction(IOConnectionInfo ioc, bool useFileTransaction)
|
||||
{
|
||||
if (Offline)
|
||||
throw new IOException("offline");
|
||||
return new TestFileTransaction(ioc, useFileTransaction, Offline);
|
||||
}
|
||||
|
||||
public class TestFileTransaction : IWriteTransaction
|
||||
{
|
||||
private readonly bool _offline;
|
||||
private readonly FileTransactionEx _transaction;
|
||||
|
||||
public TestFileTransaction(IOConnectionInfo ioc, bool useFileTransaction, bool offline)
|
||||
{
|
||||
_offline = offline;
|
||||
_transaction = new FileTransactionEx(ioc, useFileTransaction);
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
public Stream OpenFile()
|
||||
{
|
||||
if (_offline)
|
||||
throw new IOException("offline");
|
||||
return _transaction.OpenWrite();
|
||||
}
|
||||
|
||||
public void CommitWrite()
|
||||
{
|
||||
if (_offline)
|
||||
throw new IOException("offline");
|
||||
_transaction.CommitWrite();
|
||||
}
|
||||
}
|
||||
|
||||
public bool CompleteIoId()
|
||||
{
|
||||
throw new NotImplementedException();
|
||||
}
|
||||
|
||||
public bool? FileExists()
|
||||
{
|
||||
throw new NotImplementedException();
|
||||
}
|
||||
|
||||
public string GetFilenameWithoutPathAndExt(IOConnectionInfo ioc)
|
||||
{
|
||||
return _builtIn.GetFilenameWithoutPathAndExt(ioc);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Tests correct behavior in case that either remote or cache are not available
|
||||
/// </summary>
|
||||
[TestMethod]
|
||||
public void TestMakeAccessibleWhenOffline()
|
||||
{
|
||||
SetupFileStorage();
|
||||
|
||||
//read the file once. Should now be in the cache.
|
||||
MemoryStream fileContents = ReadToMemoryStream(_fileStorage, CachingTestFile);
|
||||
|
||||
//check it's the correct data:
|
||||
Assert.AreEqual(MemoryStreamToString(fileContents), _defaultCacheFileContents);
|
||||
|
||||
//let the base file storage go offline:
|
||||
_testFileStorage.Offline = true;
|
||||
|
||||
//now try to read the file again:
|
||||
MemoryStream fileContents2 = ReadToMemoryStream(_fileStorage, CachingTestFile);
|
||||
|
||||
AssertEqual(fileContents, fileContents2);
|
||||
|
||||
Assert.IsTrue(_testCacheSupervisor.CouldntOpenFromRemoteCalled);
|
||||
Assert.IsFalse(_testCacheSupervisor.CouldntSaveToRemoteCalled);
|
||||
|
||||
|
||||
}
|
||||
|
||||
private string MemoryStreamToString(MemoryStream stream)
|
||||
{
|
||||
stream.Position = 0;
|
||||
StreamReader r = new StreamReader(stream);
|
||||
return r.ReadToEnd();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// tests correct behaviour after modifiying the local cache (with the remote file being unchanged) and remote being
|
||||
/// either unavailable or available
|
||||
/// </summary>
|
||||
[TestMethod]
|
||||
public void TestSyncOnLoadWhenLocalFileChanged()
|
||||
{
|
||||
SetupFileStorage();
|
||||
|
||||
|
||||
//read the file once. Should now be in the cache.
|
||||
ReadToMemoryStream(_fileStorage, CachingTestFile);
|
||||
|
||||
Assert.IsFalse(_testCacheSupervisor.CouldntOpenFromRemoteCalled);
|
||||
Assert.IsFalse(_testCacheSupervisor.CouldntSaveToRemoteCalled);
|
||||
|
||||
//let the base file storage go offline:
|
||||
_testFileStorage.Offline = true;
|
||||
|
||||
//write something to the cache:
|
||||
string newContent = "new content";
|
||||
WriteContentToCacheFile(newContent);
|
||||
|
||||
Assert.IsFalse(_testCacheSupervisor.CouldntOpenFromRemoteCalled);
|
||||
Assert.IsTrue(_testCacheSupervisor.CouldntSaveToRemoteCalled);
|
||||
_testCacheSupervisor.CouldntSaveToRemoteCalled = false;
|
||||
|
||||
//now try to read the file again:
|
||||
MemoryStream fileContents2 = ReadToMemoryStream(_fileStorage, CachingTestFile);
|
||||
|
||||
Assert.IsTrue(_testCacheSupervisor.CouldntOpenFromRemoteCalled);
|
||||
Assert.IsFalse(_testCacheSupervisor.CouldntSaveToRemoteCalled);
|
||||
_testCacheSupervisor.CouldntOpenFromRemoteCalled = false;
|
||||
|
||||
//should return the written content:
|
||||
Assert.AreEqual(MemoryStreamToString(fileContents2), newContent);
|
||||
|
||||
//now go online and read again. This should trigger a sync and the modified data must be returned
|
||||
_testFileStorage.Offline = false;
|
||||
MemoryStream fileContents3 = ReadToMemoryStream(_fileStorage, CachingTestFile);
|
||||
Assert.AreEqual(MemoryStreamToString(fileContents3), newContent);
|
||||
|
||||
Assert.IsFalse(_testCacheSupervisor.CouldntOpenFromRemoteCalled);
|
||||
Assert.IsFalse(_testCacheSupervisor.CouldntSaveToRemoteCalled);
|
||||
|
||||
//ensure the data on the remote was synced:
|
||||
MemoryStream fileContents4 = ReadToMemoryStream(_testFileStorage, CachingTestFile);
|
||||
Assert.AreEqual(MemoryStreamToString(fileContents4), newContent);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// tests correct behaviour after modifiying both the local cache and the remote file
|
||||
/// </summary>
|
||||
[TestMethod]
|
||||
public void TestLoadLocalWhenBothFilesChanged()
|
||||
{
|
||||
SetupFileStorage();
|
||||
|
||||
//read the file once. Should now be in the cache.
|
||||
ReadToMemoryStream(_fileStorage, CachingTestFile);
|
||||
|
||||
Assert.IsFalse(_testCacheSupervisor.CouldntOpenFromRemoteCalled);
|
||||
Assert.IsFalse(_testCacheSupervisor.CouldntSaveToRemoteCalled);
|
||||
|
||||
//let the base file storage go offline:
|
||||
_testFileStorage.Offline = true;
|
||||
|
||||
//write something to the cache:
|
||||
string newLocalContent = "new local content";
|
||||
WriteContentToCacheFile(newLocalContent);
|
||||
|
||||
Assert.IsFalse(_testCacheSupervisor.CouldntOpenFromRemoteCalled);
|
||||
Assert.IsTrue(_testCacheSupervisor.CouldntSaveToRemoteCalled);
|
||||
_testCacheSupervisor.CouldntSaveToRemoteCalled = false;
|
||||
|
||||
//write something to the remote file:
|
||||
File.WriteAllText(CachingTestFile, "new remote content");
|
||||
|
||||
//go online again:
|
||||
_testFileStorage.Offline = false;
|
||||
|
||||
//now try to read the file again:
|
||||
MemoryStream fileContents2 = ReadToMemoryStream(_fileStorage, CachingTestFile);
|
||||
|
||||
//should return the local content:
|
||||
Assert.AreEqual(MemoryStreamToString(fileContents2), newLocalContent);
|
||||
|
||||
//but a notification about the conflict should be made:
|
||||
Assert.IsFalse(_testCacheSupervisor.CouldntOpenFromRemoteCalled);
|
||||
Assert.IsFalse(_testCacheSupervisor.CouldntSaveToRemoteCalled);
|
||||
Assert.IsTrue(_testCacheSupervisor.NotifyOpenFromLocalDueToConflictCalled);
|
||||
_testCacheSupervisor.NotifyOpenFromLocalDueToConflictCalled = false;
|
||||
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
[TestMethod]
|
||||
public void TestSaveToRemote()
|
||||
{
|
||||
SetupFileStorage();
|
||||
|
||||
//read the file once. Should now be in the cache.
|
||||
ReadToMemoryStream(_fileStorage, CachingTestFile);
|
||||
|
||||
//write something to the cache:
|
||||
string newContent = "new content";
|
||||
WriteContentToCacheFile(newContent);
|
||||
|
||||
Assert.IsFalse(_testCacheSupervisor.CouldntOpenFromRemoteCalled);
|
||||
Assert.IsFalse(_testCacheSupervisor.CouldntSaveToRemoteCalled);
|
||||
|
||||
Assert.AreEqual(newContent, File.ReadAllText(CachingTestFile));
|
||||
}
|
||||
|
||||
private void WriteContentToCacheFile(string newContent)
|
||||
{
|
||||
|
||||
using (var trans = _fileStorage.OpenWriteTransaction(IocForCacheFile, true))
|
||||
{
|
||||
StreamWriter sw = new StreamWriter(trans.OpenFile());
|
||||
sw.Write(newContent);
|
||||
sw.Flush();
|
||||
sw.Close();
|
||||
trans.CommitWrite();
|
||||
}
|
||||
}
|
||||
|
||||
protected IOConnectionInfo IocForCacheFile
|
||||
{
|
||||
get { return new IOConnectionInfo() { Path = CachingTestFile }; }
|
||||
}
|
||||
|
||||
private void SetupFileStorage()
|
||||
{
|
||||
_testFileStorage = new TestFileStorage();
|
||||
_testCacheSupervisor = new TestCacheSupervisor();
|
||||
_fileStorage = new CachingFileStorage(_testFileStorage, Application.Context.CacheDir.Path, _testCacheSupervisor);
|
||||
_fileStorage.ClearCache();
|
||||
File.WriteAllText(CachingTestFile, _defaultCacheFileContents);
|
||||
}
|
||||
|
||||
private static MemoryStream ReadToMemoryStream(IFileStorage fileStorage, string filename)
|
||||
{
|
||||
Stream fileStream = fileStorage.OpenFileForRead(new IOConnectionInfo() {Path = filename});
|
||||
MemoryStream fileContents = new MemoryStream();
|
||||
fileStream.CopyTo(fileContents);
|
||||
fileStream.Close();
|
||||
return fileContents;
|
||||
}
|
||||
|
||||
|
||||
static bool StreamEquals(Stream stream1, Stream stream2)
|
||||
{
|
||||
const int bufferSize = 2048;
|
||||
byte[] buffer1 = new byte[bufferSize]; //buffer size
|
||||
byte[] buffer2 = new byte[bufferSize];
|
||||
while (true)
|
||||
{
|
||||
int count1 = stream1.Read(buffer1, 0, bufferSize);
|
||||
int count2 = stream2.Read(buffer2, 0, bufferSize);
|
||||
|
||||
if (count1 != count2)
|
||||
return false;
|
||||
|
||||
if (count1 == 0)
|
||||
return true;
|
||||
|
||||
// You might replace the following with an efficient "memcmp"
|
||||
if (!buffer1.Take(count1).SequenceEqual(buffer2.Take(count2)))
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
private void AssertEqual(MemoryStream s1, MemoryStream s2)
|
||||
{
|
||||
s1.Seek(0,0);
|
||||
s2.Seek(0, 0);
|
||||
Assert.AreEqual(s1.Length, s2.Length);
|
||||
Assert.AreEqual(0, s1.Position);
|
||||
Assert.AreEqual(0, s2.Position);
|
||||
Assert.IsTrue(StreamEquals(s1, s2));
|
||||
}
|
||||
|
||||
//todo test delete
|
||||
}
|
||||
}
|
||||
@@ -232,26 +232,18 @@ namespace Kp2aUnitTests
|
||||
}
|
||||
|
||||
|
||||
|
||||
/*
|
||||
[TestMethod]
|
||||
public void TestSaveAsWhenReadOnly()
|
||||
{
|
||||
Assert.Fail("TODO: Test ");
|
||||
}
|
||||
|
||||
|
||||
[TestMethod]
|
||||
public void TestReloadWhenCancelSync()
|
||||
{
|
||||
//when a change is detected and the user cancels saving, the app should display the "file was modified - reload?" question.
|
||||
Assert.Fail("TODO: Test ");
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void TestSaveAsWhenSyncError()
|
||||
{
|
||||
Assert.Fail("TODO: Test ");
|
||||
}
|
||||
}*/
|
||||
|
||||
[TestMethod]
|
||||
public void TestLoadAndSave_TestIdenticalFiles()
|
||||
|
||||
Reference in New Issue
Block a user