@Before public void setUp() throws Exception { fileStorage = new HdfsFileStorage(); }
@Test public void testUploadJarWithDir() throws Exception { Map<String, String> config = new HashMap<>(); config.put(HdfsFileStorage.CONFIG_FSURL, "file:///"); config.put(HdfsFileStorage.CONFIG_DIRECTORY, HDFS_DIR); fileStorage.init(config); File file = File.createTempFile("test", ".tmp"); file.deleteOnExit(); List<String> lines = Arrays.asList("test-line-1", "test-line-2"); Files.write(file.toPath(), lines, Charset.forName("UTF-8")); String jarFileName = "test.jar"; fileStorage.delete(jarFileName); fileStorage.upload(new FileInputStream(file), jarFileName); InputStream inputStream = fileStorage.download(jarFileName); List<String> actual = IOUtils.readLines(inputStream); Assert.assertEquals(lines, actual); }
@Test(expected = RuntimeException.class) public void testInitWithoutFsUrl() throws Exception { fileStorage.init(new HashMap<String, String>()); }
@Test public void testUploadJar() throws Exception { Map<String, String> config = new HashMap<>(); config.put(HdfsFileStorage.CONFIG_FSURL, "file:///"); fileStorage.init(config); File file = File.createTempFile("test", ".tmp"); file.deleteOnExit(); List<String> lines = Arrays.asList("test-line-1", "test-line-2"); Files.write(file.toPath(), lines, Charset.forName("UTF-8")); String jarFileName = "test.jar"; fileStorage.delete(jarFileName); fileStorage.upload(new FileInputStream(file), jarFileName); InputStream inputStream = fileStorage.download(jarFileName); List<String> actual = IOUtils.readLines(inputStream); Assert.assertEquals(lines, actual); }