82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
|
// +build kubernetes
|
||
|
|
|
||
|
|
package kubernetes
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/base64"
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
"os/exec"
|
||
|
|
"regexp"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func setupClient(t *testing.T) {
|
||
|
|
files := []string{"token", "ca.crt"}
|
||
|
|
for _, f := range files {
|
||
|
|
cmd := exec.Command("kubectl", "get", "secrets", "-o",
|
||
|
|
fmt.Sprintf(`jsonpath="{.items[?(@.metadata.annotations['kubernetes\.io/service-account\.name']=='micro-runtime')].data.%s}"`,
|
||
|
|
strings.ReplaceAll(f, ".", "\\.")))
|
||
|
|
if outp, err := cmd.Output(); err != nil {
|
||
|
|
t.Fatalf("Failed to set k8s token %s", err)
|
||
|
|
} else {
|
||
|
|
outq := outp[1 : len(outp)-1]
|
||
|
|
decoded, err := base64.StdEncoding.DecodeString(string(outq))
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Failed to set k8s token %s '%s'", err, outq)
|
||
|
|
}
|
||
|
|
if err := ioutil.WriteFile("/var/run/secrets/kubernetes.io/serviceaccount/"+f, decoded, 0755); err != nil {
|
||
|
|
t.Fatalf("Error setting up k8s %s", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
outp, err := exec.Command("kubectl", "config", "view", "-o", `jsonpath='{.clusters[?(@.name=="kind-kind")].cluster.server}'`).Output()
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Cannot find server for kind %s", err)
|
||
|
|
}
|
||
|
|
serverHost := string(outp)
|
||
|
|
|
||
|
|
split := strings.Split(serverHost[9:len(serverHost)-1], ":")
|
||
|
|
os.Setenv("KUBERNETES_SERVICE_HOST", split[0])
|
||
|
|
os.Setenv("KUBERNETES_SERVICE_PORT", split[1])
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNamespaceCreateDelete(t *testing.T) {
|
||
|
|
defer func() {
|
||
|
|
exec.Command("kubectl", "delete", "namespace", "foobar").Run()
|
||
|
|
}()
|
||
|
|
setupClient(t)
|
||
|
|
r := NewRuntime()
|
||
|
|
if err := r.CreateNamespace("foobar"); err != nil {
|
||
|
|
t.Fatalf("Unexpected error creating namespace %s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if !namespaceExists(t, "foobar") {
|
||
|
|
t.Fatalf("Namespace foobar not found")
|
||
|
|
}
|
||
|
|
if err := r.DeleteNamespace("foobar"); err != nil {
|
||
|
|
t.Fatalf("Unexpected error deleting namespace %s", err)
|
||
|
|
}
|
||
|
|
if namespaceExists(t, "foobar") {
|
||
|
|
t.Fatalf("Namespace foobar still exists")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func namespaceExists(t *testing.T, ns string) bool {
|
||
|
|
cmd := exec.Command("kubectl", "get", "namespaces")
|
||
|
|
outp, err := cmd.Output()
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Unexpected error listing namespaces %s", err)
|
||
|
|
}
|
||
|
|
exists, err := regexp.Match(ns+"\\s+Active", outp)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Error listing namespaces %s", err)
|
||
|
|
}
|
||
|
|
return exists
|
||
|
|
|
||
|
|
}
|