Add persistence disabled test
This commit is contained in:
Arthur Barr
@@ -16,176 +16,38 @@ limitations under the License.
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"os/exec"
|
||||
"runtime"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"golang.org/x/sys/unix"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/client-go/kubernetes"
|
||||
"k8s.io/client-go/pkg/api/v1"
|
||||
"k8s.io/client-go/tools/clientcmd"
|
||||
)
|
||||
|
||||
var namespace = "default"
|
||||
|
||||
// runCommand runs an OS command. On Linux it waits for the command to
|
||||
// complete and returns the exit status (return code).
|
||||
// TODO: duplicated from cmd/runmqserver/main.go
|
||||
func runCommand(t *testing.T, name string, arg ...string) (string, int, error) {
|
||||
t.Logf("Running command: %v %v", name, strings.Trim(fmt.Sprintf("%v", arg), "[]"))
|
||||
cmd := exec.Command(name, arg...)
|
||||
// Run the command and wait for completion
|
||||
out, err := cmd.CombinedOutput()
|
||||
if err != nil {
|
||||
var rc int
|
||||
// Only works on Linux
|
||||
if runtime.GOOS == "linux" {
|
||||
var ws unix.WaitStatus
|
||||
unix.Wait4(cmd.Process.Pid, &ws, 0, nil)
|
||||
rc = ws.ExitStatus()
|
||||
} else {
|
||||
rc = -1
|
||||
}
|
||||
if rc == 0 {
|
||||
return string(out), rc, nil
|
||||
}
|
||||
return string(out), rc, err
|
||||
}
|
||||
return string(out), 0, nil
|
||||
}
|
||||
|
||||
func helmInstall(t *testing.T, cs *kubernetes.Clientset, release string, values ...string) {
|
||||
chart := "../../charts/ibm-mqadvanced-server-prod"
|
||||
image := "mycluster.icp:8500/default/mq-devserver"
|
||||
tag := "latest"
|
||||
arg := []string{
|
||||
"install",
|
||||
"--debug",
|
||||
chart,
|
||||
"--name",
|
||||
release,
|
||||
"--set",
|
||||
"image.repository=" + image,
|
||||
"--set",
|
||||
"image.tag=" + tag,
|
||||
"--set",
|
||||
"image.pullSecret=admin.registrykey",
|
||||
}
|
||||
for _, value := range values {
|
||||
arg = append(arg, "--set", value)
|
||||
}
|
||||
out, _, err := runCommand(t, "helm", arg...)
|
||||
t.Log(out)
|
||||
if err != nil {
|
||||
t.Error(out)
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func helmDelete(t *testing.T, release string) {
|
||||
out, _, err := runCommand(t, "helm", "delete", "--purge", release)
|
||||
if err != nil {
|
||||
t.Error(out)
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func helmDeleteWithPVC(t *testing.T, cs *kubernetes.Clientset, release string) {
|
||||
helmDelete(t, release)
|
||||
pvcs, err := cs.CoreV1().PersistentVolumeClaims(namespace).List(metav1.ListOptions{
|
||||
LabelSelector: "release=" + release,
|
||||
})
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
for _, pvc := range pvcs.Items {
|
||||
t.Logf("Deleting persistent volume claim: %v", pvc.Name)
|
||||
err := cs.CoreV1().PersistentVolumeClaims(namespace).Delete(pvc.Name, &metav1.DeleteOptions{})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func kubeLogin(t *testing.T) *kubernetes.Clientset {
|
||||
kc := os.Getenv("HOME") + "/.kube/config"
|
||||
c, err := clientcmd.BuildConfigFromFlags("", kc)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
cs, err := kubernetes.NewForConfig(c)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
return cs
|
||||
}
|
||||
|
||||
func waitForReady(t *testing.T, cs *kubernetes.Clientset, release string) {
|
||||
pods := getPodsForHelmRelease(t, cs, release)
|
||||
pod := pods.Items[0]
|
||||
podName := pod.Name
|
||||
// Wait for the queue manager container to be started...
|
||||
for {
|
||||
pod, err := cs.CoreV1().Pods(namespace).Get(podName, metav1.GetOptions{})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(pod.Status.ContainerStatuses) > 0 {
|
||||
// Got a container now, but it could still be in state "ContainerCreating"
|
||||
// TODO: Check the status here properly
|
||||
time.Sleep(3 * time.Second)
|
||||
break
|
||||
}
|
||||
}
|
||||
// Exec into the container to check if it's ready
|
||||
for {
|
||||
// Current version of Kubernetes Go client doesn't support "exec", so run this via the command line
|
||||
// TODO: If we run "chkmqready" here, it doesn't seem to work
|
||||
out, _, err := runCommand(t, "kubectl", "exec", podName, "--", "dspmq")
|
||||
//out, rc, err := runCommand(t, "kubectl", "exec", podName, "--", "chkmqready")
|
||||
if err != nil {
|
||||
t.Error(out)
|
||||
out2, _, err2 := runCommand(t, "kubectl", "describe", "pod", podName)
|
||||
if err2 == nil {
|
||||
t.Log(out2)
|
||||
}
|
||||
t.Fatal(err)
|
||||
}
|
||||
if strings.Contains(out, "Running") {
|
||||
t.Log("MQ is ready")
|
||||
return
|
||||
}
|
||||
// if rc == 0 {
|
||||
// return
|
||||
// }
|
||||
time.Sleep(1 * time.Second)
|
||||
}
|
||||
}
|
||||
|
||||
func getPodsForHelmRelease(t *testing.T, cs *kubernetes.Clientset, release string) *v1.PodList {
|
||||
pods, err := cs.CoreV1().Pods(namespace).List(metav1.ListOptions{
|
||||
LabelSelector: "release=" + release,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
return pods
|
||||
}
|
||||
|
||||
func TestHelmGoldenPath(t *testing.T) {
|
||||
cs := kubeLogin(t)
|
||||
release := strings.ToLower(t.Name())
|
||||
helmInstall(t, cs, release, "license=accept", "persistence.useDynamicProvisioning=false")
|
||||
defer helmDeleteWithPVC(t, cs, release)
|
||||
|
||||
defer helmDelete(t, release)
|
||||
defer helmDeletePVC(t, cs, release)
|
||||
waitForReady(t, cs, release)
|
||||
pods := getPodsForHelmRelease(t, cs, release)
|
||||
if len(pods.Items) != 1 {
|
||||
t.Fatalf("Expected 1 pod with the right label, found %v pods", len(pods.Items))
|
||||
}
|
||||
|
||||
func TestPersistenceDisabled(t *testing.T) {
|
||||
cs := kubeLogin(t)
|
||||
release := strings.ToLower(t.Name())
|
||||
helmInstall(t, cs, release, "license=accept", "persistence.enabled=false")
|
||||
defer helmDelete(t, release)
|
||||
waitForReady(t, cs, release)
|
||||
|
||||
// Check that no PVCs were created
|
||||
pvcs, err := cs.CoreV1().PersistentVolumeClaims(namespace).List(metav1.ListOptions{
|
||||
LabelSelector: "release=" + release,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(pvcs.Items) > 0 {
|
||||
t.Errorf("Expected no PVC, found %v (%+v)", len(pvcs.Items), pvcs.Items)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user