2022-03-13 15:17:18 +01:00
|
|
|
package internal
|
|
|
|
|
|
|
|
import (
|
|
|
|
"errors"
|
|
|
|
"os"
|
|
|
|
"path/filepath"
|
|
|
|
"testing"
|
2022-04-15 17:30:12 +02:00
|
|
|
|
|
|
|
"github.com/netbirdio/netbird/util"
|
|
|
|
"github.com/stretchr/testify/assert"
|
2022-03-13 15:17:18 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
func TestReadConfig(t *testing.T) {
|
|
|
|
}
|
|
|
|
|
2022-04-15 17:30:12 +02:00
|
|
|
func TestGetConfig(t *testing.T) {
|
2022-03-13 15:17:18 +01:00
|
|
|
managementURL := "https://test.management.url:33071"
|
2022-04-15 17:30:12 +02:00
|
|
|
adminURL := "https://app.admin.url"
|
2022-03-13 15:17:18 +01:00
|
|
|
path := filepath.Join(t.TempDir(), "config.json")
|
|
|
|
preSharedKey := "preSharedKey"
|
|
|
|
|
|
|
|
// case 1: new config has to be generated
|
2022-04-15 17:30:12 +02:00
|
|
|
config, err := GetConfig(managementURL, adminURL, path, preSharedKey)
|
2022-03-13 15:17:18 +01:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
assert.Equal(t, config.ManagementURL.String(), managementURL)
|
|
|
|
assert.Equal(t, config.PreSharedKey, preSharedKey)
|
|
|
|
|
|
|
|
if _, err := os.Stat(path); errors.Is(err, os.ErrNotExist) {
|
|
|
|
t.Errorf("config file was expected to be created under path %s", path)
|
|
|
|
}
|
|
|
|
|
|
|
|
// case 2: existing config -> fetch it
|
2022-04-15 17:30:12 +02:00
|
|
|
config, err = GetConfig(managementURL, adminURL, path, preSharedKey)
|
2022-03-13 15:17:18 +01:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
assert.Equal(t, config.ManagementURL.String(), managementURL)
|
|
|
|
assert.Equal(t, config.PreSharedKey, preSharedKey)
|
|
|
|
|
|
|
|
// case 3: existing config, but new managementURL has been provided -> update config
|
|
|
|
newManagementURL := "https://test.newManagement.url:33071"
|
2022-04-15 17:30:12 +02:00
|
|
|
config, err = GetConfig(newManagementURL, adminURL, path, preSharedKey)
|
2022-03-13 15:17:18 +01:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
assert.Equal(t, config.ManagementURL.String(), newManagementURL)
|
|
|
|
assert.Equal(t, config.PreSharedKey, preSharedKey)
|
|
|
|
|
|
|
|
// read once more to make sure that config file has been updated with the new management URL
|
|
|
|
readConf, err := util.ReadJson(path, config)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
assert.Equal(t, readConf.(*Config).ManagementURL.String(), newManagementURL)
|
|
|
|
}
|