GQL-125: Add RestoreCitationRevision mutation and retrieving collections associated to the citation records (#174)

* GQL-125: Add RestoreCitationRevision mutation

* GQL-125: Adding tests

* GQL-125: Fixing naming

* GQL-125: Allowing associated collections to be returned from Citation

* GQL-125: Updating tests to use Name

Author: william-valencia

src/datasources/__tests__/citation.test.js

@@ -2,7 +2,8 @@ import nock from 'nock'
 
 import {
   deleteCitation as citationSourceDelete,
-  fetchCitations as citationSourceFetch
+  fetchCitations as citationSourceFetch,
+  restoreCitationRevision as citationSourceRestoreRevision
 } from '../citation'
 
 let requestInfo
@@ -314,6 +315,104 @@ describe('citation#fetch', () => {
   })
 })
 
+describe('restoreCitationRevision', () => {
+  const OLD_ENV = process.env
+
+  beforeEach(() => {
+    vi.resetAllMocks()
+
+    vi.restoreAllMocks()
+
+    process.env = { ...OLD_ENV }
+
+    process.env.cmrRootUrl = 'http://example-cmr.com'
+
+    // Default requestInfo
+    requestInfo = {
+      name: 'citationDelete',
+      alias: 'citationDelete',
+      args: {
+        conceptId: 'CIT100000-EDSC',
+        revisionId: '1'
+      },
+      fieldsByTypeName: {
+        CitationMutationResponse: {
+          conceptId: {
+            name: 'conceptId',
+            alias: 'conceptId',
+            args: {},
+            fieldsByTypeName: {}
+          },
+          revisionId: {
+            name: 'revisionId',
+            alias: 'revisionId',
+            args: {},
+            fieldsByTypeName: {}
+          }
+        }
+      }
+    }
+  })
+
+  afterEach(() => {
+    process.env = OLD_ENV
+  })
+
+  test('returns the CMR results', async () => {
+    nock(/example-cmr/)
+      .defaultReplyHeaders({
+        'CMR-Request-Id': 'abcd-1234-efgh-5678'
+      })
+      .get('/search/citations.umm_json?concept_id=CIT100000-EDSC&all_revisions=true')
+      .reply(200, {
+        items: [{
+          meta: {
+            'concept-id': 'CIT100000-EDSC',
+            'native-id': '1b9d6bcd-bbfd-4b2d-9b5d-ab8dfbbd4bed',
+            'revision-id': 1
+          },
+          umm: {
+            Name: 'Tortor Elit Fusce Quam Risus'
+          }
+        }, {
+          meta: {
+            'concept-id': 'CIT100000-EDSC',
+            'native-id': '1b9d6bcd-bbfd-4b2d-9b5d-ab8dfbbd4bed',
+            'revision-id': 2
+          },
+          umm: {
+            Name: 'Adipiscing Cras Etiam Venenatis'
+          }
+        }]
+      })
+
+    nock(/example-cmr/)
+      .defaultReplyHeaders({
+        'CMR-Request-Id': 'abcd-1234-efgh-5678'
+      })
+      .put('/ingest/providers/EDSC/citations/1b9d6bcd-bbfd-4b2d-9b5d-ab8dfbbd4bed')
+      .reply(200, {
+        'concept-id': 'CIT100000-EDSC',
+        'revision-id': '3'
+      })
+
+    const response = await citationSourceRestoreRevision({
+      revisionId: '1',
+      conceptId: 'CIT100000-EDSC'
+    }, {
+      headers: {
+        'Client-Id': 'eed-test-graphql',
+        'CMR-Request-Id': 'abcd-1234-efgh-5678'
+      }
+    }, requestInfo)
+
+    expect(response).toEqual({
+      conceptId: 'CIT100000-EDSC',
+      revisionId: '3'
+    })
+  })
+})
+
 describe('citation#delete', () => {
   const OLD_ENV = process.env
 

src/datasources/citation.js

@@ -1,7 +1,10 @@
 import citationKeyMap from '../utils/umm/citationKeyMap.json'
+import Citation from '../cmr/concepts/citation'
 
+import { cmrQuery } from '../utils/cmrQuery'
+import { findPreviousRevision } from '../utils/findPreviousRevision'
+import { getProviderFromConceptId } from '../utils/getProviderFromConceptId'
 import { parseRequestedFields } from '../utils/parseRequestedFields'
-import Citation from '../cmr/concepts/citation'
 
 export const fetchCitations = async (params, context, parsedInfo) => {
   const { headers } = context
@@ -20,6 +23,56 @@ export const fetchCitations = async (params, context, parsedInfo) => {
   return citation.getFormattedResponse()
 }
 
+export const restoreCitationRevision = async (args, context, parsedInfo) => {
+  const { headers } = context
+
+  const requestInfo = parseRequestedFields(parsedInfo, citationKeyMap, 'citation')
+
+  const {
+    ingestKeys
+  } = requestInfo
+
+  const citation = new Citation(headers, requestInfo, args)
+
+  const {
+    conceptId,
+    revisionId
+  } = args
+
+  const previousRevisions = await cmrQuery({
+    conceptType: 'citations',
+    options: {
+      format: 'umm_json'
+    },
+    params: {
+      conceptId,
+      allRevisions: true
+    }
+  })
+
+  const { data: responseData } = previousRevisions
+
+  // Find the requested revision for the provided concept
+  const previousRevision = findPreviousRevision(responseData, revisionId)
+
+  const { meta, umm } = previousRevision
+
+  const { 'native-id': nativeId } = meta
+
+  // Query CMR
+  citation.ingest({
+    nativeId,
+    providerId: getProviderFromConceptId(conceptId),
+    ...umm
+  }, ingestKeys, headers)
+
+  // Parse the response from CMR
+  await citation.parseIngest(requestInfo)
+
+  // Return a formatted JSON response
+  return citation.getFormattedIngestResponse()
+}
+
 export const deleteCitation = async (args, context, parsedInfo) => {
   const { headers } = context
 

src/graphql/handler.js

@@ -55,7 +55,8 @@ import {
 
 import {
   deleteCitation as citationSourceDelete,
-  fetchCitations as citationSourceFetch
+  fetchCitations as citationSourceFetch,
+  restoreCitationRevision as citationSourceRestoreRevision
 } from '../datasources/citation'
 
 import {
@@ -235,6 +236,7 @@ export default startServerAndCreateLambdaHandler(
           associationSourceDelete,
           citationSourceDelete,
           citationSourceFetch,
+          citationSourceRestoreRevision,
           collectionDraftProposalSource,
           collectionDraftSource,
           collectionSourceDelete,

src/resolvers/__tests__/__mocks__/mockServer.js

@@ -38,7 +38,8 @@ import {
 
 import {
   deleteCitation as citationSourceDelete,
-  fetchCitations as citationSourceFetch
+  fetchCitations as citationSourceFetch,
+  restoreCitationRevision as citationSourceRestoreRevision
 } from '../../../datasources/citation'
 
 import {
@@ -108,6 +109,7 @@ export const buildContextValue = (extraContext) => ({
     associationSourceDelete,
     citationSourceDelete,
     citationSourceFetch,
+    citationSourceRestoreRevision,
     collectionDraftProposalSource,
     collectionDraftSource,
     collectionSourceDelete,