implement and test:

- deployments - deployment - createDeployment
dmlond · Feb 28, 2020 · b161f3e · b161f3e
1 parent b578ae1
commit b161f3e
Show file tree

Hide file tree

Showing 2 changed files with 84 additions and 0 deletions.
diff --git a/src/__tests__/helpers/projectServiceClient.spec.js b/src/__tests__/helpers/projectServiceClient.spec.js
@@ -337,4 +337,49 @@ describe('projectServiceClient', () => {
       testProjectServiceApi(expectedProjectServicePath, expectedProjectServiceSendMethod, subject, expectedExtraPayload, expectedModelId, environmentPayload);        
     });
 
+    describe('.deployments', () => {
+      let expectedModelId = "abc-123-xyz";
+      let expectedProjectServicePath = '/api/v1/ai_models/'+expectedModelId+'/deployments';
+      let expectedProjectServiceSendMethod = 'get'
+
+      function subject(mid,s,f) {
+          projectServiceClient.deployments(mid,s,f);
+      }
+
+      testProjectServiceApi(expectedProjectServicePath, expectedProjectServiceSendMethod, subject, {}, expectedModelId);
+    });
+
+    describe('.createDeployment', () => {
+      let expectedModelId = "abc-123-xyz";
+      let expectedProjectServicePath = '/api/v1/ai_models/'+expectedModelId+'/deployments';
+      let expectedProjectServiceSendMethod = 'post';
+      let newDeploymentPayload = {
+        commit_sha: 'abc123xyz',
+        image: 'docker.io/image',
+        entrypoint: 'python3'
+      };
+      let expectedExtraPayload = {
+        data: {
+          deployment: newDeploymentPayload
+        }
+      };
+      function subject(mid, p, s,f) {
+          projectServiceClient.createDeployment(mid, p, s,f);
+      }
+
+      testProjectServiceApi(expectedProjectServicePath, expectedProjectServiceSendMethod, subject, expectedExtraPayload, expectedModelId, newDeploymentPayload);        
+    });
+
+    describe('.deployment', () => {
+      let expectedModelId = "abc-123-xyz";
+      let expectedDeploymentId = 'xyz-abc-123';
+      let expectedProjectServicePath = '/api/v1/ai_models/'+expectedModelId+'/deployments/'+expectedDeploymentId;
+      let expectedProjectServiceSendMethod = 'get'
+
+      function subject(mid, did, s,f) {
+          projectServiceClient.deployment(mid, did, s,f);
+      }
+
+      testProjectServiceApi(expectedProjectServicePath, expectedProjectServiceSendMethod, subject, {}, expectedModelId, expectedDeploymentId);
+    });
 });
diff --git a/src/helpers/projectServiceClient.js b/src/helpers/projectServiceClient.js
@@ -192,5 +192,44 @@ const projectServiceClient = {
             failureHandler
         ) 
     },
+
+    deployments(modelId, successHandler, failureHandler) {
+        var deploymentIndex = `${config.proxy_uri}/api/v1/ai_models/${modelId}/deployments`;
+        this.send(
+            {
+                url: deploymentIndex,
+                method: 'get'
+            },
+            successHandler,
+            failureHandler
+        )            
+    },
+
+    deployment(modelId, deploymentId, successHandler, failureHandler) {
+        var deploymentProfile = `${config.proxy_uri}/api/v1/ai_models/${modelId}/deployments/${deploymentId}`;
+        this.send(
+            {
+                url: deploymentProfile,
+                method: 'get'
+            },
+            successHandler,
+            failureHandler
+        )
+    },
+
+    createDeployment(modelId, payload, successHandler, failureHandler) {
+        var deploymentIndex = `${config.proxy_uri}/api/v1/ai_models/${modelId}/deployments`;
+        this.send(
+            {
+                url: deploymentIndex,
+                method: 'post',
+                data: {
+                    deployment: payload
+                }
+            },
+            successHandler,
+            failureHandler
+        ) 
+    },
 };
 export default projectServiceClient;