SftpFileStreamTest_Dispose_SessionOpen.cs 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. using System;
  2. using System.Globalization;
  3. using System.IO;
  4. using Microsoft.VisualStudio.TestTools.UnitTesting;
  5. using Moq;
  6. using Renci.SshNet.Sftp;
  7. namespace Renci.SshNet.Tests.Classes.Sftp
  8. {
  9. [TestClass]
  10. public class SftpFileStreamTest_Dispose_SessionOpen : SftpFileStreamTestBase
  11. {
  12. private SftpFileStream _target;
  13. private string _path;
  14. private byte[] _handle;
  15. private uint _bufferSize;
  16. private uint _readBufferSize;
  17. private uint _writeBufferSize;
  18. protected override void SetupData()
  19. {
  20. base.SetupData();
  21. var random = new Random();
  22. _path = random.Next().ToString(CultureInfo.InvariantCulture);
  23. _handle = GenerateRandom(2, random);
  24. _bufferSize = (uint)random.Next(1, 1000);
  25. _readBufferSize = (uint)random.Next(1, 1000);
  26. _writeBufferSize = (uint)random.Next(1, 1000);
  27. }
  28. protected override void SetupMocks()
  29. {
  30. _ = SftpSessionMock.InSequence(MockSequence)
  31. .Setup(p => p.RequestOpen(_path, Flags.Write | Flags.Truncate, false))
  32. .Returns(_handle);
  33. _ = SftpSessionMock.InSequence(MockSequence)
  34. .Setup(p => p.CalculateOptimalReadLength(_bufferSize))
  35. .Returns(_readBufferSize);
  36. _ = SftpSessionMock.InSequence(MockSequence)
  37. .Setup(p => p.CalculateOptimalWriteLength(_bufferSize, _handle))
  38. .Returns(_writeBufferSize);
  39. _ = SftpSessionMock.InSequence(MockSequence)
  40. .Setup(p => p.IsOpen).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.Truncate,
  50. FileAccess.Write,
  51. (int)_bufferSize);
  52. }
  53. protected override void Act()
  54. {
  55. _target.Dispose();
  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. }