🐛 Serialize the success value, remove equals override

[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 5837846..da1bb90 100644 (file)
--- a/src/test/java/net/pterodactylus/rhynodge/states/StateManagerTest.java
+++ b/src/test/java/net/pterodactylus/rhynodge/states/StateManagerTest.java
@@ -1,19 +1,14 @@
 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;
 
 import com.fasterxml.jackson.annotation.JsonProperty;
-import com.google.common.base.Optional;
 import com.google.common.io.Files;
 import org.hamcrest.Matchers;
 import org.junit.BeforeClass;
@@ -21,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}.
  *
@@ -41,7 +43,7 @@ public class StateManagerTest {
        public StateManagerTest() throws IOException {
                tempFolder.create();
                statePath = tempFolder.newFolder();
-               stateManager = new StateManager(statePath.getPath());
+               stateManager = new StateManager(StateDirectory.of(statePath.getPath()));
        }
 
        @Test
@@ -95,15 +97,10 @@ public class StateManagerTest {
                        super(success);
                }
 
+               @Nonnull
                @Override
-               public boolean equals(Object object) {
-                       if (!(object instanceof TestState)) {
-                               return false;
-                       }
-                       TestState testState = (TestState) object;
-                       return equal(exception(), testState.exception())
-                                       && (failCount() == testState.failCount())
-                                       && (time() == testState.time());
+               protected String plainText() {
+                       return "Test";
                }
 
        }
@@ -113,6 +110,12 @@ public class StateManagerTest {
                @JsonProperty
                private final Object someObject = new Object();
 
+               @Nonnull
+               @Override
+               protected String plainText() {
+                       return "Invalid";
+               }
+
        }
 
 }