add aff_raw field to augment pipeline message

adsabs · Jan 8, 2020 · 0ab8b32 · 0ab8b32
1 parent ad4778a
commit 0ab8b32
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 4 deletions.
diff --git a/adsmsg/protobuf/augmentrecord_pb2.py b/adsmsg/protobuf/augmentrecord_pb2.py
diff --git a/adsmsg/tests/test_affiliation_record.py b/adsmsg/tests/test_affiliation_record.py
@@ -71,6 +71,9 @@ def test_affiliation_response(self):
              "aff_id": ['placeholder'],
              "institution": [
                 "CfA"
+             ],
+             'aff_raw':  [
+                "Harvard-Smithsonian Center for Astrophysics, Cambridge, MA."
              ]
         }
         a = AugmentAffiliationResponseRecord(**d)
@@ -82,6 +85,7 @@ def test_affiliation_response(self):
         self.assertEqual(a.aff_canonical, d['aff_canonical'])
         self.assertEqual(a.aff_facet_hier, d['aff_facet_hier'])
         self.assertEqual(a.aff_id, d['aff_id'])
+        self.assertEqual(a.aff_raw, d['aff_raw'])
         self.assertEqual(a.institution, d['institution'])
 
     def test_affiliation_response_list(self):
@@ -108,6 +112,9 @@ def test_affiliation_response_list(self):
              "aff_id": ['placeholder'],
              "institution": [
                 "CfA"
+             ],
+             'aff_raw':  [
+                "Harvard-Smithsonian Center for Astrophysics, Cambridge, MA."
              ]
         }
         d_list = [d]
@@ -124,4 +131,5 @@ def test_affiliation_response_list(self):
             self.assertEqual(m.affiliation_responses[i].aff_canonical, d_list[i]['aff_canonical'])
             self.assertEqual(m.affiliation_responses[i].aff_facet_hier, d_list[i]['aff_facet_hier'])
             self.assertEqual(m.affiliation_responses[i].aff_id, d_list[i]['aff_id'])
+            self.assertEqual(m.affiliation_responses[i].aff_raw, d_list[i]['aff_raw'])
             self.assertEqual(m.affiliation_responses[i].institution, d_list[i]['institution'])
diff --git a/specs/augmentrecord.proto b/specs/augmentrecord.proto
@@ -32,6 +32,7 @@ message AugmentAffiliationResponseRecord {
     repeated string aff_facet_hier = 8;
     repeated string aff_id = 9;
     repeated string institution = 10;
+    repeated string aff_raw = 11;
 }
 
 // also support lists of the above