Add the new methods for integration endpoints

novuhq · Oct 8, 2023 · f60f6d4 · f60f6d4
1 parent 1449de3
commit f60f6d4
Show file tree

Hide file tree

Showing 5 changed files with 164 additions and 1 deletion.
diff --git a/lib/integration.go b/lib/integration.go
@@ -14,6 +14,8 @@ type IIntegration interface {
 	GetWebhookSupportStatus(ctx context.Context, providerId string) (bool, error)
 	Update(ctx context.Context, integrationId string, request UpdateIntegrationRequest) (*IntegrationResponse, error)
 	Delete(ctx context.Context, integrationId string) (*IntegrationResponse, error)
+	SetIntegrationAsPrimary(ctx context.Context, integrationId string) (*SetIntegrationAsPrimaryResponse, error)
+	GetChannelLimit(ctx context.Context, channelType string) (*IntegrationChannelLimitResponse, error)
 }
 
 type IntegrationService service
@@ -148,3 +150,39 @@ func (i IntegrationService) Delete(ctx context.Context, integrationId string) (*
 
 	return &response, nil
 }
+
+func (i IntegrationService) SetIntegrationAsPrimary(ctx context.Context, integrationId string) (*SetIntegrationAsPrimaryResponse, error) {
+	var response SetIntegrationAsPrimaryResponse
+
+	URL := i.client.config.BackendURL.JoinPath("integrations", integrationId, "set-primary")
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodPost, URL.String(), http.NoBody)
+	if err != nil {
+		return nil, err
+	}
+
+	_, err = i.client.sendRequest(req, &response)
+	if err != nil {
+		return nil, err
+	}
+
+	return &response, nil
+}
+
+func (i IntegrationService) GetChannelLimit(ctx context.Context, channelType string) (*IntegrationChannelLimitResponse, error) {
+	var response IntegrationChannelLimitResponse
+
+	URL := i.client.config.BackendURL.JoinPath("integrations", channelType, "limit")
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodGet, URL.String(), http.NoBody)
+	if err != nil {
+		return nil, err
+	}
+
+	_, err = i.client.sendRequest(req, &response)
+	if err != nil {
+		return nil, err
+	}
+
+	return &response, nil
+}
diff --git a/lib/integration_test.go b/lib/integration_test.go
@@ -254,3 +254,55 @@ func TestDeleteActiveIntegration_Success(t *testing.T) {
 
 	require.NoError(t, err)
 }
+
+func TestSetIntegrationAsPrimary_Success(t *testing.T) {
+	const integrationId = "IntegrationId"
+
+	var response *lib.SetIntegrationAsPrimaryResponse
+	fileToStruct(filepath.Join("../testdata", "set_integration_as_primary_response.json"), &response)
+
+	httpServer := IntegrationTestServer(t, IntegrationServerOptions[interface{}]{
+		ExpectedRequest: IntegrationRequestDetails[interface{}]{
+			Url:    fmt.Sprintf("/v1/integrations/%s/set-primary", integrationId),
+			Method: http.MethodPost,
+		},
+		ExpectedResponse: IntegrationResponseDetails{
+			StatusCode: http.StatusOK,
+			Body:       response,
+		},
+	})
+
+	ctx := context.Background()
+	novuClient := lib.NewAPIClient(novuApiKey, &lib.Config{BackendURL: lib.MustParseURL(httpServer.URL)})
+
+	res, err := novuClient.IntegrationsApi.SetIntegrationAsPrimary(ctx, integrationId)
+
+	assert.Equal(t, response, res)
+	require.NoError(t, err)
+}
+
+func TestGetChannelLimit_Success(t *testing.T) {
+	const channelType = "ChannelType"
+
+	var response *lib.IntegrationChannelLimitResponse
+	fileToStruct(filepath.Join("../testdata", "integration_channel_limit_response.json"), &response)
+
+	httpServer := IntegrationTestServer(t, IntegrationServerOptions[interface{}]{
+		ExpectedRequest: IntegrationRequestDetails[interface{}]{
+			Url:    fmt.Sprintf("/v1/integrations/%s/limit", channelType),
+			Method: http.MethodGet,
+		},
+		ExpectedResponse: IntegrationResponseDetails{
+			StatusCode: http.StatusOK,
+			Body:       response,
+		},
+	})
+
+	ctx := context.Background()
+	novuClient := lib.NewAPIClient(novuApiKey, &lib.Config{BackendURL: lib.MustParseURL(httpServer.URL)})
+
+	res, err := novuClient.IntegrationsApi.GetChannelLimit(ctx, channelType)
+
+	assert.Equal(t, response, res)
+	require.NoError(t, err)
+}
diff --git a/lib/model.go b/lib/model.go
@@ -382,6 +382,30 @@ type GetIntegrationsResponse struct {
 	Data []Integration `json:"data"`
 }
 
