mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-07-30 23:42:01 +08:00
cffa511cf3
* feat(user): support sso and api key * feat(user): remove ee references from pkg * feat(user): remove ee references from pkg * feat(user): related client changes * feat(user): remove the sso available check * feat(user): fix go tests * feat(user): move the middleware from ee to pkg * feat(user): some more error code cleanup * feat(user): some more error code cleanup * feat(user): skip flaky UI tests * feat(user): some more error code cleanup
93 lines
3.2 KiB
Go
93 lines
3.2 KiB
Go
package integrations
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/SigNoz/signoz/pkg/emailing"
|
|
"github.com/SigNoz/signoz/pkg/emailing/noopemailing"
|
|
"github.com/SigNoz/signoz/pkg/instrumentation/instrumentationtest"
|
|
"github.com/SigNoz/signoz/pkg/modules/organization/implorganization"
|
|
"github.com/SigNoz/signoz/pkg/modules/user/impluser"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestIntegrationLifecycle(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
mgr, store := NewTestIntegrationsManager(t)
|
|
ctx := context.Background()
|
|
|
|
organizationModule := implorganization.NewModule(implorganization.NewStore(store))
|
|
providerSettings := instrumentationtest.New().ToProviderSettings()
|
|
emailing, _ := noopemailing.New(context.Background(), providerSettings, emailing.Config{})
|
|
userModule := impluser.NewModule(impluser.NewStore(store, providerSettings), nil, emailing, providerSettings)
|
|
user, apiErr := createTestUser(organizationModule, userModule)
|
|
if apiErr != nil {
|
|
t.Fatalf("could not create test user: %v", apiErr)
|
|
}
|
|
|
|
ii := true
|
|
installedIntegrationsFilter := &IntegrationsFilter{
|
|
IsInstalled: &ii,
|
|
}
|
|
|
|
installedIntegrations, apiErr := mgr.ListIntegrations(
|
|
ctx, user.OrgID, installedIntegrationsFilter,
|
|
)
|
|
require.Nil(apiErr)
|
|
require.Equal([]IntegrationsListItem{}, installedIntegrations)
|
|
|
|
availableIntegrations, apiErr := mgr.ListIntegrations(ctx, user.OrgID, nil)
|
|
require.Nil(apiErr)
|
|
require.Equal(2, len(availableIntegrations))
|
|
require.False(availableIntegrations[0].IsInstalled)
|
|
require.False(availableIntegrations[1].IsInstalled)
|
|
|
|
testIntegrationConfig := map[string]interface{}{}
|
|
installed, apiErr := mgr.InstallIntegration(
|
|
ctx, user.OrgID, availableIntegrations[1].Id, testIntegrationConfig,
|
|
)
|
|
require.Nil(apiErr)
|
|
require.Equal(installed.Id, availableIntegrations[1].Id)
|
|
|
|
integration, apiErr := mgr.GetIntegration(ctx, user.OrgID, availableIntegrations[1].Id)
|
|
require.Nil(apiErr)
|
|
require.Equal(integration.Id, availableIntegrations[1].Id)
|
|
require.NotNil(integration.Installation)
|
|
|
|
installedIntegrations, apiErr = mgr.ListIntegrations(
|
|
ctx, user.OrgID, installedIntegrationsFilter,
|
|
)
|
|
require.Nil(apiErr)
|
|
require.Equal(1, len(installedIntegrations))
|
|
require.Equal(availableIntegrations[1].Id, installedIntegrations[0].Id)
|
|
|
|
availableIntegrations, apiErr = mgr.ListIntegrations(ctx, user.OrgID, nil)
|
|
require.Nil(apiErr)
|
|
require.Equal(2, len(availableIntegrations))
|
|
require.False(availableIntegrations[0].IsInstalled)
|
|
require.True(availableIntegrations[1].IsInstalled)
|
|
|
|
apiErr = mgr.UninstallIntegration(ctx, user.OrgID, installed.Id)
|
|
require.Nil(apiErr)
|
|
|
|
integration, apiErr = mgr.GetIntegration(ctx, user.OrgID, availableIntegrations[1].Id)
|
|
require.Nil(apiErr)
|
|
require.Equal(integration.Id, availableIntegrations[1].Id)
|
|
require.Nil(integration.Installation)
|
|
|
|
installedIntegrations, apiErr = mgr.ListIntegrations(
|
|
ctx, user.OrgID, installedIntegrationsFilter,
|
|
)
|
|
require.Nil(apiErr)
|
|
require.Equal(0, len(installedIntegrations))
|
|
|
|
availableIntegrations, apiErr = mgr.ListIntegrations(ctx, user.OrgID, nil)
|
|
require.Nil(apiErr)
|
|
require.Equal(2, len(availableIntegrations))
|
|
require.False(availableIntegrations[0].IsInstalled)
|
|
require.False(availableIntegrations[1].IsInstalled)
|
|
}
|