tests
This commit is contained in:
parent
d8f5be1149
commit
48c9070893
|
@ -33,7 +33,6 @@ public class FileCsvCollectorPlugin implements CollectorPlugin {
|
||||||
|
|
||||||
private static final Logger log = LoggerFactory.getLogger(FileCsvCollectorPlugin.class);
|
private static final Logger log = LoggerFactory.getLogger(FileCsvCollectorPlugin.class);
|
||||||
|
|
||||||
// TODO USE HDFS FILESSYSTEM
|
|
||||||
private final FileSystem fileSystem;
|
private final FileSystem fileSystem;
|
||||||
|
|
||||||
public FileCsvCollectorPlugin(final FileSystem fileSystem) {
|
public FileCsvCollectorPlugin(final FileSystem fileSystem) {
|
||||||
|
|
|
@ -43,7 +43,6 @@ public class HttpCsvCollectorPlugin implements CollectorPlugin {
|
||||||
|
|
||||||
private static final Logger log = LoggerFactory.getLogger(HttpCsvCollectorPlugin.class);
|
private static final Logger log = LoggerFactory.getLogger(HttpCsvCollectorPlugin.class);
|
||||||
|
|
||||||
// TODO USE HDFS FILESSYSTEM FOR TEMP FILE
|
|
||||||
private final FileSystem fileSystem;
|
private final FileSystem fileSystem;
|
||||||
|
|
||||||
private final HttpConnector2 httpConnector;
|
private final HttpConnector2 httpConnector;
|
||||||
|
|
|
@ -0,0 +1,48 @@
|
||||||
|
package eu.dnetlib.dhp.collection.plugin.csv;
|
||||||
|
|
||||||
|
import static org.junit.jupiter.api.Assertions.assertTrue;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
import org.apache.commons.lang3.StringUtils;
|
||||||
|
import org.apache.hadoop.conf.Configuration;
|
||||||
|
import org.apache.hadoop.fs.FileSystem;
|
||||||
|
import org.dom4j.DocumentException;
|
||||||
|
import org.dom4j.DocumentHelper;
|
||||||
|
import org.junit.jupiter.api.Disabled;
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
|
||||||
|
import eu.dnetlib.dhp.collection.ApiDescriptor;
|
||||||
|
import eu.dnetlib.dhp.collection.plugin.CollectorPlugin;
|
||||||
|
import eu.dnetlib.dhp.common.collection.CollectorException;
|
||||||
|
|
||||||
|
@Disabled
|
||||||
|
class FileCsvCollectorPluginTest {
|
||||||
|
|
||||||
|
private FileCsvCollectorPlugin plugin;
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void testCollect() throws CollectorException, DocumentException, IOException {
|
||||||
|
|
||||||
|
this.plugin = new FileCsvCollectorPlugin(FileSystem.getLocal(new Configuration()));
|
||||||
|
|
||||||
|
final ApiDescriptor api = new ApiDescriptor();
|
||||||
|
api.setBaseUrl("file:///tmp/test.csv");
|
||||||
|
api.setProtocol(CollectorPlugin.NAME.fileCSV.name());
|
||||||
|
api.getParams().put("header", "true");
|
||||||
|
api.getParams().put("identifier", "0");
|
||||||
|
api.getParams().put("separator", ",");
|
||||||
|
api.getParams().put("quote", "\"");
|
||||||
|
|
||||||
|
final List<String> list = this.plugin.collect(api, null).toList();
|
||||||
|
assertTrue(list.size() > 0);
|
||||||
|
|
||||||
|
for (final String xml : list) {
|
||||||
|
System.out.println(xml);
|
||||||
|
assertTrue(StringUtils.isNotBlank(xml));
|
||||||
|
assertTrue(StringUtils.isNotBlank((DocumentHelper.parseText(xml).valueOf("//*[@isId='true']"))));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,16 @@
|
||||||
|
package eu.dnetlib.dhp.collection.plugin.csv;
|
||||||
|
|
||||||
|
import static org.junit.jupiter.api.Assertions.fail;
|
||||||
|
|
||||||
|
import org.junit.jupiter.api.Disabled;
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
|
||||||
|
@Disabled
|
||||||
|
class HttpCsvCollectorPluginTest {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void testCollect() {
|
||||||
|
fail("Not yet implemented");
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in New Issue