/** Tests that reporting a lost file from the file system informs the client about this file. */ @Test public void reportLostFileTest() throws Exception { AlluxioURI path = new AlluxioURI("test"); mAlluxioLineageFileSystem.reportLostFile(path); Mockito.verify(mLineageMasterClient).reportLostFile("test"); // verify client is released Mockito.verify(mLineageContext).releaseMasterClient(mLineageMasterClient); }
/** Tests that a {@link DummyFileOutputStream} is returned. */ @Test public void getDummyOutStreamTest() throws Exception { AlluxioURI path = new AlluxioURI("test"); Mockito.when(mLineageMasterClient.reinitializeFile("test", TEST_BLOCK_SIZE, 0)).thenReturn(-1L); CreateFileOptions options = CreateFileOptions.defaults().setBlockSizeBytes(TEST_BLOCK_SIZE).setTtl(0); FileOutStream outStream = mAlluxioLineageFileSystem.createFile(path, options); Assert.assertTrue(outStream instanceof DummyFileOutputStream); // verify client is released Mockito.verify(mLineageContext).releaseMasterClient(mLineageMasterClient); }
/** Sets up all dependencies before running a test. */ @Before public void before() { mLineageMasterClient = PowerMockito.mock(LineageMasterClient.class); mLineageContext = PowerMockito.mock(LineageContext.class); Mockito.when(mLineageContext.acquireMasterClient()).thenReturn(mLineageMasterClient); Whitebox.setInternalState(LineageContext.class, "INSTANCE", mLineageContext); mAlluxioLineageFileSystem = LineageFileSystem.get(); Whitebox.setInternalState(mAlluxioLineageFileSystem, "mLineageContext", mLineageContext); FileSystemContext fileSystemContext = PowerMockito.mock(FileSystemContext.class); FileSystemMasterClient fileSystemMasterClient = PowerMockito.mock(FileSystemMasterClient.class); Mockito.when(fileSystemContext.acquireMasterClient()).thenReturn(fileSystemMasterClient); Whitebox.setInternalState(FileSystemContext.class, "INSTANCE", fileSystemContext); Whitebox.setInternalState(mAlluxioLineageFileSystem, "mContext", fileSystemContext); }