feature: add mechanism to pass callback to @kfarranger-server

src/index.js

@@ -10,11 +10,15 @@ import { port, egoURL, projectId, esHost } from './env';
 import { version, dependencies } from '../package.json';
 import { shortUrlStatic, statistics, survival, searchByIds } from './endpoints';
 import { onlyAdminMutations, injectBodyHttpHeaders } from './middleware';
+import { postProcessSaveSet } from './utils/saveSet';
+import SQS from 'aws-sdk/clients/sqs';
 
 const app = express();
 const http = Server(app);
 const io = socketIO(http);
 
+const sqs = new SQS({ apiVersion: '2012-11-05' });
+
 app.use(cors());
 
 /*
@@ -52,7 +56,13 @@ app.use(
       },
       {
         type: 'allow',
-        route: [`/(.*)/graphql`, `/(.*)/graphql/(.*)`, `/(.*)/download`, `/survival`, '/searchByIds'],
+        route: [
+          `/(.*)/graphql`,
+          `/(.*)/graphql/(.*)`,
+          `/(.*)/download`,
+          `/survival`,
+          '/searchByIds',
+        ],
         status: ['approved'],
         role: 'user',
       },
@@ -82,6 +92,11 @@ Arranger({
     context: ({ jwt }) => ({ jwt }),
     middleware: [onlyAdminMutations],
   },
+  callbacks: {
+    resolvers: {
+      saveSet: postProcessSaveSet(sqs),
+    },
+  },
 }).then(router => {
   app.use(router);
   http.listen(port, async () => {

src/utils/saveSet.js

@@ -0,0 +1,6 @@
+import { sqsQueueUrl } from '../env';
+
+export const postProcessSaveSet = (sqs) => async data => {
+  const params = { MessageBody: JSON.stringify(data), QueueUrl: sqsQueueUrl };
+  return sqs.sendMessage(params).promise();
+};

src/utils/saveSet.test.js

@@ -0,0 +1,57 @@
+import AWSMock from 'aws-sdk-mock';
+import AWS from 'aws-sdk';
+import { postProcessSaveSet } from './saveSet';
+import SQS from 'aws-sdk/clients/sqs';
+import { describe, it, beforeEach, afterEach } from 'mocha';
+import { expect } from 'chai';
+
+const mockSaveSetData = {
+  setId: '123xyyz',
+  createdAt: Date.now(),
+  ids: ['id1', '1d2'],
+  size: 2,
+  sqon: {
+    op: 'and',
+    content: [
+      {
+        op: 'in',
+        content: {
+          field: 'observed_phenotype.name',
+          value: ['Abnormality of the cardiovascular system (HP:0001626)'],
+        },
+      },
+    ],
+  },
+  type: 'participant',
+  userId: 'd1c3f78f-a68f-46c4-a44f-9d1249d6337e',
+  path: 'kf_id',
+  sort: [],
+  tag: 'test',
+};
+
+describe('Test case for SQS SendMessage', () => {
+  beforeEach(() => {
+    AWSMock.setSDKInstance(AWS);
+  });
+
+  afterEach(() => {
+    AWSMock.restore();
+  });
+
+  it('should return a successful response', async () => {
+    AWSMock.mock('SQS', 'sendMessage', () => Promise.resolve({ MessageId: '123' }));
+    const sqs = new AWS.SQS({ apiVersion: '2012-11-05' });
+    const res = await postProcessSaveSet(sqs)(mockSaveSetData);
+    expect(res.MessageId).to.equal('123');
+  });
+
+  it('should throw when the queue does not respond', async () => {
+    AWSMock.mock('SQS', 'sendMessage', () => Promise.reject('could not send data to queue'));
+    const sqs = new AWS.SQS({ apiVersion: '2012-11-05' });
+    try {
+      await postProcessSaveSet(sqs)(mockSaveSetData);
+    } catch (err) {
+      expect(err).to.eql('could not send data to queue');
+    }
+  });
+});