From 1695d45bd41a772319d357642df536cd8634faf2 Mon Sep 17 00:00:00 2001
From: "miriam.baglioni" <miriam.baglioni@isti.cnr.it>
Date: Fri, 30 Jul 2021 17:57:01 +0200
Subject: [PATCH] Hosted By Map - Test class to verify the preparation of the
 intermediate information

---
 .../oa/graph/hostedbymap/TestPrepare.scala    | 53 +++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/hostedbymap/TestPrepare.scala

diff --git a/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/hostedbymap/TestPrepare.scala b/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/hostedbymap/TestPrepare.scala
new file mode 100644
index 000000000..f01e4f59f
--- /dev/null
+++ b/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/hostedbymap/TestPrepare.scala
@@ -0,0 +1,53 @@
+package eu.dnetlib.dhp.oa.graph.hostedbymap
+
+import com.fasterxml.jackson.databind.ObjectMapper
+import eu.dnetlib.dhp.oa.graph.hostedbymap.model.DatasourceInfo
+import org.apache.spark.SparkConf
+import org.apache.spark.sql.{Dataset, SparkSession}
+import org.junit.jupiter.api.Assertions.{assertEquals, assertTrue}
+import org.junit.jupiter.api.Test
+
+class TestPrepare extends java.io.Serializable{
+
+  @Test
+  def testPrepareDatasource():Unit = {
+    val conf = new SparkConf()
+    conf.setMaster("local[*]")
+    conf.set("spark.driver.host", "localhost")
+    val spark: SparkSession =
+      SparkSession
+        .builder()
+        .appName(getClass.getSimpleName)
+        .config(conf)
+        .getOrCreate()
+    val path = getClass.getResource("datasource.json").getPath
+
+    val ds :Dataset[DatasourceInfo]= SparkPrepareHostedByInfoToApply.prepareDatasourceInfo(spark, path)
+
+    val mapper :ObjectMapper = new ObjectMapper()
+
+    assertEquals(9,  ds.count)
+
+    assertEquals(8, ds.filter(hbi => !hbi.getIssn.equals("")).count)
+    assertEquals(5, ds.filter(hbi => !hbi.getEissn.equals("")).count)
+    assertEquals(0, ds.filter(hbi => !hbi.getLissn.equals("")).count)
+
+    assertEquals(0, ds.filter(hbi => hbi.getIssn.equals("") && hbi.getEissn.equals("") && hbi.getLissn.equals("")).count)
+
+    assertTrue(ds.filter(hbi => hbi.getIssn.equals("0212-8365")).count == 1)
+    assertTrue(ds.filter(hbi => hbi.getEissn.equals("2253-900X")).count == 1)
+    assertTrue(ds.filter(hbi => hbi.getIssn.equals("0212-8365") && hbi.getEissn.equals("2253-900X")).count == 1)
+    assertTrue(ds.filter(hbi => hbi.getIssn.equals("0212-8365") && hbi.getOfficialname.equals("Thémata")).count == 1)
+    assertTrue(ds.filter(hbi => hbi.getIssn.equals("0212-8365") && hbi.getId.equals("10|doajarticles::abbc9265bea9ff62776a1c39785af00c")).count == 1)
+    ds.foreach(hbi => assertTrue(hbi.getId.startsWith("10|")))
+    ds.foreach(e => println(mapper.writeValueAsString(e)))
+    spark.close()
+  }
+
+  @Test
+  def testPrepareHostedByMap():Unit = {
+
+  }
+
+
+}