grafana/pkg/services/ssosettings/ssosettingsimpl/service_test.go

package ssosettingsimpl

import (
	"context"
	"fmt"
	"testing"

	"github.com/grafana/grafana/pkg/infra/log"
	"github.com/grafana/grafana/pkg/services/accesscontrol"
	"github.com/grafana/grafana/pkg/services/accesscontrol/acimpl"
	"github.com/grafana/grafana/pkg/services/auth/identity"
	"github.com/grafana/grafana/pkg/services/ssosettings"
	"github.com/grafana/grafana/pkg/services/ssosettings/models"
	"github.com/grafana/grafana/pkg/services/ssosettings/ssosettingstests"
	"github.com/grafana/grafana/pkg/services/user"
	"github.com/grafana/grafana/pkg/setting"
	"github.com/stretchr/testify/require"
)

func TestSSOSettingsService_GetForProvider(t *testing.T) {
	testCases := []struct {
		name    string
		setup   func(env testEnv)
		want    *models.SSOSetting
		wantErr bool
	}{
		{
			name: "should return successfully",
			setup: func(env testEnv) {
				env.store.ExpectedSSOSetting = &models.SSOSetting{
					Provider: "github",
					Settings: map[string]interface{}{
						"enabled": true,
					},
					Source: models.DB,
				}
			},
			want: &models.SSOSetting{
				Provider: "github",
				Settings: map[string]interface{}{
					"enabled": true,
				},
			},
			wantErr: false,
		},
		{
			name:    "should return error if store returns an error different than not found",
			setup:   func(env testEnv) { env.store.ExpectedError = fmt.Errorf("error") },
			want:    nil,
			wantErr: true,
		},
		{
			name: "should fallback to strategy if store returns not found",
			setup: func(env testEnv) {
				env.store.ExpectedError = ssosettings.ErrNotFound
				env.fallbackStrategy.ExpectedIsMatch = true
				env.fallbackStrategy.ExpectedConfig = map[string]interface{}{
					"enabled": true,
				}
			},
			want: &models.SSOSetting{
				Provider: "github",
				Settings: map[string]interface{}{
					"enabled": true,
				},
				Source: models.System,
			},
			wantErr: false,
		},
		{
			name: "should return error if the fallback strategy was not found",
			setup: func(env testEnv) {
				env.store.ExpectedError = ssosettings.ErrNotFound
				env.fallbackStrategy.ExpectedIsMatch = false
			},
			want:    nil,
			wantErr: true,
		},
		{
			name: "should return error if fallback strategy returns error",
			setup: func(env testEnv) {
				env.store.ExpectedError = ssosettings.ErrNotFound
				env.fallbackStrategy.ExpectedIsMatch = true
				env.fallbackStrategy.ExpectedError = fmt.Errorf("error")
			},
			want:    nil,
			wantErr: true,
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			env := setupTestEnv(t)
			if tc.setup != nil {
				tc.setup(env)
			}

			actual, err := env.service.GetForProvider(context.Background(), "github")

			if tc.wantErr {
				require.Error(t, err)
				return
			}

			require.NoError(t, err)
			require.Equal(t, tc.want, actual)
		})
	}
}

