SftpFileStreamTest_Close_SessionOpen.cs 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. using System;
  2. using System.IO;
  3. using Microsoft.VisualStudio.TestTools.UnitTesting;
  4. using Moq;
  5. using Renci.SshNet.Sftp;
  6. namespace Renci.SshNet.Tests.Classes.Sftp
  7. {
  8. [TestClass]
  9. public class SftpFileStreamTest_Close_SessionOpen : SftpFileStreamTestBase
  10. {
  11. private SftpFileStream _target;
  12. private string _path;
  13. private byte[] _handle;
  14. private uint _bufferSize;
  15. private uint _readBufferSize;
  16. private uint _writeBufferSize;
  17. protected override void SetupData()
  18. {
  19. base.SetupData();
  20. var random = new Random();
  21. _path = random.Next().ToString();
  22. _handle = GenerateRandom(2, random);
  23. _bufferSize = (uint) random.Next(1, 1000);
  24. _readBufferSize = (uint) random.Next(1, 1000);
  25. _writeBufferSize = (uint) random.Next(1, 1000);
  26. }
  27. protected override void SetupMocks()
  28. {
  29. _ = SftpSessionMock.InSequence(MockSequence)
  30. .Setup(p => p.RequestOpen(_path, Flags.Read, false))
  31. .Returns(_handle);
  32. _ = SftpSessionMock.InSequence(MockSequence)
  33. .Setup(p => p.CalculateOptimalReadLength(_bufferSize))
  34. .Returns(_readBufferSize);
  35. _ = SftpSessionMock.InSequence(MockSequence)
  36. .Setup(p => p.CalculateOptimalWriteLength(_bufferSize, _handle))
  37. .Returns(_writeBufferSize);
  38. _ = SftpSessionMock.InSequence(MockSequence)
  39. .Setup(p => p.IsOpen)
  40. .Returns(true);
  41. _ = SftpSessionMock.InSequence(MockSequence)
  42. .Setup(p => p.RequestClose(_handle));
  43. }
  44. protected override void Arrange()
  45. {
  46. base.Arrange();
  47. _target = new SftpFileStream(SftpSessionMock.Object,
  48. _path,
  49. FileMode.Open,
  50. FileAccess.Read,
  51. (int) _bufferSize);
  52. }
  53. protected override void Act()
  54. {
  55. _target.Close();
  56. }
  57. [TestMethod]
  58. public void IsOpenOnSftpSessionShouldBeInvokedOnce()
  59. {
  60. SftpSessionMock.Verify(p => p.IsOpen, Times.Once);
  61. }
  62. [TestMethod]
  63. public void RequestCloseOnSftpSessionShouldBeInvokedOnce()
  64. {
  65. SftpSessionMock.Verify(p => p.RequestClose(_handle), Times.Once);
  66. }
  67. }
  68. }