e2e tests
Karolis Rusenas
parent
bd00613e7e
commit
379affe794
|
|
@ -0,0 +1,187 @@
|
|||
package tests
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"net/http"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/keel-hq/keel/types"
|
||||
|
||||
apps_v1 "k8s.io/api/apps/v1"
|
||||
"k8s.io/api/core/v1"
|
||||
meta_v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
)
|
||||
|
||||
var dockerHub0150Webhook = `{
|
||||
"push_data": {
|
||||
"pushed_at": 1497467660,
|
||||
"images": [],
|
||||
"tag": "0.0.15",
|
||||
"pusher": "karolisr"
|
||||
},
|
||||
"repository": {
|
||||
"status": "Active",
|
||||
"description": "",
|
||||
"is_trusted": false,
|
||||
"repo_url": "https://hub.docker.com/r/webhook-demo",
|
||||
"owner": "karolisr",
|
||||
"is_official": false,
|
||||
"is_private": false,
|
||||
"name": "keel",
|
||||
"namespace": "karolisr",
|
||||
"star_count": 0,
|
||||
"comment_count": 0,
|
||||
"date_created": 1497032538,
|
||||
"repo_name": "karolisr/webhook-demo"
|
||||
}
|
||||
}`
|
||||
|
||||
func TestSemverUpdate(t *testing.T) {
|
||||
|
||||
// stop := make(chan struct{})
|
||||
context, cancel := context.WithCancel(context.Background())
|
||||
// defer close(ctx)
|
||||
defer cancel()
|
||||
|
||||
go startKeel(context)
|
||||
|
||||
_, kcs := getKubernetesClient()
|
||||
|
||||
t.Run("UpdateThroughDockerHubWebhook", func(t *testing.T) {
|
||||
|
||||
testNamespace := createNamespaceForTest()
|
||||
defer deleteTestNamespace(testNamespace)
|
||||
|
||||
dep := &apps_v1.Deployment{
|
||||
meta_v1.TypeMeta{},
|
||||
meta_v1.ObjectMeta{
|
||||
Name: "deployment-1",
|
||||
Namespace: testNamespace,
|
||||
Labels: map[string]string{types.KeelPolicyLabel: "all"},
|
||||
Annotations: map[string]string{},
|
||||
},
|
||||
apps_v1.DeploymentSpec{
|
||||
Selector: &meta_v1.LabelSelector{
|
||||
MatchLabels: map[string]string{
|
||||
"app": "wd-1",
|
||||
},
|
||||
},
|
||||
Template: v1.PodTemplateSpec{
|
||||
ObjectMeta: meta_v1.ObjectMeta{
|
||||
Labels: map[string]string{
|
||||
"app": "wd-1",
|
||||
"release": "1",
|
||||
},
|
||||
},
|
||||
Spec: v1.PodSpec{
|
||||
Containers: []v1.Container{
|
||||
v1.Container{
|
||||
Name: "wd-1",
|
||||
Image: "karolisr/webhook-demo:0.0.14",
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
apps_v1.DeploymentStatus{},
|
||||
}
|
||||
|
||||
_, err := kcs.AppsV1().Deployments(testNamespace).Create(dep)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to create deployment: %s", err)
|
||||
}
|
||||
// giving some time to get started
|
||||
// TODO: replace with a readiness check function to wait for 1/1 READY
|
||||
time.Sleep(2 * time.Second)
|
||||
|
||||
// sending webhook
|
||||
client := http.DefaultClient
|
||||
buf := bytes.NewBufferString(dockerHub0150Webhook)
|
||||
req, err := http.NewRequest("POST", "http://localhost:9300/v1/webhooks/dockerhub", buf)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to create req: %s", err)
|
||||
}
|
||||
resp, err := client.Do(req)
|
||||
if err != nil {
|
||||
t.Errorf("failed to make a webhook request to keel: %s", err)
|
||||
}
|
||||
|
||||
if resp.StatusCode != 200 {
|
||||
t.Errorf("unexpected webhook response from keel: %d", resp.StatusCode)
|
||||
}
|
||||
|
||||
time.Sleep(2 * time.Second)
|
||||
|
||||
updated, err := kcs.AppsV1().Deployments(testNamespace).Get(dep.ObjectMeta.Name, meta_v1.GetOptions{})
|
||||
if err != nil {
|
||||
t.Fatalf("failed to get deployment: %s", err)
|
||||
}
|
||||
|
||||
if updated.Spec.Template.Spec.Containers[0].Image != "karolisr/webhook-demo:0.0.15" {
|
||||
t.Errorf("expected 'karolisr/webhook-demo:0.0.15', got: '%s'", updated.Spec.Template.Spec.Containers[0].Image)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("UpdateThroughDockerHubPolling", func(t *testing.T) {
|
||||
|
||||
testNamespace := createNamespaceForTest()
|
||||
defer deleteTestNamespace(testNamespace)
|
||||
|
||||
dep := &apps_v1.Deployment{
|
||||
meta_v1.TypeMeta{},
|
||||
meta_v1.ObjectMeta{
|
||||
Name: "deployment-1",
|
||||
Namespace: testNamespace,
|
||||
Labels: map[string]string{
|
||||
types.KeelPolicyLabel: "major",
|
||||
types.KeelTriggerLabel: "poll",
|
||||
},
|
||||
Annotations: map[string]string{},
|
||||
},
|
||||
apps_v1.DeploymentSpec{
|
||||
Selector: &meta_v1.LabelSelector{
|
||||
MatchLabels: map[string]string{
|
||||
"app": "wd-1",
|
||||
},
|
||||
},
|
||||
Template: v1.PodTemplateSpec{
|
||||
ObjectMeta: meta_v1.ObjectMeta{
|
||||
Labels: map[string]string{
|
||||
"app": "wd-1",
|
||||
"release": "1",
|
||||
},
|
||||
},
|
||||
Spec: v1.PodSpec{
|
||||
Containers: []v1.Container{
|
||||
v1.Container{
|
||||
Name: "wd-1",
|
||||
Image: "keelhq/push-workflow-example:0.1.0-dev",
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
apps_v1.DeploymentStatus{},
|
||||
}
|
||||
|
||||
_, err := kcs.AppsV1().Deployments(testNamespace).Create(dep)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to create deployment: %s", err)
|
||||
}
|
||||
// giving some time to get started
|
||||
// TODO: replace with a readiness check function to wait for 1/1 READY
|
||||
time.Sleep(2 * time.Second)
|
||||
|
||||
updated, err := kcs.AppsV1().Deployments(testNamespace).Get(dep.ObjectMeta.Name, meta_v1.GetOptions{})
|
||||
if err != nil {
|
||||
t.Fatalf("failed to get deployment: %s", err)
|
||||
}
|
||||
|
||||
if updated.Spec.Template.Spec.Containers[0].Image != "keelhq/push-workflow-example:0.5.0-dev" {
|
||||
t.Errorf("expected 'keelhq/push-workflow-example:0.5.0-dev', got: '%s'", updated.Spec.Template.Spec.Containers[0].Image)
|
||||
}
|
||||
})
|
||||
|
||||
}
|
||||
|
|
@ -0,0 +1,93 @@
|
|||
package tests
|
||||
|
||||
import (
|
||||
"context"
|
||||
"flag"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
|
||||
"k8s.io/api/core/v1"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/client-go/kubernetes"
|
||||
"k8s.io/client-go/rest"
|
||||
"k8s.io/client-go/tools/clientcmd"
|
||||
|
||||
// load the gcp plugin (required to authenticate against GKE clusters).
|
||||
_ "k8s.io/client-go/plugin/pkg/client/auth/gcp"
|
||||
|
||||
log "github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
var kubeConfig = flag.String("kubeconfig", "", "Path to Kubernetes config file")
|
||||
|
||||
func getKubeConfig() string {
|
||||
if *kubeConfig != "" {
|
||||
return *kubeConfig
|
||||
}
|
||||
|
||||
return filepath.Join(os.Getenv("HOME"), ".kube", "config")
|
||||
}
|
||||
|
||||
func getKubernetesClient() (*rest.Config, *kubernetes.Clientset) {
|
||||
// use the current context in kubeconfig
|
||||
config, err := clientcmd.BuildConfigFromFlags("", getKubeConfig())
|
||||
if err != nil {
|
||||
panic(err.Error())
|
||||
}
|
||||
|
||||
// create the clientset
|
||||
clientSet, err := kubernetes.NewForConfig(config)
|
||||
if err != nil {
|
||||
panic(err.Error())
|
||||
}
|
||||
|
||||
return config, clientSet
|
||||
}
|
||||
|
||||
func createNamespaceForTest() string {
|
||||
_, clientset := getKubernetesClient()
|
||||
ns := &v1.Namespace{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
GenerateName: "keel-e2e-test-",
|
||||
},
|
||||
}
|
||||
cns, err := clientset.CoreV1().Namespaces().Create(ns)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
log.Infof("test namespace '%s' created", cns.Name)
|
||||
|
||||
return cns.Name
|
||||
}
|
||||
|
||||
func deleteTestNamespace(namespace string) error {
|
||||
|
||||
defer log.Infof("test namespace '%s' deleted", namespace)
|
||||
_, clientset := getKubernetesClient()
|
||||
deleteOptions := metav1.DeleteOptions{}
|
||||
return clientset.CoreV1().Namespaces().Delete(namespace, &deleteOptions)
|
||||
}
|
||||
|
||||
func startKeel(ctx context.Context) error {
|
||||
|
||||
log.Info("keel started")
|
||||
defer log.Info("keel stopped")
|
||||
|
||||
cmd := "keel"
|
||||
args := []string{"--no-incluster", "--kubeconfig", getKubeConfig()}
|
||||
c := exec.CommandContext(ctx, cmd, args...)
|
||||
c.Stdout = os.Stdout
|
||||
c.Stderr = os.Stderr
|
||||
|
||||
go func() {
|
||||
<-ctx.Done()
|
||||
err := c.Process.Kill()
|
||||
if err != nil {
|
||||
log.Errorf("failed to kill keel process: %s", err)
|
||||
}
|
||||
}()
|
||||
|
||||
return c.Run()
|
||||
}
|
||||
Loading…
Reference in New Issue