From fe0a7870e67f3f19ad55e715df2fe132770e54ca Mon Sep 17 00:00:00 2001
From: Sandro La Bruzzo <sandro.labruzzo@isti.cnr.it>
Date: Thu, 8 Oct 2020 10:33:12 +0200
Subject: [PATCH] Added test to check if merge authors works

---
 .../SparkScholexplorerAggregationTest.scala   | 54 +++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/sx/graph/SparkScholexplorerAggregationTest.scala

diff --git a/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/sx/graph/SparkScholexplorerAggregationTest.scala b/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/sx/graph/SparkScholexplorerAggregationTest.scala
new file mode 100644
index 0000000000..4d83057f2e
--- /dev/null
+++ b/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/sx/graph/SparkScholexplorerAggregationTest.scala
@@ -0,0 +1,54 @@
+package eu.dnetlib.dhp.sx.graph
+
+import com.fasterxml.jackson.databind.{ObjectMapper, SerializationFeature}
+import eu.dnetlib.dhp.schema.scholexplorer.DLIPublication
+import eu.dnetlib.dhp.sx.ebi.EBIAggregator
+import org.apache.spark.sql.{Dataset, Encoder, Encoders, SparkSession}
+import org.junit.jupiter.api.Assertions._
+import org.junit.jupiter.api.Test
+
+import scala.io.Source
+
+class SparkScholexplorerAggregationTest {
+
+
+  @Test
+  def testFunderRelationshipsMapping(): Unit = {
+    val publications = Source.fromInputStream(getClass.getResourceAsStream("publication.json")).mkString
+
+    var s: List[DLIPublication] = List[DLIPublication]()
+
+    val m: ObjectMapper = new ObjectMapper()
+
+    m.enable(SerializationFeature.INDENT_OUTPUT)
+
+    for (line <- publications.lines) {
+      s = m.readValue(line, classOf[DLIPublication]) :: s
+
+
+    }
+
+
+    implicit val pubEncoder: Encoder[DLIPublication] = Encoders.kryo[DLIPublication]
+    val spark: SparkSession = SparkSession.builder().appName("Test").master("local[*]").getOrCreate()
+
+
+    val ds: Dataset[DLIPublication] = spark.createDataset(spark.sparkContext.parallelize(s)).as[DLIPublication]
+
+    val unique = ds.map(d => (d.getId, d))(Encoders.tuple(Encoders.STRING, pubEncoder))
+      .groupByKey(_._1)(Encoders.STRING)
+      .agg(EBIAggregator.getDLIPublicationAggregator().toColumn)
+      .map(p => p._2)
+
+    val uniquePubs: DLIPublication = unique.first()
+
+    s.foreach(pp => assertFalse(pp.getAuthor.isEmpty))
+
+
+    assertNotNull(uniquePubs.getAuthor)
+    assertFalse(uniquePubs.getAuthor.isEmpty)
+
+
+  }
+
+}