From f31c2e9461069ebc78e79646114980e4b9681735 Mon Sep 17 00:00:00 2001
From: "miriam.baglioni" <miriam.baglioni@isti.cnr.it>
Date: Tue, 11 Aug 2020 15:49:25 +0200
Subject: [PATCH] enabled test

---
 .../graph/ExtractRelationFromEntityTest.java  | 105 +++++++-----------
 1 file changed, 38 insertions(+), 67 deletions(-)

diff --git a/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/graph/ExtractRelationFromEntityTest.java b/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/graph/ExtractRelationFromEntityTest.java
index 706089e4d..109458355 100644
--- a/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/graph/ExtractRelationFromEntityTest.java
+++ b/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/graph/ExtractRelationFromEntityTest.java
@@ -27,7 +27,6 @@ import eu.dnetlib.dhp.oa.graph.dump.community.CommunityMap;
 import eu.dnetlib.dhp.schema.dump.oaf.graph.Relation;
 import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpService;
 
-@Disabled
 public class ExtractRelationFromEntityTest {
 
 	private static final ObjectMapper OBJECT_MAPPER = new ObjectMapper();
@@ -39,34 +38,6 @@ public class ExtractRelationFromEntityTest {
 	private static final Logger log = LoggerFactory
 		.getLogger(ExtractRelationFromEntityTest.class);
 
-	private static CommunityMap map = new CommunityMap();
-
-	static {
-		map.put("egi", "EGI Federation");
-		map.put("fet-fp7", "FET FP7");
-		map.put("fet-h2020", "FET H2020");
-		map.put("clarin", "CLARIN");
-		map.put("fam", "Fisheries and Aquaculture Management");
-		map.put("ni", "Neuroinformatics");
-		map.put("mes", "European Marine Scinece");
-		map.put("instruct", "Instruct-Eric");
-		map.put("rda", "Research Data Alliance");
-		map.put("elixir-gr", "ELIXIR GR");
-		map.put("aginfra", "Agricultural and Food Sciences");
-		map.put("dariah", "DARIAH EU");
-		map.put("risis", "RISI");
-		map.put("ee", "SDSN - Greece");
-		map.put("oa-pg", "EC Post-Grant Open Access Pilot");
-		map.put("beopen", "Transport Research");
-		map.put("euromarine", "Euromarine");
-		map.put("ifremer", "Ifremer");
-		map.put("dh-ch", "Digital Humanities and Cultural Heritage");
-		map.put("science-innovation-policy", "Science and Innovation Policy Studies");
-		map.put("covid-19", "COVID-19");
-		map.put("enrmaps", "Energy Research");
-		map.put("epos", "EPOS");
-
-	}
 
 	@BeforeAll
 	public static void beforeAll() throws IOException {
@@ -97,43 +68,43 @@ public class ExtractRelationFromEntityTest {
 		spark.stop();
 	}
 
-//	@Test
-//	public void test1() {
-//
-//		final String sourcePath = getClass()
-//			.getResource("/eu/dnetlib/dhp/oa/graph/dump/resultDump/singelRecord_pub.json")
-//			.getPath();
-//
-//		final String communityMapPath = getClass()
-//			.getResource("/eu/dnetlib/dhp/oa/graph/dump/communityMapPath/communitymap.json")
-//			.getPath();
-//
-//		Extractor ex = new Extractor();
-//		ex
-//			.run(
-//				false, sourcePath, workingDir.toString() + "/relation",
-//				// eu.dnetlib.dhp.schema.oaf.Publication.class, communityMapPath);
-//				eu.dnetlib.dhp.schema.oaf.Publication.class, map);
-//
-//		final JavaSparkContext sc = JavaSparkContext.fromSparkContext(spark.sparkContext());
-//
-//		JavaRDD<Relation> tmp = sc
-//			.textFile(workingDir.toString() + "/relation")
-//			.map(item -> OBJECT_MAPPER.readValue(item, Relation.class));
-//
-//		org.apache.spark.sql.Dataset<Relation> verificationDataset = spark
-//			.createDataset(tmp.rdd(), Encoders.bean(Relation.class));
-//
-//		Assertions
-//			.assertEquals(
-//				9,
-//				verificationDataset.filter("source.id = '50|dedup_wf_001::15270b996fa8fd2fb5723daeab3685c3'").count());
-//
-//		Assertions
-//			.assertEquals(
-//				9,
-//				verificationDataset.filter("source.id = '50|dedup_wf_001::15270b996fa8fd2fb5723daxab3685c3'").count());
-//
-//	}
+	@Test
+	public void test1() {
+
+		final String sourcePath = getClass()
+			.getResource("/eu/dnetlib/dhp/oa/graph/dump/resultDump/singelRecord_pub.json")
+			.getPath();
+
+		final String communityMapPath = getClass()
+			.getResource("/eu/dnetlib/dhp/oa/graph/dump/communityMapPath/communitymap.json")
+			.getPath();
+
+		Extractor ex = new Extractor();
+		ex
+			.run(
+				false, sourcePath, workingDir.toString() + "/relation",
+				// eu.dnetlib.dhp.schema.oaf.Publication.class, communityMapPath);
+				eu.dnetlib.dhp.schema.oaf.Publication.class, communityMapPath);
+
+		final JavaSparkContext sc = JavaSparkContext.fromSparkContext(spark.sparkContext());
+
+		JavaRDD<Relation> tmp = sc
+			.textFile(workingDir.toString() + "/relation")
+			.map(item -> OBJECT_MAPPER.readValue(item, Relation.class));
+
+		org.apache.spark.sql.Dataset<Relation> verificationDataset = spark
+			.createDataset(tmp.rdd(), Encoders.bean(Relation.class));
+
+		Assertions
+			.assertEquals(
+				9,
+				verificationDataset.filter("source.id = '50|dedup_wf_001::15270b996fa8fd2fb5723daeab3685c3'").count());
+
+		Assertions
+			.assertEquals(
+				9,
+				verificationDataset.filter("source.id = '50|dedup_wf_001::15270b996fa8fd2fb5723daxab3685c3'").count());
+
+	}
 
 }