From 64d644d8b8fcaeb427b2377b50e6f56adb3fd316 Mon Sep 17 00:00:00 2001
From: akwizgran <akwizgran@users.sourceforge.net>
Date: Fri, 5 Dec 2014 19:22:35 +0000
Subject: [PATCH] Clean up output streams in unit test.

This fixes some lint warnings and may also fix dev task #102.
---
 .../transport/StreamWriterImplTest.java       | 42 ++++++++++++++++++-
 1 file changed, 41 insertions(+), 1 deletion(-)

diff --git a/briar-tests/src/org/briarproject/transport/StreamWriterImplTest.java b/briar-tests/src/org/briarproject/transport/StreamWriterImplTest.java
index 8fc9cd6f1a..e55e9092d0 100644
--- a/briar-tests/src/org/briarproject/transport/StreamWriterImplTest.java
+++ b/briar-tests/src/org/briarproject/transport/StreamWriterImplTest.java
@@ -31,7 +31,7 @@ public class StreamWriterImplTest extends BriarTestCase {
 	}
 
 	@Test
-	public void testFlushWithoutBufferedDataWritesFrame() throws Exception {
+	public void testFlushWithoutBufferedDataOnlyFlushes() throws Exception {
 		Mockery context = new Mockery();
 		final FrameWriter writer = context.mock(FrameWriter.class);
 		StreamWriterImpl w = new StreamWriterImpl(writer, FRAME_LENGTH);
@@ -41,6 +41,16 @@ public class StreamWriterImplTest extends BriarTestCase {
 		}});
 		w.flush();
 		context.assertIsSatisfied();
+
+		// Clean up
+		context.checking(new Expectations() {{
+			// Closing the writer writes a final frame and flushes again
+			oneOf(writer).writeFrame(with(any(byte[].class)), with(0),
+					with(true));
+			oneOf(writer).flush();
+		}});
+		w.close();
+		context.assertIsSatisfied();
 	}
 
 	@Test
@@ -59,6 +69,16 @@ public class StreamWriterImplTest extends BriarTestCase {
 		w.write(0);
 		w.flush();
 		context.assertIsSatisfied();
+
+		// Clean up
+		context.checking(new Expectations() {{
+			// Closing the writer writes a final frame and flushes again
+			oneOf(writer).writeFrame(with(any(byte[].class)), with(0),
+					with(true));
+			oneOf(writer).flush();
+		}});
+		w.close();
+		context.assertIsSatisfied();
 	}
 
 	@Test
@@ -75,6 +95,16 @@ public class StreamWriterImplTest extends BriarTestCase {
 			w.write(0);
 		}
 		context.assertIsSatisfied();
+
+		// Clean up
+		context.checking(new Expectations() {{
+			// Closing the writer writes a final frame and flushes again
+			oneOf(writer).writeFrame(with(any(byte[].class)), with(0),
+					with(true));
+			oneOf(writer).flush();
+		}});
+		w.close();
+		context.assertIsSatisfied();
 	}
 
 	@Test
@@ -96,6 +126,16 @@ public class StreamWriterImplTest extends BriarTestCase {
 		w.write(b);
 		w.write(b);
 		context.assertIsSatisfied();
+
+		// Clean up
+		context.checking(new Expectations() {{
+			// Closing the writer writes a final frame and flushes again
+			oneOf(writer).writeFrame(with(any(byte[].class)), with(0),
+					with(true));
+			oneOf(writer).flush();
+		}});
+		w.close();
+		context.assertIsSatisfied();
 	}
 
 	@Test
-- 
GitLab