Check ref content.

hltcoe · Oct 17, 2023 · a7fb843 · a7fb843
1 parent f03b86a
commit a7fb843
Showing 1 changed file with 29 additions and 3 deletions.
diff --git a/test_nodejs/communication_fu_test.js b/test_nodejs/communication_fu_test.js
@@ -7,9 +7,8 @@ describe("addInternalReferences", function() {
   it("Returns Communication-level maps", function() {
     const comm = new concrete.communication.Communication();
     comm.initFromTJSONProtocolObject(commJSONData);
-    expect(
-      comm.addInternalReferences()
-    ).to.have.all.keys([
+    const refs = comm.addInternalReferences();
+    expect(refs).to.have.all.keys([
       "entityForUUID",
       "entityMentionForUUID",
       "sectionForUUID",
@@ -18,5 +17,32 @@ describe("addInternalReferences", function() {
       "situationMentionForUUID",
       "tokenizationForUUID"
     ]);
+    comm.sectionList.forEach((section) => {
+      expect(refs.sectionForUUID[section.uuid.uuidString]).to.equal(section);
+      section.sentenceList.forEach((sentence) => {
+        expect(refs.sentenceForUUID[sentence.uuid.uuidString]).to.equal(sentence);
+        expect(refs.tokenizationForUUID[sentence.tokenization.uuid.uuidString]).to.equal(sentence.tokenization);
+      });
+    });
+    comm.entitySetList.forEach((entitySet) =>
+      entitySet.entityList.forEach((entity) =>
+        expect(refs.entityForUUID[entity.uuid.uuidString]).to.equal(entity)
+      )
+    );
+    comm.entityMentionSetList.forEach((entityMentionSet) =>
+      entityMentionSet.mentionList.forEach((mention) =>
+        expect(refs.entityMentionForUUID[mention.uuid.uuidString]).to.equal(mention)
+      )
+    );
+    comm.situationSetList.forEach((situationSet) =>
+      situationSet.situationList.forEach((situation) =>
+        expect(refs.situationForUUID[situation.uuid.uuidString]).to.equal(situation)
+      )
+    );
+    comm.situationMentionSetList.forEach((situationMentionSet) =>
+      situationMentionSet.mentionList.forEach((mention) =>
+        expect(refs.situationMentionForUUID[mention.uuid.uuidString]).to.equal(mention)
+      )
+    );
   });
 });