package webhooks_test

import (
	"bytes"
	"io/ioutil"
	"testing"

	"github.com/nirmata/kube-policy/pkg/webhooks"
	"gotest.tools/assert"
	rest "k8s.io/client-go/rest"
)

func TestExtractCA_EmptyBundle(t *testing.T) {
	CAFile := "resources/CAFile"

	config := &rest.Config{
		TLSClientConfig: rest.TLSClientConfig{
			CAData: nil,
			CAFile: CAFile,
		},
	}

	expected, err := ioutil.ReadFile(CAFile)
	assert.Assert(t, err == nil)
	actual := webhooks.ExtractCA(config)
	assert.Assert(t, bytes.Equal(expected, actual))
}

func TestExtractCA_EmptyCAFile(t *testing.T) {
	CABundle := []byte(`LS0tLS1CRUdJTiBDRVJUSUZJQ0FURS0tLS0tCk1JSUN5RENDQWJDZ0F3SUJBZ0lCQURBTkJna3Foa2lHOXcwQkFRc0ZBREFWTVJNd0VRWURWUVFERXdwcmRXSmwKY201bGRHVnpNQjRYRFRFNU1ETXhPVEUwTURjd05Gb1hEVEk1TURNeE5qRTBNRGN3TkZvd0ZURVRNQkVHQTFVRQpBeE1LYTNWaVpYSnVaWFJsY3pDQ0FTSXdEUVlKS29aSWh2Y05BUUVCQlFBRGdnRVBBRENDQVFvQ2dnRUJBTStQClVLVmExcm9tQndOZzdqNnBBSGo5TDQ4RVJpdEplRzRXM1pUYmNMNWNKbnVTQmFsc1h1TWpQTGZmbUV1VEZIdVAKenRqUlBEUHcreEg1d3VTWFF2U0tIaXF2VE1pUm9DSlJFa09sQXpIa1dQM0VrdnUzNzRqZDVGV3Q3NEhnRk91cApIZ1ZwdUxPblczK2NDVE5iQ3VkeDFMVldRbGgwQzJKbm1Lam5uS1YrTkxzNFJVaVk1dk91ekpuNHl6QldLRjM2CmJLZ3ZDOVpMWlFSM3dZcnJNZWllYzBnWVY2VlJtaGgxSjRDV3V1UWd0ckM2d2NJanFWZFdEUlJyNHFMdEtDcDIKQVNIZmNieitwcEdHblJ5Z2FzcWNJdnpiNUVwV3NIRGtHRStUUW5WQ0JmTmsxN0NEOTZBQ1pmRWVybzEvWE16MgpRbzZvcUE0dnF5ZkdWWVU5RVZFQ0F3RUFBYU1qTUNFd0RnWURWUjBQQVFIL0JBUURBZ0trTUE4R0ExVWRFd0VCCi93UUZNQU1CQWY4d0RRWUpLb1pJaHZjTkFRRUxCUUFEZ2dFQkFNWFVpUVJpdUc4cGdzcHMrZTdGZWdCdEJOZEcKZlFUdHVLRWFUZ0U0RjQwamJ3UmdrN25DTHlsSHgvRG04aVRRQmsyWjR4WnNuY0huRys4SkwrckRLdlJBSE5iVQpsYnpReXA1V3FwdjdPcThwZ01wU0o5bTdVY3BGZmRVZkorNW43aXFnTGdMb3lhNmtRVTR2Rk0yTE1rWjI5NVpxCmVId0hnREo5Z3IwWGNyOWM1L2tRdkxFc2Z2WU5QZVhuamNyWXlDb2JNcVduSElxeVd3cHM1VTJOaGgraXhSZEIKbzRRL3RJS04xOU93WGZBaVc5SENhNzZMb3ZXaUhPU2UxVnFzK1h1N1A5ckx4eW1vQm91aFcxVmZ0bUo5Qy9vTAp3cFVuNnlXRCttY0tkZ3J5QTFjTWJ4Q281bUd6YTNLaFk1QTd5eDQ1cThkSEIzTWU4d0FCam1wWEs0ST0KLS0tLS1FTkQgQ0VSVElGSUNBVEUtLS0tLQo=`)

	config := &rest.Config{
		TLSClientConfig: rest.TLSClientConfig{
			CAData: CABundle,
			CAFile: "",
		},
	}

	actual := webhooks.ExtractCA(config)
	assert.Assert(t, bytes.Equal(CABundle, actual))
}

func TestExtractCA_EmptyConfig(t *testing.T) {
	config := &rest.Config{
		TLSClientConfig: rest.TLSClientConfig{
			CAData: nil,
			CAFile: "",
		},
	}

	actual := webhooks.ExtractCA(config)
	assert.Assert(t, actual == nil)
}

func TestExtractCA_InvalidFile(t *testing.T) {
	config := &rest.Config{
		TLSClientConfig: rest.TLSClientConfig{
			CAData: nil,
			CAFile: "somenonexistingfile",
		},
	}

	actual := webhooks.ExtractCA(config)
	assert.Assert(t, actual == nil)
}