forked from coop-cloud/abra
80 lines
2.1 KiB
Go
80 lines
2.1 KiB
Go
package client_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"coopcloud.tech/abra/pkg/client"
|
|
dContext "github.com/docker/cli/cli/context"
|
|
dCliContextStore "github.com/docker/cli/cli/context/store"
|
|
)
|
|
|
|
type TestContext struct {
|
|
context dCliContextStore.Metadata
|
|
expected_endpoint string
|
|
}
|
|
|
|
func dockerContext(host, key string) TestContext {
|
|
dockerContext := dCliContextStore.Metadata{
|
|
Name: "foo",
|
|
Metadata: nil,
|
|
Endpoints: map[string]interface{}{
|
|
key: dContext.EndpointMetaBase{
|
|
Host: host,
|
|
SkipTLSVerify: false,
|
|
},
|
|
},
|
|
}
|
|
return TestContext{
|
|
context: dockerContext,
|
|
expected_endpoint: host,
|
|
}
|
|
}
|
|
|
|
func TestCreateContext(t *testing.T) {
|
|
err := client.CreateContext("testCreate", "wronguser", "wrongport")
|
|
if err == nil {
|
|
t.Error("client.CreateContext(\"testCreate\", \"wronguser\", \"wrongport\") should have failed but didn't return an error")
|
|
}
|
|
err = client.CreateContext("testCreate", "", "")
|
|
if err != nil {
|
|
t.Errorf("Couldn't create context: %s", err)
|
|
}
|
|
}
|
|
|
|
func TestDeleteContext(t *testing.T) {
|
|
err := client.DeleteContext("default")
|
|
if err == nil {
|
|
t.Errorf("client.DeleteContext(\"default\") should have failed but didn't return an error")
|
|
}
|
|
|
|
err = client.DeleteContext("testCreate")
|
|
if err != nil {
|
|
t.Errorf("client.DeleteContext(\"testCreate\") failed: %s", err)
|
|
}
|
|
err = client.DeleteContext("testCreate")
|
|
if err == nil {
|
|
t.Errorf("client.DeleteContext(\"testCreate\") should have failed this time (attempt to delete non-existent context) but didn't return an error")
|
|
}
|
|
}
|
|
|
|
func TestGetContextEndpoint(t *testing.T) {
|
|
var testDockerContexts = []TestContext{
|
|
dockerContext("ssh://foobar", "docker"),
|
|
dockerContext("ssh://foobar", "k8"),
|
|
}
|
|
for _, context := range testDockerContexts {
|
|
endpoint, err := client.GetContextEndpoint(context.context)
|
|
if err != nil {
|
|
if err.Error() != "context lacks Docker endpoint" {
|
|
t.Error(err)
|
|
}
|
|
} else {
|
|
if endpoint != context.expected_endpoint {
|
|
t.Errorf("did not get correct context endpoint. Expected: %s, received: %s", context.expected_endpoint, endpoint)
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
}
|