+type IntegrationChannelLimitResponse struct {
+	Data struct {
+		Limit int `json:"limit"`
+		Count int `json:"count"`
+	} `json:"data"`
+}
+
+type SetIntegrationAsPrimaryResponse struct {
+	Data struct {
+		ID             string                 `json:"_id"`
+		EnvironmentID  string                 `json:"_environmentId"`
+		OrganizationID string                 `json:"_organizationId"`
+		Name           string                 `json:"name"`
+		Identifier     string                 `json:"identifier"`
+		ProviderID     string                 `json:"providerId"`
+		Channel        string                 `json:"channel"`
+		Credentials    IntegrationCredentials `json:"credentials"`
+		Active         bool                   `json:"active"`
+		Deleted        bool                   `json:"deleted"`
+		DeletedAt      string                 `json:"deletedAt"`
+		DeletedBy      string                 `json:"deletedBy"`
+		Primary        bool                   `json:"primary"`
+	} `json:"data"`
+}
 type BulkTriggerOptions struct {
 	Name          interface{} `json:"name,omitempty"`
 	To            interface{} `json:"to,omitempty"`
@@ -437,5 +461,5 @@ type BlueprintByTemplateIdResponse struct {
 
 type BlueprintGroupByCategoryResponse struct {
 	General []interface{} `json:"general,omitempty"`
-	Popular interface{} `json:"popular,omitempty"`
+	Popular interface{}   `json:"popular,omitempty"`
 }
diff --git a/testdata/integration_channel_limit_response.json b/testdata/integration_channel_limit_response.json
@@ -0,0 +1,6 @@
+{
+	"data": {
+		"limit": 0,
+		"count": 0
+	}
+}
diff --git a/testdata/set_integration_as_primary_response.json b/testdata/set_integration_as_primary_response.json
@@ -0,0 +1,43 @@
+{
+	"data": {
+		"_id": "string",
+		"_environmentId": "string",
+		"_organizationId": "string",
+		"name": "string",
+		"identifier": "string",
+		"providerId": "string",
+		"channel": "in_app",
+		"credentials": {
+			"apiKey": "string",
+			"user": "string",
+			"secretKey": "string",
+			"domain": "string",
+			"password": "string",
+			"host": "string",
+			"port": "string",
+			"secure": true,
+			"region": "string",
+			"accountSid": "string",
+			"messageProfileId": "string",
+			"token": "string",
+			"from": "string",
+			"senderName": "string",
+			"projectName": "string",
+			"applicationId": "string",
+			"clientId": "string",
+			"requireTls": true,
+			"ignoreTls": true,
+			"baseUrl": "string",
+			"webhookUrl": "string",
+			"redirectUrl": "string",
+			"hmac": true,
+			"serviceAccount": "string",
+			"ipPoolName": "string"
+		},
+		"active": true,
+		"deleted": true,
+		"deletedAt": "string",
+		"deletedBy": "string",
+		"primary": true
+	}
+}