102 lines
2.7 KiB
Java
102 lines
2.7 KiB
Java
|
|
package eu.dnetlib.dhp.oa.graph.merge;
|
|
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
|
import static org.junit.jupiter.api.Assertions.assertNotNull;
|
|
|
|
import java.io.IOException;
|
|
import java.util.Optional;
|
|
|
|
import org.apache.commons.io.IOUtils;
|
|
import org.junit.jupiter.api.BeforeEach;
|
|
import org.junit.jupiter.api.Test;
|
|
|
|
import com.fasterxml.jackson.databind.DeserializationFeature;
|
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
|
|
|
import eu.dnetlib.dhp.schema.oaf.Datasource;
|
|
|
|
class MergeGraphTableSparkJobTest {
|
|
|
|
private ObjectMapper mapper;
|
|
|
|
@BeforeEach
|
|
public void setUp() {
|
|
mapper = new ObjectMapper().configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
|
|
}
|
|
|
|
@Test
|
|
void testMerge() throws IOException {
|
|
Datasource d = MergeGraphTableSparkJob
|
|
.mergeDatasource(
|
|
d("datasource_cris.json"),
|
|
d("datasource_openaire2.0.json"));
|
|
|
|
assertEquals("10|274269ac6f3b::2a2e2793b500f3f7b47ef24b1a9277b7", d.getId());
|
|
assertNotNull(d.getOriginalId());
|
|
assertEquals(2, d.getOriginalId().size());
|
|
assertNotNull(d.getCollectedfrom());
|
|
assertEquals(2, d.getCollectedfrom().size());
|
|
assertNotNull(d.getPid());
|
|
assertEquals(1, d.getPid().size());
|
|
}
|
|
|
|
@Test
|
|
void testMergeCompatibility() throws IOException {
|
|
assertEquals(
|
|
"openaire-cris_1.1",
|
|
MergeGraphTableSparkJob
|
|
.mergeDatasource(
|
|
d("datasource_cris.json"),
|
|
d("datasource_UNKNOWN.json"))
|
|
.getOpenairecompatibility()
|
|
.getClassid());
|
|
assertEquals(
|
|
"openaire-cris_1.1",
|
|
MergeGraphTableSparkJob
|
|
.mergeDatasource(
|
|
d("datasource_UNKNOWN.json"),
|
|
d("datasource_cris.json"))
|
|
.getOpenairecompatibility()
|
|
.getClassid());
|
|
assertEquals(
|
|
"driver-openaire2.0",
|
|
MergeGraphTableSparkJob
|
|
.mergeDatasource(
|
|
d("datasource_native.json"),
|
|
d("datasource_driver-openaire2.0.json"))
|
|
.getOpenairecompatibility()
|
|
.getClassid());
|
|
assertEquals(
|
|
"driver-openaire2.0",
|
|
MergeGraphTableSparkJob
|
|
.mergeDatasource(
|
|
d("datasource_driver-openaire2.0.json"),
|
|
d("datasource_native.json"))
|
|
.getOpenairecompatibility()
|
|
.getClassid());
|
|
assertEquals(
|
|
"openaire4.0",
|
|
MergeGraphTableSparkJob
|
|
.mergeDatasource(
|
|
d("datasource_notCompatible.json"),
|
|
d("datasource_openaire4.0.json"))
|
|
.getOpenairecompatibility()
|
|
.getClassid());
|
|
assertEquals(
|
|
"notCompatible",
|
|
MergeGraphTableSparkJob
|
|
.mergeDatasource(
|
|
d("datasource_notCompatible.json"),
|
|
d("datasource_UNKNOWN.json"))
|
|
.getOpenairecompatibility()
|
|
.getClassid());
|
|
}
|
|
|
|
private Optional<Datasource> d(String file) throws IOException {
|
|
String json = IOUtils.toString(getClass().getResourceAsStream(file));
|
|
return Optional.of(mapper.readValue(json, Datasource.class));
|
|
}
|
|
|
|
}
|