added unit tests for #58

cryptomator · May 22, 2019 · 5916093 · 5916093
1 parent 1b8bf36
commit 5916093
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/src/test/java/org/cryptomator/cryptofs/ch/CleartextFileChannelTest.java b/src/test/java/org/cryptomator/cryptofs/ch/CleartextFileChannelTest.java
@@ -442,6 +442,29 @@ public void writeAfterEof() throws IOException {
 			Assertions.assertEquals(210l, inTest.size());
 		}
 
+		@Test
+		@DisplayName("write header if it isn't already written")
+		public void testWriteHeaderIfNeeded() throws IOException {
+			when(options.writable()).thenReturn(true);
+
+			inTest.force(true);
+			inTest.force(true);
+			inTest.force(true);
+
+			Mockito.verify(ciphertextFileChannel, Mockito.times(1)).write(Mockito.any(), Mockito.eq(0l));
+		}
+
+		@Test
+		@DisplayName("don't write header if it is already written")
+		public void testDontRewriteHeader() throws IOException {
+			when(options.writable()).thenReturn(true);
+			inTest = new CleartextFileChannel(ciphertextFileChannel, header, false, readWriteLock, cryptor, chunkCache, options, fileSize, lastModified, attributeViewSupplier, exceptionsDuringWrite, closeListener, stats);
+
+			inTest.force(true);
+
+			Mockito.verify(ciphertextFileChannel, Mockito.never()).write(Mockito.any(), Mockito.eq(0l));
+		}
+
 	}
 
 	@Nested