Browse Source

Perform some cleanup, add another test.

Sam Jaffe 5 years ago
parent
commit
25d45d7306

+ 12 - 5
src/test/java/org/leumasjaffe/recipe/controller/FileControllerTest.java

@@ -4,8 +4,8 @@ import static org.junit.jupiter.api.Assertions.*;
 import static org.mockito.Mockito.*;
 
 import java.awt.Component;
-import java.awt.ComponentOrientation;
 import java.io.StringReader;
+import java.io.StringWriter;
 import java.util.Optional;
 
 import org.junit.jupiter.api.BeforeEach;
@@ -15,13 +15,10 @@ import org.mockito.ArgumentCaptor;
 import org.mockito.InOrder;
 import org.mockito.Mock;
 import org.mockito.MockitoAnnotations;
-import org.mockito.Spy;
 
 class FileControllerTest {
 	@SuppressWarnings("serial")
-	private static class StubComponent extends Component implements FileController.Model {
-		@Override public void setModel(Recipe model) {}
-	}
+	private static abstract class StubComponent extends Component implements FileController.Model {}
 	
 	StubComponent stub;
 	FileController<StubComponent> controller;
@@ -77,5 +74,15 @@ class FileControllerTest {
 		
 		verify(controller).errorPopup(any());
 	}
+	
+	@Test
+	void testCanWriteRecipeToStream() {
+		StringWriter writer = new StringWriter();
+		doReturn(Optional.of(writer)).when(controller).getSaved();
+		
+		controller.create();
+		controller.save();
+		assertNotEquals("", writer.toString());
+	}
 
 }