Feat: workflow trigger new resources (#623)

client/api_client.go

@@ -116,6 +116,8 @@ type ApiClientInterface interface {
 	CustomFlowAssign(assignments []CustomFlowAssignment) error
 	CustomFlowUnassign(assignments []CustomFlowAssignment) error
 	CustomFlowGetAssignments(assignments []CustomFlowAssignment) ([]CustomFlowAssignment, error)
+	SubscribeWorkflowTrigger(environmentId string, payload WorkflowTriggerEnvironments) error
+	UnsubscribeWorkflowTrigger(environmentId string, payload WorkflowTriggerEnvironments) error
 }
 
 func NewApiClient(client http.HttpClientInterface, defaultOrganizationId string) ApiClientInterface {

client/workflow_triggers.go

@@ -8,9 +8,13 @@ type WorkflowTriggerUpsertPayload struct {
 	DownstreamEnvironmentIds []string `json:"downstreamEnvironmentIds"`
 }
 
+type WorkflowTriggerEnvironments struct {
+	DownstreamEnvironmentIds []string `json:"downstreamEnvironmentIds"`
+}
+
 func (client *ApiClient) WorkflowTrigger(environmentId string) ([]WorkflowTrigger, error) {
 	var result []WorkflowTrigger
-	err := client.http.Get("environments/"+environmentId+"/downstream", nil, &result)
+	err := client.http.Get("/environments/"+environmentId+"/downstream", nil, &result)
 	if err != nil {
 		return []WorkflowTrigger{}, err
 	}
@@ -27,3 +31,11 @@ func (client *ApiClient) WorkflowTriggerUpsert(environmentId string, request Wor
 	}
 	return result, nil
 }
+
+func (client *ApiClient) SubscribeWorkflowTrigger(environmentId string, payload WorkflowTriggerEnvironments) error {
+	return client.http.Post("/environments/"+environmentId+"/downstream/subscribe", payload, nil)
+}
+
+func (client *ApiClient) UnsubscribeWorkflowTrigger(environmentId string, payload WorkflowTriggerEnvironments) error {
+	return client.http.Post("/environments/"+environmentId+"/downstream/unsubscribe", payload, nil)
+}

client/workflow_triggers_test.go

@@ -41,7 +41,7 @@ var _ = Describe("Workflow Triggers", func() {
 	Describe("WorkflowTrigger", func() {
 		BeforeEach(func() {
 			httpCall = mockHttpClient.EXPECT().
-				Get("environments/"+environmentId+"/downstream", nil, gomock.Any()).
+				Get("/environments/"+environmentId+"/downstream", nil, gomock.Any()).
 				Do(func(path string, request interface{}, response *[]WorkflowTrigger) {
 					*response = mockTrigger
 				})
@@ -52,4 +52,40 @@ var _ = Describe("Workflow Triggers", func() {
 			Expect(triggers).To(Equal(mockTrigger))
 		})
 	})
+
+	Describe("SubscribeWorkflowTrigger", func() {
+		var err error
+
+		subscribePayload := WorkflowTriggerEnvironments{
+			DownstreamEnvironmentIds: []string{"1", "2"},
+		}
+
+		BeforeEach(func() {
+			httpCall = mockHttpClient.EXPECT().Post("/environments/"+environmentId+"/downstream/subscribe", subscribePayload, nil)
+			httpCall.Times(1)
+			err = apiClient.SubscribeWorkflowTrigger(environmentId, subscribePayload)
+		})
+
+		It("Should not return an error", func() {
+			Expect(err).To(BeNil())
+		})
+	})
+
+	Describe("UnsubscribeWorkflowTrigger", func() {
+		var err error
+
+		unsubscribePayload := WorkflowTriggerEnvironments{
+			DownstreamEnvironmentIds: []string{"1", "2"},
+		}
+
+		BeforeEach(func() {
+			httpCall = mockHttpClient.EXPECT().Post("/environments/"+environmentId+"/downstream/unsubscribe", unsubscribePayload, nil)
+			httpCall.Times(1)
+			err = apiClient.UnsubscribeWorkflowTrigger(environmentId, unsubscribePayload)
+		})
+
+		It("Should not return an error", func() {
+			Expect(err).To(BeNil())
+		})
+	})
 })