use mockito instead

T1l3 · Jul 29, 2016 · 59181c8 · 59181c8
1 parent 37e0e63
commit 59181c8
Showing 1 changed file with 9 additions and 61 deletions.
diff --git a/core/src/test/java/org/elasticsearch/env/ESFileStoreTests.java b/core/src/test/java/org/elasticsearch/env/ESFileStoreTests.java
@@ -25,69 +25,17 @@
 import java.nio.file.attribute.FileAttributeView;
 import java.nio.file.attribute.FileStoreAttributeView;
 
-public class ESFileStoreTests extends ESTestCase {
-    static class MockFileStore extends FileStore {
-
-        public long usableSpace = -1;
-
-        @Override
-        public String type() {
-            return "mock";
-        }
-
-        @Override
-        public String name() {
-            return "mock";
-        }
-
-        @Override
-        public String toString() {
-            return "mock";
-        }
-
-        @Override
-        public boolean isReadOnly() {
-            return false;
-        }
-
-        @Override
-        public long getTotalSpace() throws IOException {
-            return usableSpace*3;
-        }
-
-        @Override
-        public long getUsableSpace() throws IOException {
-            return usableSpace;
-        }
-
-        @Override
-        public long getUnallocatedSpace() throws IOException {
-            return usableSpace*2;
-        }
-
-        @Override
-        public boolean supportsFileAttributeView(Class<? extends FileAttributeView> type) {
-            return false;
-        }
-
-        @Override
-        public boolean supportsFileAttributeView(String name) {
-            return false;
-        }
-
-        @Override
-        public <V extends FileStoreAttributeView> V getFileStoreAttributeView(Class<V> type) {
-            return null;
-        }
-
-        @Override
-        public Object getAttribute(String attribute) throws IOException {
-            return null;
-        }
-    }
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
 
+public class ESFileStoreTests extends ESTestCase {
     public void testNegativeSpace() throws Exception {
-        FileStore store = new ESFileStore(new MockFileStore());
+        FileStore mocked = mock(FileStore.class);
+        when(mocked.getUsableSpace()).thenReturn(-1L);
+        when(mocked.getTotalSpace()).thenReturn(-1L);
+        when(mocked.getUnallocatedSpace()).thenReturn(-1L);
+        assertEquals(-1, mocked.getUsableSpace());
+        FileStore store = new ESFileStore(mocked);
         assertEquals(Long.MAX_VALUE, store.getUsableSpace());
         assertEquals(Long.MAX_VALUE, store.getTotalSpace());
         assertEquals(Long.MAX_VALUE, store.getUnallocatedSpace());