Fix CryptoStream.Dispose

[mono.git] / mcs / class / corlib / Test / System.Security.Cryptography / CryptoStreamTest.cs

diff --git a/mcs/class/corlib/Test/System.Security.Cryptography/CryptoStreamTest.cs b/mcs/class/corlib/Test/System.Security.Cryptography/CryptoStreamTest.cs
index c4bfae156a5d6951f7664c322d1ca5a5a138e445..60746defc1cff1dba0ee6b0f3f54db9130733536 100644 (file)
--- a/mcs/class/corlib/Test/System.Security.Cryptography/CryptoStreamTest.cs
+++ b/mcs/class/corlib/Test/System.Security.Cryptography/CryptoStreamTest.cs
@@ -1389,6 +1389,46 @@ namespace MonoTests.System.Security.Cryptography {
 #endif
                }
 
+               [Test]
+               public void ReadModeDispose_FinalBlock ()
+               {
+                       using (SHA1 sha1 = SHA1.Create()) {
+                               using (MemoryStream mem = new MemoryStream(new byte[] { 1, 2, 3 }, false))
+                                       using (CryptoStream cs = new CryptoStream(mem, sha1, CryptoStreamMode.Read))
+                                       {
+                                       }
+                               byte b = sha1.Hash [0]; // This will throw if TransformFinalBlock not called in sha1
+                               GC.KeepAlive (b); // just the warning...
+                       }
+                }
+
+               [Test]
+               public void CustomDisposeCalled ()
+               {
+                       using (MemoryStream mem = new MemoryStream(new byte[] { 1, 2, 3 }, false)) {
+                               MyCryptoStream cs;
+                               using (cs = new MyCryptoStream (mem, SHA1.Create()))
+                               {
+                               }
+                               Assert.IsTrue (cs.DisposeCalled, "#1");
+                       }
+               }
+
+               class MyCryptoStream : CryptoStream {
+                       public bool DisposeCalled { get; private set;}
+
+                       public MyCryptoStream(Stream stream, ICryptoTransform transform)
+                                               : base(stream, transform, CryptoStreamMode.Read)
+                       {
+                       }
+
+                       protected override void Dispose(bool disposing)
+                       {
+                               base.Dispose(disposing);
+                               DisposeCalled = true;
+                       }
+               }
+
                class ExpandTransform : ICryptoTransform {
 
                        public bool CanReuseTransform {