Michael Mandrus
542a1bf3ac
* implement querying gms for snapshot status * add some documentation * provide snapshot resources after snapshot is created * add rate limiting to backend * fix compilation error * fix typo * add unit tests * finish merge * lint * swagger gen * more testing * remove duplicate test * address a couple PR comments * update switch statement to a map * add timeouts to gms client through the http client * remove extra whitespace * put method back where it was so the PR is less confusing * fix tests * add todo * fix final unit test
57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
package gmsclient
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func Test_buildBasePath(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
// Domain is required
|
|
_, err := NewGMSClient(&setting.Cfg{
|
|
CloudMigration: setting.CloudMigrationSettings{
|
|
GMSDomain: "",
|
|
},
|
|
})
|
|
require.Error(t, err)
|
|
|
|
// Domain is required
|
|
c, err := NewGMSClient(&setting.Cfg{
|
|
CloudMigration: setting.CloudMigrationSettings{
|
|
GMSDomain: "non-empty",
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
client := c.(*gmsClientImpl)
|
|
|
|
tests := []struct {
|
|
description string
|
|
domain string
|
|
clusterSlug string
|
|
expected string
|
|
}{
|
|
{
|
|
description: "domain starts with http://localhost, should return domain",
|
|
domain: "http://localhost:8080",
|
|
clusterSlug: "anything",
|
|
expected: "http://localhost:8080",
|
|
},
|
|
{
|
|
description: "domain doesn't start with http://localhost, should build a string using the domain and clusterSlug",
|
|
domain: "gms-dev",
|
|
clusterSlug: "us-east-1",
|
|
expected: "https://cms-us-east-1.gms-dev/cloud-migrations",
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.description, func(t *testing.T) {
|
|
client.cfg.CloudMigration.GMSDomain = tt.domain
|
|
assert.Equal(t, tt.expected, client.buildBasePath(tt.clusterSlug))
|
|
})
|
|
}
|
|
}
|