func TestSSOSettingsService_List(t *testing.T) {
	defaultIdentity := &user.SignedInUser{
		UserID: 1,
		OrgID:  1,
		Permissions: map[int64]map[string][]string{
			1: {
				accesscontrol.ActionSettingsRead: {accesscontrol.ScopeSettingsAll},
			},
		},
	}

	scopedIdentity := &user.SignedInUser{
		UserID: 1,
		OrgID:  1,
		Permissions: map[int64]map[string][]string{
			1: {
				accesscontrol.ActionSettingsRead: []string{
					accesscontrol.Scope("settings", "auth.azuread", "*"),
					accesscontrol.Scope("settings", "auth.github", "*"),
				},
			},
		},
	}
	testCases := []struct {
		name     string
		setup    func(env testEnv)
		identity identity.Requester
		want     []*models.SSOSetting
		wantErr  bool
	}{
		{
			name: "should return successfully",
			setup: func(env testEnv) {
				env.store.ExpectedSSOSettings = []*models.SSOSetting{
					{
						Provider: "github",
						Settings: map[string]interface{}{
							"enabled": true,
						},
						Source: models.DB,
					},
					{
						Provider: "okta",
						Settings: map[string]interface{}{
							"enabled": false,
						},
						Source: models.DB,
					},
				}
				env.fallbackStrategy.ExpectedIsMatch = true
				env.fallbackStrategy.ExpectedConfig = map[string]interface{}{
					"enabled": false,
				}
			},
			identity: defaultIdentity,
			want: []*models.SSOSetting{
				{
					Provider: "github",
					Settings: map[string]interface{}{
						"enabled": true,
					},
					Source: models.DB,
				},
				{
					Provider: "okta",
					Settings: map[string]interface{}{
						"enabled": false,
					},
					Source: models.DB,
				},
				{
					Provider: "gitlab",
					Settings: map[string]interface{}{
						"enabled": false,
					},
					Source: models.System,
				},
				{
					Provider: "generic_oauth",
					Settings: map[string]interface{}{
						"enabled": false,
					},
					Source: models.System,
				},
				{
					Provider: "google",
					Settings: map[string]interface{}{
						"enabled": false,
					},
					Source: models.System,
				},
				{
					Provider: "azuread",
					Settings: map[string]interface{}{
						"enabled": false,
					},
					Source: models.System,
				},
				{
					Provider: "grafana_com",
					Settings: map[string]interface{}{
						"enabled": false,
					},
					Source: models.System,
				},
			},
			wantErr: false,
		},
		{
			name: "should return the settings that the user has access to",
			setup: func(env testEnv) {
				env.store.ExpectedSSOSettings = []*models.SSOSetting{
					{
						Provider: "github",
						Settings: map[string]interface{}{
							"enabled": true,
						},
						Source: models.DB,
					},
					{
						Provider: "okta",
						Settings: map[string]interface{}{
							"enabled": false,
						},
						Source: models.DB,
					},
				}
				env.fallbackStrategy.ExpectedIsMatch = true
				env.fallbackStrategy.ExpectedConfig = map[string]interface{}{
					"enabled": false,
				}
			},
			identity: scopedIdentity,
			want: []*models.SSOSetting{
				{
					Provider: "github",
					Settings: map[string]interface{}{
						"enabled": true,
					},
					Source: models.DB,
				},
				{
					Provider: "azuread",
					Settings: map[string]interface{}{
						"enabled": false,
					},
					Source: models.System,
				},
			},
			wantErr: false,
		},
		{
			name:     "should return error if store returns an error",
			setup:    func(env testEnv) { env.store.ExpectedError = fmt.Errorf("error") },
			identity: defaultIdentity,
			want:     nil,
			wantErr:  true,
		},
		{
			name: "should use the fallback strategy if store returns empty list",
			setup: func(env testEnv) {
				env.store.ExpectedSSOSettings = []*models.SSOSetting{}
				env.fallbackStrategy.ExpectedIsMatch = true
				env.fallbackStrategy.ExpectedConfig = map[string]interface{}{
					"enabled": false,
				}
			},
			identity: defaultIdentity,
			want: []*models.SSOSetting{
				{
					Provider: "github",
					Settings: map[string]interface{}{
						"enabled": false,
					},
					Source: models.System,
				},
				{
					Provider: "okta",
					Settings: map[string]interface{}{
						"enabled": false,
					},
					Source: models.System,
				},
				{
					Provider: "gitlab",
					Settings: map[string]interface{}{
						"enabled": false,
					},
					Source: models.System,
				},
				{
					Provider: "generic_oauth",
					Settings: map[string]interface{}{
						"enabled": false,
					},
					Source: models.System,
				},
				{
					Provider: "google",
					Settings: map[string]interface{}{
						"enabled": false,
					},
					Source: models.System,
				},
				{
					Provider: "azuread",
					Settings: map[string]interface{}{
						"enabled": false,
					},
					Source: models.System,
				},
				{
					Provider: "grafana_com",
					Settings: map[string]interface{}{
						"enabled": false,
					},
					Source: models.System,
				},
			},
			wantErr: false,
		},
		{
			name: "should return error if any of the fallback strategies was not found",
			setup: func(env testEnv) {
				env.store.ExpectedSSOSettings = []*models.SSOSetting{}
				env.fallbackStrategy.ExpectedIsMatch = false
			},
			identity: defaultIdentity,
			want:     nil,
			wantErr:  true,
		},
	}
	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			env := setupTestEnv(t)
			if tc.setup != nil {
				tc.setup(env)
			}

			actual, err := env.service.List(context.Background(), tc.identity)

			if tc.wantErr {
				require.Error(t, err)
				return
			}

			require.NoError(t, err)
			require.ElementsMatch(t, tc.want, actual)
		})
	}
}

func setupTestEnv(t *testing.T) testEnv {
	store := ssosettingstests.NewFakeStore()
	fallbackStrategy := ssosettingstests.NewFakeFallbackStrategy()

	accessControl := acimpl.ProvideAccessControl(setting.NewCfg())
	svc := &SSOSettingsService{
		log:          log.NewNopLogger(),
		store:        store,
		ac:           accessControl,
		fbStrategies: []ssosettings.FallbackStrategy{fallbackStrategy},
	}
	return testEnv{
		service:          svc,
		store:            store,
		ac:               accessControl,
		fallbackStrategy: fallbackStrategy,
	}
}

type testEnv struct {
	service          *SSOSettingsService
	store            *ssosettingstests.FakeStore
	ac               accesscontrol.AccessControl
	fallbackStrategy *ssosettingstests.FakeFallbackStrategy
}