main_test.go

package main

import (
	"bytes"
	"net/http"
	"net/http/httptest"
	"testing"

	"go.b8s.dev/rollingpin/config"
	"go.b8s.dev/rollingpin/kube"
	"go.uber.org/zap"
)

func TestHarborWebhookEndToEnd(t *testing.T) {
	// Set up test HTTP request
	payload := `{
		"type": "PUSH_ARTIFACT",
		"occur_at": 1586922308,
		"operator": "admin",
		"event_data": {
			"resources": [{
				"digest": "sha256:8a9e9863dbb6e10edb5adfe917c00da84e1700fa76e7ed02476aa6e6fb8ee0d8",
				"tag": "latest",
				"resource_url": "cr.b8s.dev/library/debian:v2"
			}],
			"repository": {
				"date_created": 1586922308,
				"name": "debian",
				"namespace": "library",
				"repo_full_name": "library/debian",
				"repo_type": "private"
			}
		}
	}`
	req, _ := http.NewRequest("POST", "/webhooks/harbor", bytes.NewBufferString(payload))
	req.Header.Add("authorization", "Bearer abc1234")
	resp := httptest.NewRecorder()

	// Set up fake kubernetes api client
	fakeClient, _ := kube.NewFake()
	fakeClient.CreateDeployment(
		&kube.Deployment{
			Namespace: "default",
			Name:      "test-deployment",
			Containers: []*kube.Container{
				{Name: "app", Image: "cr.b8s.dev/library/debian:v1"},
			},
		},
	)

	// Set up our own app's stuff
	conf := &config.Config{
		AuthToken: "abc1234",
		Mappings: []config.ImageMapping{
			{
				Namespace:      "default",
				DeploymentName: "test-deployment",
				ImageName:      "library/debian",
			},
		},
	}
	log, _ := zap.NewProduction()

	// Execute request
	r := buildRouter(conf, log, fakeClient)
	r.ServeHTTP(resp, req)

	// Assertions
	if resp.Code != 200 {
		t.Errorf("Expected 200 response got: %d", resp.Code)
	}
	if resp.Body.String() != `{"ok":true}` {
		t.Errorf("Expected OK response got: %s", resp.Body.String())
	}

	newDeploy, _ := fakeClient.GetDeployment("default", "test-deployment")
	newImageName := newDeploy.Containers[0].Image
	if newImageName != "cr.b8s.dev/library/debian:v2" {
		t.Errorf("Expected deployment to be updated but was not! Image was: %s", newImageName)
	}
}

func TestDirectWebhookEndToEnd(t *testing.T) {
	// Set up test HTTP request
	payload := `{
		"image_url": "cr.b8s.dev/library/debian:v2",
		"repository_name": "library/debian"
	}`
	req, _ := http.NewRequest("POST", "/webhooks/direct", bytes.NewBufferString(payload))
	req.Header.Add("authorization", "Bearer abc1234")
	resp := httptest.NewRecorder()

	// Set up fake kubernetes api client
	fakeClient, _ := kube.NewFake()
	fakeClient.CreateDeployment(
		&kube.Deployment{
			Namespace: "default",
			Name:      "test-deployment",
			Containers: []*kube.Container{
				{Name: "app", Image: "cr.b8s.dev/library/debian:v1"},
			},
		},
	)

	// Set up our own app's stuff
	conf := &config.Config{
		AuthToken: "abc1234",
		Mappings: []config.ImageMapping{
			{
				Namespace:      "default",
				DeploymentName: "test-deployment",
				ImageName:      "library/debian",
			},
		},
	}
	log, _ := zap.NewProduction()

	// Execute request
	r := buildRouter(conf, log, fakeClient)
	r.ServeHTTP(resp, req)

	// Assertions
	if resp.Code != 200 {
		t.Errorf("Expected 200 response got: %d", resp.Code)
	}
	if resp.Body.String() != `{"ok":true}` {
		t.Errorf("Expected OK response got: %s", resp.Body.String())
	}

	newDeploy, _ := fakeClient.GetDeployment("default", "test-deployment")
	newImageName := newDeploy.Containers[0].Image
	if newImageName != "cr.b8s.dev/library/debian:v2" {
		t.Errorf("Expected deployment to be updated but was not! Image was: %s", newImageName)
	}
}