コード例 #1
0
        public void Execute_IfFileDoesntExistOnServerButSegmentExists_ShouldNotSendSegment()
        {
            // Arrange
            const string filePath = "C:\\MyFolder\\1.txt";

            var fileSystem    = MockFileSystem(filePath);
            var configuration = new Mock <IConfiguration>();

            configuration.Setup(conf => conf["segment-size"]).Returns("10");
            var hashAlg  = MD5.Create();
            var hashSize = hashAlg.HashSize / 8;

            // Act
            var strategy = new ChangeTransferStrategy(configuration.Object, fileSystem.Object, hashAlg);
            var buffer   = new byte[45];

            buffer[27] = 1; // Answer from server that we need this file
            var memoryStream = new MemoryStream(buffer);

            strategy.Execute(GetFileModel(filePath), memoryStream);

            // Assert
            memoryStream.Seek(0, SeekOrigin.Begin);
            var reader = new BinaryReader(memoryStream);

            memoryStream.Seek(InitialDataSize + hashSize + 1, SeekOrigin.Begin);  // Skip initiation data + filehash + server response
            var hash = reader.ReadBytes(hashSize);

            Assert.Equal(16, hash.Length);
            var otherValues = reader.ReadBytes(10);

            Assert.Equal(0, otherValues?.Length);
        }
コード例 #2
0
        public void Execute_IfFileExistsOnServer_ShouldNotSendFile()
        {
            // Arrange
            const string filePath = "C:\\MyFolder\\1.txt";

            var fileSystem    = MockFileSystem(filePath);
            var configuration = new Mock <IConfiguration>();
            var hashAlg       = MD5.Create();

            // Act
            var strategy     = new ChangeTransferStrategy(configuration.Object, fileSystem.Object, hashAlg);
            var memoryStream = new MemoryStream();

            strategy.Execute(GetFileModel(filePath), memoryStream);

            // Assert
            memoryStream.Seek(0, SeekOrigin.Begin);
            var reader = new BinaryReader(memoryStream);

            memoryStream.Seek(InitialDataSize, SeekOrigin.Begin); // Skip initiation data

            var hash = reader.ReadBytes(hashAlg.HashSize);

            Assert.Equal(16, hash.Length);

            var otherValues = reader.ReadBytes(10);

            Assert.Equal(0, otherValues?.Length);
        }
コード例 #3
0
        public void Execute_WithNullStream_ShouldShrowException()
        {
            // Arrange
            var fileSystem    = new Mock <IFileSystem>();
            var configuration = new Mock <IConfiguration>();
            var hash          = new Mock <HashAlgorithm>();

            // Act
            var strategy = new ChangeTransferStrategy(configuration.Object, fileSystem.Object, hash.Object);

            // Assert
            Assert.Throws <ArgumentNullException>(() => strategy.Execute(new RenameFileModel(), null));
        }
コード例 #4
0
        public void Execute_ShouldInitiateDialogWithCorrectValues()
        {
            // Arrange
            const string filePath = "C:\\MyFolder\\1.txt";

            var file = new Mock <FileBase>();

            file.Setup(f => f.OpenRead(filePath)).Returns(Stream.Null);
            var fileSystem = new Mock <IFileSystem>();

            fileSystem.Setup(fs => fs.File).Returns(file.Object);

            var configuration = new Mock <IConfiguration>();
            var hash          = new Mock <HashAlgorithm>();

            // Act
            var strategy     = new ChangeTransferStrategy(configuration.Object, fileSystem.Object, hash.Object);
            var memoryStream = new MemoryStream();

            strategy.Execute(GetFileModel(filePath), memoryStream);

            // Assert
            memoryStream.Seek(0, SeekOrigin.Begin);
            var reader = new BinaryReader(memoryStream);

            var version = reader.ReadBytes(1).Single();

            Assert.Equal(1, version);

            var action = (FileAction)reader.ReadBytes(1).Single();

            Assert.Equal(FileAction.Change, action);

            var type = (FileType)reader.ReadBytes(1).Single();

            Assert.Equal(FileType.Unknown, type);


            var size = BitConverter.ToInt32(reader.ReadBytes(4), 0);

            Assert.Equal(5, size);

            var filename = Encoding.GetEncoding("UTF-8").GetString(reader.ReadBytes(size));

            Assert.Equal("1.txt", filename);
        }
コード例 #5
0
        public void Execute_IfFileDoesntExist_ShouldSendAllSegments()
        {
            // Arrange
            const string filePath = "C:\\MyFolder\\1.txt";

            var fileSystem    = MockFileSystem(filePath);
            var configuration = new Mock <IConfiguration>();

            configuration.Setup(conf => conf["segment-size"]).Returns("5");
            var hashAlg  = MD5.Create();
            var hashSize = hashAlg.HashSize / 8;

            // Act
            var strategy = new ChangeTransferStrategy(configuration.Object, fileSystem.Object, hashAlg);
            var buffer   = new byte[73];

            buffer[28] = 1; // Answer from server that we need this file
            buffer[45] = 1; // Answer from server that we need first segment
            buffer[67] = 1; // Answer from server that we need second segment
            var memoryStream = new MemoryStream(buffer);

            strategy.Execute(GetFileModel(filePath), memoryStream);

            // Assert
            memoryStream.Seek(0, SeekOrigin.Begin);
            var reader = new BinaryReader(memoryStream);

            // Skip initiation data + filehash + server response + segment hash + server resposne
            memoryStream.Seek(InitialDataSize + hashSize + 1 + hashSize + 1, SeekOrigin.Begin);

            var firstSegment = reader.ReadBytes(5);

            Assert.Equal(5, firstSegment?.Length);
            Assert.Equal(new byte[] { 1, 2, 3, 4, 5 }, firstSegment);

            memoryStream.ReadByte();    // Skip server response
            reader.ReadBytes(hashSize); // Skip hash

            var secondSegment = reader.ReadBytes(5);

            Assert.Equal(5, secondSegment?.Length);
            Assert.Equal(new byte[] { 6, 7, 8, 9, 0 }, secondSegment);
        }