♻️ Move output generation to state

[rhynodge.git] / src / test / java / net / pterodactylus / rhynodge / states / StateManagerTest.java

diff --git a/src/test/java/net/pterodactylus/rhynodge/states/StateManagerTest.java b/src/test/java/net/pterodactylus/rhynodge/states/StateManagerTest.java
index 0821b68..8f250ab 100644 (file)
--- a/src/test/java/net/pterodactylus/rhynodge/states/StateManagerTest.java
+++ b/src/test/java/net/pterodactylus/rhynodge/states/StateManagerTest.java
@@ -1,18 +1,12 @@
 package net.pterodactylus.rhynodge.states;
 
-import static com.google.common.base.Charsets.UTF_8;
-import static com.google.common.base.Objects.equal;
-import static org.apache.log4j.Level.OFF;
-import static org.apache.log4j.Logger.getLogger;
-import static org.hamcrest.MatcherAssert.assertThat;
-import static org.hamcrest.Matchers.is;
-
 import java.io.File;
 import java.io.IOException;
 import java.util.Optional;
+import javax.annotation.Nonnull;
 
 import net.pterodactylus.rhynodge.State;
-import net.pterodactylus.rhynodge.states.StateManager.Directory;
+import net.pterodactylus.rhynodge.states.StateManager.StateDirectory;
 
 import com.fasterxml.jackson.annotation.JsonProperty;
 import com.google.common.io.Files;
@@ -22,6 +16,13 @@ import org.junit.Rule;
 import org.junit.Test;
 import org.junit.rules.TemporaryFolder;
 
+import static com.google.common.base.Charsets.UTF_8;
+import static com.google.common.base.Objects.equal;
+import static org.apache.log4j.Level.OFF;
+import static org.apache.log4j.Logger.getLogger;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+
 /**
  * Unit test for {@link StateManager}.
  *
@@ -42,7 +43,7 @@ public class StateManagerTest {
        public StateManagerTest() throws IOException {
                tempFolder.create();
                statePath = tempFolder.newFolder();
-               stateManager = new StateManager(Directory.of(statePath.getPath()));
+               stateManager = new StateManager(StateDirectory.of(statePath.getPath()));
        }
 
        @Test
@@ -107,6 +108,12 @@ public class StateManagerTest {
                                        && (time() == testState.time());
                }
 
+               @Nonnull
+               @Override
+               protected String plainText() {
+                       return "Test";
+               }
+
        }
 
        public static class InvalidState extends AbstractState {
@@ -114,6 +121,12 @@ public class StateManagerTest {
                @JsonProperty
                private final Object someObject = new Object();
 
+               @Nonnull
+               @Override
+               protected String plainText() {
+                       return "Invalid";
+               }
+
        }
 
 }