renamed applySchedule to setSchedule to be more consistent with singu…

…lar resource endpont usage implement and set environment and setEnvironment
dmlond · Feb 28, 2020 · b578ae1 · b578ae1
1 parent 32cb5f6
commit b578ae1
Show file tree

Hide file tree

Showing 2 changed files with 65 additions and 3 deletions.
diff --git a/src/__tests__/helpers/projectServiceClient.spec.js b/src/__tests__/helpers/projectServiceClient.spec.js
@@ -284,7 +284,7 @@ describe('projectServiceClient', () => {
       testProjectServiceApi(expectedProjectServicePath, expectedProjectServiceSendMethod, subject, {}, expectedModelId);
     });
 
-    describe('.applySchedule', () => {
+    describe('.setSchedule', () => {
       let expectedModelId = "abc-123-xyz";
       let expectedProjectServicePath = '/api/v1/ai_models'+"/"+expectedModelId+'/schedule';
       let expectedProjectServiceSendMethod = 'post';
@@ -297,9 +297,44 @@ describe('projectServiceClient', () => {
         }
       };
       function subject(mid, p, s,f) {
-          projectServiceClient.applySchedule(mid, p, s,f);
+          projectServiceClient.setSchedule(mid, p, s,f);
       }
 
       testProjectServiceApi(expectedProjectServicePath, expectedProjectServiceSendMethod, subject, expectedExtraPayload, expectedModelId, scedulePayload);        
     });
+
+    describe('.environment', () => {
+      let expectedModelId = "abc-123-xyz";
+      let expectedProjectServicePath = '/api/v1/ai_models'+"/"+expectedModelId+'/environment';
+      let expectedProjectServiceSendMethod = 'get'
+
+      function subject(mid,s,f) {
+          projectServiceClient.environment(mid,s,f);
+      }
+
+      testProjectServiceApi(expectedProjectServicePath, expectedProjectServiceSendMethod, subject, {}, expectedModelId);
+    });
+
+    describe('.setEnvironment', () => {
+      let expectedModelId = "abc-123-xyz";
+      let expectedProjectServicePath = '/api/v1/ai_models'+"/"+expectedModelId+'/environment';
+      let expectedProjectServiceSendMethod = 'post';
+      let environmentPayload = {
+        variables:{
+          PIPELINE_USER: "theUserz",
+          PIPELINE_PASSWORD: "a1bn23"
+        }
+      };
+      let expectedExtraPayload = {
+        data: {
+          environment: environmentPayload
+        }
+      };
+      function subject(mid, p, s,f) {
+          projectServiceClient.setEnvironment(mid, p, s,f);
+      }
+
+      testProjectServiceApi(expectedProjectServicePath, expectedProjectServiceSendMethod, subject, expectedExtraPayload, expectedModelId, environmentPayload);        
+    });
+
 });
diff --git a/src/helpers/projectServiceClient.js b/src/helpers/projectServiceClient.js
@@ -151,7 +151,7 @@ const projectServiceClient = {
         )
     },
 
-    applySchedule(modelId, payload, successHandler, failureHandler) {
+    setSchedule(modelId, payload, successHandler, failureHandler) {
         var scheduleUrl = `${config.proxy_uri}/api/v1/ai_models/${modelId}/schedule`;
         this.send(
             {
@@ -165,5 +165,32 @@ const projectServiceClient = {
             failureHandler
         ) 
     },
+
+    environment(modelId, successHandler, failureHandler) {
+        var environmentUrl = `${config.proxy_uri}/api/v1/ai_models/${modelId}/environment`;
+        this.send(
+            {
+                url: environmentUrl,
+                method: 'get'
+            },
+            successHandler,
+            failureHandler
+        )
+    },
+
+    setEnvironment(modelId, payload, successHandler, failureHandler) {
+        var environmentUrl = `${config.proxy_uri}/api/v1/ai_models/${modelId}/environment`;
+        this.send(
+            {
+                url: environmentUrl,
+                method: 'post',
+                data: {
+                    environment: payload
+                }
+            },
+            successHandler,
+            failureHandler
+        ) 
+    },
 };
 export default projectServiceClient;