diff --git a/core/src/test/java/org/elasticsearch/env/ESFileStoreTests.java b/core/src/test/java/org/elasticsearch/env/ESFileStoreTests.java index a0e33979a03..7ce278708a0 100644 --- 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.FileStore; import java.nio.file.attribute.FileAttributeView; import java.nio.file.attribute.FileStoreAttributeView; +import static org.mockito.Mockito.mock; +import static org.mockito.Mockito.when; + 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 type) { - return false; - } - - @Override - public boolean supportsFileAttributeView(String name) { - return false; - } - - @Override - public V getFileStoreAttributeView(Class type) { - return null; - } - - @Override - public Object getAttribute(String attribute) throws IOException { - return null; - } - } - 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());