netbird/management/cmd/admin_config_test.go

package cmd

import (
	"context"
	"path"
	"path/filepath"
	"testing"

	"github.com/stretchr/testify/require"

	nbconfig "github.com/netbirdio/netbird/management/internals/server/config"
	"github.com/netbirdio/netbird/management/server/idp"
)

func TestApplyAdminDatadirOverrideRelocatesDefaultIDPStorage(t *testing.T) {
	oldDatadir := filepath.Join(t.TempDir(), "old")
	newDatadir := filepath.Join(t.TempDir(), "new")

	for _, defaultFile := range []string{
		"",
		filepath.Join(oldDatadir, "idp.db"),
		path.Join(oldDatadir, "idp.db"),
	} {
		t.Run(defaultFile, func(t *testing.T) {
			cfg := &nbconfig.Config{
				EmbeddedIdP: &idp.EmbeddedIdPConfig{
					Enabled: true,
					Storage: idp.EmbeddedStorageConfig{
						Type: "sqlite3",
						Config: idp.EmbeddedStorageTypeConfig{
							File: defaultFile,
						},
					},
				},
			}
			datadir := oldDatadir
			oldAdminDatadir := adminDatadir
			adminDatadir = newDatadir
			t.Cleanup(func() { adminDatadir = oldAdminDatadir })

			applyAdminDatadirOverride(cfg, &datadir)

			require.Equal(t, newDatadir, datadir)
			require.Equal(t, filepath.Join(newDatadir, "idp.db"), cfg.EmbeddedIdP.Storage.Config.File)
		})
	}
}

func TestOpenAdminEventStoreMissingEncryptionKeyReturnsNilInterface(t *testing.T) {
	eventStore, err := openAdminEventStore(context.Background(), &nbconfig.Config{}, t.TempDir())
	require.Error(t, err)
	require.Contains(t, err.Error(), "encryption key")
	require.Nil(t, eventStore)
}

func TestApplyAdminDatadirOverrideKeepsExplicitIDPStorage(t *testing.T) {
	oldDatadir := filepath.Join(t.TempDir(), "old")
	newDatadir := filepath.Join(t.TempDir(), "new")
	explicitFile := filepath.Join(t.TempDir(), "custom-idp.db")
	cfg := &nbconfig.Config{
		EmbeddedIdP: &idp.EmbeddedIdPConfig{
			Enabled: true,
			Storage: idp.EmbeddedStorageConfig{
				Type: "sqlite3",
				Config: idp.EmbeddedStorageTypeConfig{
					File: explicitFile,
				},
			},
		},
	}
	datadir := oldDatadir
	oldAdminDatadir := adminDatadir
	adminDatadir = newDatadir
	t.Cleanup(func() { adminDatadir = oldAdminDatadir })

	applyAdminDatadirOverride(cfg, &datadir)

	require.Equal(t, newDatadir, datadir)
	require.Equal(t, explicitFile, cfg.EmbeddedIdP.Storage.Config.File)
}