♻️ 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 839b389..8f250ab 100644 (file)
--- a/src/test/java/net/pterodactylus/rhynodge/states/StateManagerTest.java
+++ b/src/test/java/net/pterodactylus/rhynodge/states/StateManagerTest.java
@@ -1,15 +1,9 @@
 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.StateDirectory;
@@ -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}.
  *
@@ -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";
+               }
+
        }
 
 }