From 69e9ea9eeb458d3d92b52b39e74e00698f5d5f9e Mon Sep 17 00:00:00 2001
From: "miriam.baglioni" <miriam.baglioni@isti.cnr.it>
Date: Thu, 23 Dec 2021 10:15:30 +0100
Subject: [PATCH] [Graph Dump] Test for extraction of rels from entities
 extended

---
 .../ExtractRelationFromEntityTest.java        | 34 +++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/complete/ExtractRelationFromEntityTest.java b/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/complete/ExtractRelationFromEntityTest.java
index 511236d82..93d520eb4 100644
--- a/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/complete/ExtractRelationFromEntityTest.java
+++ b/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/complete/ExtractRelationFromEntityTest.java
@@ -115,6 +115,23 @@ public class ExtractRelationFromEntityTest {
 					.getReltype()
 					.getType());
 
+		Assertions
+				.assertEquals(
+						"context", verificationDataset
+								.filter((FilterFunction<Relation>) row -> row.getSource().getId().startsWith("00"))
+								.collectAsList()
+								.get(0)
+								.getSource()
+								.getType());
+
+		Assertions
+				.assertEquals(
+						"result", verificationDataset
+								.filter((FilterFunction<Relation>) row -> row.getSource().getId().startsWith("00"))
+								.collectAsList()
+								.get(0)
+								.getTarget()
+								.getType());
 		Assertions
 			.assertEquals(
 				"IsRelatedTo", verificationDataset
@@ -133,6 +150,23 @@ public class ExtractRelationFromEntityTest {
 					.getReltype()
 					.getType());
 
+		Assertions
+				.assertEquals(
+						"context", verificationDataset
+								.filter((FilterFunction<Relation>) row -> row.getTarget().getId().startsWith("00"))
+								.collectAsList()
+								.get(0)
+								.getTarget()
+								.getType());
+
+		Assertions
+				.assertEquals(
+						"result", verificationDataset
+								.filter((FilterFunction<Relation>) row -> row.getTarget().getId().startsWith("00"))
+								.collectAsList()
+								.get(0)
+								.getSource()
+								.getType());
 	}
 
 }