mirror of
https://github.com/ipedrazas/drone-helm.git
synced 2026-06-04 18:24:13 +08:00
changing to a plugin package to allow reuse
This commit is contained in:
Joachim Hill-Grannec
@@ -0,0 +1,398 @@
|
||||
package plugin
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"os"
|
||||
"os/exec"
|
||||
"regexp"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"github.com/alecthomas/template"
|
||||
)
|
||||
|
||||
var HELM_BIN = "/bin/helm"
|
||||
var KUBECONFIG = "/root/.kube/kubeconfig"
|
||||
|
||||
type (
|
||||
// Config maps the params we need to run Helm
|
||||
Config struct {
|
||||
APIServer string `json:"api_server"`
|
||||
Token string `json:"token"`
|
||||
Certificate string `json:"certificate"`
|
||||
ServiceAccount string `json:"service_account"`
|
||||
KubeConfig string `json:"kube_config"`
|
||||
HelmCommand string `json:"helm_command"`
|
||||
SkipTLSVerify bool `json:"tls_skip_verify"`
|
||||
Namespace string `json:"namespace"`
|
||||
Release string `json:"release"`
|
||||
Chart string `json:"chart"`
|
||||
Version string `json:"version"`
|
||||
Values string `json:"values"`
|
||||
ValuesFiles string `json:"values_files"`
|
||||
Debug bool `json:"debug"`
|
||||
DryRun bool `json:"dry_run"`
|
||||
Secrets []string `json:"secrets"`
|
||||
Prefix string `json:"prefix"`
|
||||
TillerNs string `json:"tiller_ns"`
|
||||
Wait bool `json:"wait"`
|
||||
RecreatePods bool `json:"recreate_pods"`
|
||||
Upgrade bool `json:"upgrade"`
|
||||
CanaryImage bool `json:"canary_image"`
|
||||
ClientOnly bool `json:"client_only"`
|
||||
ReuseValues bool `json:"reuse_values"`
|
||||
Timeout string `json:"timeout"`
|
||||
Force bool `json:"force"`
|
||||
HelmRepos []string `json:"helm_repos"`
|
||||
Purge bool `json:"purge"`
|
||||
UpdateDependencies bool `json:"update_dependencies"`
|
||||
}
|
||||
// Plugin default
|
||||
Plugin struct {
|
||||
Config Config
|
||||
command []string
|
||||
}
|
||||
)
|
||||
|
||||
func setHelpCommand(p *Plugin) {
|
||||
p.command = []string{""}
|
||||
}
|
||||
func setDeleteCommand(p *Plugin) {
|
||||
delete := make([]string, 2)
|
||||
delete[0] = "delete"
|
||||
delete[1] = p.Config.Release
|
||||
|
||||
if p.Config.TillerNs != "" {
|
||||
delete = append(delete, "--tiller-namespace")
|
||||
delete = append(delete, p.Config.TillerNs)
|
||||
}
|
||||
if p.Config.DryRun {
|
||||
delete = append(delete, "--dry-run")
|
||||
}
|
||||
if p.Config.Purge {
|
||||
delete = append(delete, "--purge")
|
||||
}
|
||||
|
||||
p.command = delete
|
||||
}
|
||||
|
||||
func setUpgradeCommand(p *Plugin) {
|
||||
upgrade := make([]string, 2)
|
||||
upgrade[0] = "upgrade"
|
||||
upgrade[1] = "--install"
|
||||
|
||||
if p.Config.Release != "" {
|
||||
upgrade = append(upgrade, p.Config.Release)
|
||||
}
|
||||
upgrade = append(upgrade, p.Config.Chart)
|
||||
if p.Config.Version != "" {
|
||||
upgrade = append(upgrade, "--version")
|
||||
upgrade = append(upgrade, p.Config.Version)
|
||||
}
|
||||
if p.Config.Values != "" {
|
||||
upgrade = append(upgrade, "--set")
|
||||
upgrade = append(upgrade, unQuote(p.Config.Values))
|
||||
}
|
||||
if p.Config.ValuesFiles != "" {
|
||||
for _, valuesFile := range strings.Split(p.Config.ValuesFiles, ",") {
|
||||
upgrade = append(upgrade, "--values")
|
||||
upgrade = append(upgrade, valuesFile)
|
||||
}
|
||||
}
|
||||
if p.Config.Namespace != "" {
|
||||
upgrade = append(upgrade, "--namespace")
|
||||
upgrade = append(upgrade, p.Config.Namespace)
|
||||
}
|
||||
if p.Config.TillerNs != "" {
|
||||
upgrade = append(upgrade, "--tiller-namespace")
|
||||
upgrade = append(upgrade, p.Config.TillerNs)
|
||||
}
|
||||
if p.Config.DryRun {
|
||||
upgrade = append(upgrade, "--dry-run")
|
||||
}
|
||||
if p.Config.Debug {
|
||||
upgrade = append(upgrade, "--debug")
|
||||
}
|
||||
if p.Config.Wait {
|
||||
upgrade = append(upgrade, "--wait")
|
||||
}
|
||||
if p.Config.RecreatePods {
|
||||
upgrade = append(upgrade, "--recreate-pods")
|
||||
}
|
||||
if p.Config.ReuseValues {
|
||||
upgrade = append(upgrade, "--reuse-values")
|
||||
}
|
||||
if p.Config.Timeout != "" {
|
||||
upgrade = append(upgrade, "--timeout")
|
||||
upgrade = append(upgrade, p.Config.Timeout)
|
||||
}
|
||||
if p.Config.Force {
|
||||
upgrade = append(upgrade, "--force")
|
||||
}
|
||||
p.command = upgrade
|
||||
|
||||
}
|
||||
|
||||
func setHelmCommand(p *Plugin) {
|
||||
|
||||
switch p.Config.HelmCommand {
|
||||
case "upgrade":
|
||||
setUpgradeCommand(p)
|
||||
case "delete":
|
||||
setDeleteCommand(p)
|
||||
default:
|
||||
switch os.Getenv("DRONE_BUILD_EVENT") {
|
||||
case "push", "tag", "deployment", "pull_request":
|
||||
setUpgradeCommand(p)
|
||||
case "delete":
|
||||
setDeleteCommand(p)
|
||||
default:
|
||||
setHelpCommand(p)
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
var repoExp = regexp.MustCompile(`^(?P<name>[\w-]+)=(?P<url>(http|https)://[\w-./:]+)`)
|
||||
|
||||
// parseRepo returns map of regex capture groups (name, url)
|
||||
func parseRepo(repo string) (map[string]string, error) {
|
||||
matches := repoExp.FindStringSubmatch(repo)
|
||||
if len(matches) < 1 {
|
||||
return nil, fmt.Errorf("Invalid repo definition: %s", repo)
|
||||
}
|
||||
result := make(map[string]string)
|
||||
for i, name := range repoExp.SubexpNames() {
|
||||
if i != 0 {
|
||||
result[name] = matches[i]
|
||||
}
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func doHelmRepoAdd(repo string) ([]string, error) {
|
||||
repoMap, err := parseRepo(unQuote(repo))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
repoAdd := []string{
|
||||
"repo",
|
||||
"add",
|
||||
repoMap["name"],
|
||||
repoMap["url"],
|
||||
}
|
||||
return repoAdd, nil
|
||||
}
|
||||
|
||||
func doHelmInit(p *Plugin) []string {
|
||||
init := make([]string, 1)
|
||||
init[0] = "init"
|
||||
if p.Config.TillerNs != "" {
|
||||
init = append(init, "--tiller-namespace")
|
||||
init = append(init, p.Config.TillerNs)
|
||||
}
|
||||
if p.Config.ClientOnly {
|
||||
init = append(init, "--client-only")
|
||||
}
|
||||
if p.Config.Upgrade {
|
||||
init = append(init, "--upgrade")
|
||||
}
|
||||
if p.Config.CanaryImage {
|
||||
init = append(init, "--canary-image")
|
||||
}
|
||||
|
||||
return init
|
||||
|
||||
}
|
||||
|
||||
func doDependencyUpdate(chart string) []string {
|
||||
dependencyUpdate := []string{
|
||||
"dependency",
|
||||
"update",
|
||||
chart,
|
||||
}
|
||||
|
||||
return dependencyUpdate
|
||||
}
|
||||
|
||||
// Exec default method
|
||||
func (p *Plugin) Exec() error {
|
||||
if p.Config.Debug {
|
||||
p.debugEnv()
|
||||
}
|
||||
|
||||
// create /root/.kube/config file if not exists
|
||||
if _, err := os.Stat(p.Config.KubeConfig); os.IsNotExist(err) {
|
||||
resolveSecrets(p)
|
||||
if p.Config.APIServer == "" {
|
||||
return fmt.Errorf("Error: API Server is needed to deploy.")
|
||||
}
|
||||
if p.Config.Token == "" {
|
||||
return fmt.Errorf("Error: Token is needed to deploy.")
|
||||
}
|
||||
if p.Config.SkipTLSVerify == false && p.Config.Certificate == "" {
|
||||
return fmt.Errorf("Error: Certificate is needed to deploy when SKIP_TLS_VERIFY is false.")
|
||||
}
|
||||
initialiseKubeconfig(&p.Config, KUBECONFIG, p.Config.KubeConfig)
|
||||
}
|
||||
|
||||
if p.Config.Debug {
|
||||
p.debug()
|
||||
}
|
||||
|
||||
init := doHelmInit(p)
|
||||
err := runCommand(init)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Error running helm command: " + strings.Join(init[:], " "))
|
||||
}
|
||||
|
||||
if len(p.Config.HelmRepos) > 0 {
|
||||
for _, repo := range p.Config.HelmRepos {
|
||||
repoAdd, err := doHelmRepoAdd(repo)
|
||||
if err == nil {
|
||||
if p.Config.Debug {
|
||||
log.Println("adding helm repo: " + strings.Join(repoAdd[:], " "))
|
||||
}
|
||||
|
||||
if err = runCommand(repoAdd); err != nil {
|
||||
return fmt.Errorf("Error adding helm repo: " + err.Error())
|
||||
}
|
||||
} else {
|
||||
return err
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if p.Config.UpdateDependencies {
|
||||
if err = runCommand(doDependencyUpdate(p.Config.Chart)); err != nil {
|
||||
return fmt.Errorf("Error updating dependencies: " + err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
setHelmCommand(p)
|
||||
|
||||
if p.Config.Debug {
|
||||
log.Println("helm command: " + strings.Join(p.command, " "))
|
||||
}
|
||||
|
||||
err = runCommand(p.command)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Error running helm command: " + strings.Join(p.command[:], " "))
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func initialiseKubeconfig(params *Config, source string, target string) error {
|
||||
f, err := os.Create(target)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer f.Close()
|
||||
// parse template
|
||||
t, _ := template.ParseFiles(source)
|
||||
// execute template
|
||||
return t.Execute(f, params)
|
||||
}
|
||||
|
||||
func runCommand(params []string) error {
|
||||
cmd := new(exec.Cmd)
|
||||
cmd = exec.Command(HELM_BIN, params...)
|
||||
|
||||
cmd.Stdout = os.Stdout
|
||||
cmd.Stderr = os.Stderr
|
||||
|
||||
err := cmd.Run()
|
||||
return err
|
||||
}
|
||||
|
||||
func resolveSecrets(p *Plugin) {
|
||||
p.Config.Values = resolveEnvVar(p.Config.Values, p.Config.Prefix, p.Config.Debug)
|
||||
if p.Config.APIServer == "" {
|
||||
p.Config.APIServer = resolveEnvVar("${API_SERVER}", p.Config.Prefix, p.Config.Debug)
|
||||
}
|
||||
if p.Config.Token == "" {
|
||||
p.Config.Token = resolveEnvVar("${KUBERNETES_TOKEN}", p.Config.Prefix, p.Config.Debug)
|
||||
}
|
||||
if p.Config.Certificate == "" {
|
||||
p.Config.Certificate = resolveEnvVar("${KUBERNETES_CERTIFICATE}", p.Config.Prefix, p.Config.Debug)
|
||||
}
|
||||
if p.Config.ServiceAccount == "" {
|
||||
p.Config.ServiceAccount = resolveEnvVar("${SERVICE_ACCOUNT}", p.Config.Prefix, p.Config.Debug)
|
||||
if p.Config.ServiceAccount == "" {
|
||||
p.Config.ServiceAccount = "helm"
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// getEnvVars will return [${TAG} {TAG} TAG]
|
||||
func getEnvVars(envvars string) [][]string {
|
||||
re := regexp.MustCompile(`\$(\{?(\w+)\}?)\.?`)
|
||||
extracted := re.FindAllStringSubmatch(envvars, -1)
|
||||
return extracted
|
||||
}
|
||||
|
||||
func resolveEnvVar(key string, prefix string, debug bool) string {
|
||||
envvars := getEnvVars(key)
|
||||
return replaceEnvvars(envvars, prefix, key, debug)
|
||||
}
|
||||
|
||||
func replaceEnvvars(envvars [][]string, prefix string, s string, debug bool) string {
|
||||
for _, envvar := range envvars {
|
||||
envvarName := envvar[0]
|
||||
envvarKey := envvar[2]
|
||||
prefixedKey := strings.ToUpper(prefix + "_" + envvarKey)
|
||||
envval := os.Getenv(prefixedKey)
|
||||
if debug {
|
||||
fmt.Printf("-ReplVar: %s => %s-- %s\n", prefixedKey, envvarKey, envval)
|
||||
}
|
||||
|
||||
if envval == "" {
|
||||
envval = os.Getenv(envvarKey)
|
||||
}
|
||||
|
||||
if strings.Contains(s, envvarName) {
|
||||
s = strings.Replace(s, envvarName, envval, -1)
|
||||
}
|
||||
}
|
||||
|
||||
return s
|
||||
}
|
||||
|
||||
// unQuote removes quotes if present
|
||||
func unQuote(s string) string {
|
||||
unquoted, err := strconv.Unquote(s)
|
||||
if err != nil {
|
||||
// ignore error and return original string
|
||||
return s
|
||||
}
|
||||
return unquoted
|
||||
}
|
||||
|
||||
func (p *Plugin) debugEnv() {
|
||||
// debug env vars
|
||||
for _, e := range os.Environ() {
|
||||
fmt.Println("-Var:--", e)
|
||||
}
|
||||
}
|
||||
|
||||
func (p *Plugin) debug() {
|
||||
fmt.Println(p)
|
||||
// debug plugin obj
|
||||
fmt.Printf("Api server: %s \n", p.Config.APIServer)
|
||||
fmt.Printf("Values: %s \n", p.Config.Values)
|
||||
fmt.Printf("Secrets: %s \n", p.Config.Secrets)
|
||||
fmt.Printf("Helm Repos: %s \n", p.Config.HelmRepos)
|
||||
fmt.Printf("ValuesFiles: %s \n", p.Config.ValuesFiles)
|
||||
kubeconfig, err := ioutil.ReadFile(KUBECONFIG)
|
||||
if err == nil {
|
||||
fmt.Println(string(kubeconfig))
|
||||
}
|
||||
|
||||
config, err := ioutil.ReadFile(p.Config.KubeConfig)
|
||||
if err == nil {
|
||||
fmt.Println(string(config))
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,498 @@
|
||||
package plugin
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestInitialiseKubeconfig(t *testing.T) {
|
||||
|
||||
plugin := Plugin{
|
||||
Config: Config{
|
||||
APIServer: "http://myapiserver",
|
||||
Token: "secret-token",
|
||||
Certificate: "my-cert-data",
|
||||
ServiceAccount: "default-account",
|
||||
HelmCommand: "",
|
||||
Namespace: "default",
|
||||
SkipTLSVerify: false, // if set the true with Certificate, this test will fail
|
||||
},
|
||||
}
|
||||
|
||||
configfile := "config3.test"
|
||||
initialiseKubeconfig(&plugin.Config, "../kubeconfig", configfile)
|
||||
data, err := ioutil.ReadFile(configfile)
|
||||
if err != nil {
|
||||
t.Errorf("Error reading file %v", err)
|
||||
}
|
||||
kubeConfigStr := string(data)
|
||||
|
||||
if !strings.Contains(kubeConfigStr, "secret-token") {
|
||||
t.Errorf("Kubeconfig doesn't render token")
|
||||
}
|
||||
if !strings.Contains(kubeConfigStr, "http://myapiserver") {
|
||||
t.Errorf("Kubeconfig doesn't render APIServer")
|
||||
}
|
||||
if !strings.Contains(kubeConfigStr, "default-account") {
|
||||
t.Errorf("Kubeconfig doesn't render serviceaccount")
|
||||
}
|
||||
if !strings.Contains(kubeConfigStr, "my-cert-data") {
|
||||
t.Errorf("Kubeconfig doesn't render certificate")
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetHelmCommandEmptyPushEvent(t *testing.T) {
|
||||
os.Setenv("DRONE_BUILD_EVENT", "push")
|
||||
plugin := &Plugin{
|
||||
Config: Config{
|
||||
APIServer: "http://myapiserver",
|
||||
Token: "secret-token",
|
||||
HelmCommand: "",
|
||||
Namespace: "default",
|
||||
SkipTLSVerify: true,
|
||||
Debug: true,
|
||||
DryRun: true,
|
||||
Chart: "./chart/test",
|
||||
Version: "1.2.3",
|
||||
Release: "test-release",
|
||||
Values: `"image.tag=v.0.1.0,nameOverride=my-over-app"`,
|
||||
Wait: true,
|
||||
ReuseValues: true,
|
||||
Timeout: "500",
|
||||
Force: true,
|
||||
},
|
||||
}
|
||||
setHelmCommand(plugin)
|
||||
res := strings.Join(plugin.command[:], " ")
|
||||
expected := "upgrade --install test-release ./chart/test --version 1.2.3 --set image.tag=v.0.1.0,nameOverride=my-over-app --namespace default --dry-run --debug --wait --reuse-values --timeout 500 --force"
|
||||
if res != expected {
|
||||
t.Errorf("Result is %s and we expected %s", res, expected)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetHelmCommandUpgrade(t *testing.T) {
|
||||
os.Setenv("DRONE_BUILD_EVENT", "push")
|
||||
plugin := &Plugin{
|
||||
Config: Config{
|
||||
APIServer: "http://myapiserver",
|
||||
Token: "secret-token",
|
||||
HelmCommand: "upgrade",
|
||||
Namespace: "default",
|
||||
SkipTLSVerify: true,
|
||||
Debug: true,
|
||||
DryRun: true,
|
||||
Chart: "./chart/test",
|
||||
Version: "1.2.3",
|
||||
Release: "test-release",
|
||||
Values: `"image.tag=v.0.1.0,nameOverride=my-over-app"`,
|
||||
Wait: true,
|
||||
ReuseValues: true,
|
||||
Timeout: "500",
|
||||
Force: true,
|
||||
},
|
||||
}
|
||||
setHelmCommand(plugin)
|
||||
res := strings.Join(plugin.command[:], " ")
|
||||
expected := "upgrade --install test-release ./chart/test --version 1.2.3 --set image.tag=v.0.1.0,nameOverride=my-over-app --namespace default --dry-run --debug --wait --reuse-values --timeout 500 --force"
|
||||
if res != expected {
|
||||
t.Errorf("Result is %s and we expected %s", res, expected)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetHelmDeleteCommand(t *testing.T) {
|
||||
os.Setenv("DRONE_BUILD_EVENT", "delete")
|
||||
plugin := &Plugin{
|
||||
Config: Config{
|
||||
APIServer: "http://myapiserver",
|
||||
Token: "secret-token",
|
||||
HelmCommand: "delete",
|
||||
Namespace: "default",
|
||||
SkipTLSVerify: true,
|
||||
Debug: true,
|
||||
Chart: "./chart/test",
|
||||
Release: "test-release",
|
||||
Values: "image.tag=v.0.1.0,nameOverride=my-over-app",
|
||||
Wait: true,
|
||||
},
|
||||
}
|
||||
setHelmCommand(plugin)
|
||||
res := strings.Join(plugin.command[:], " ")
|
||||
expected := "delete test-release"
|
||||
if res != expected {
|
||||
t.Errorf("Result is %s and we expected %s", res, expected)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetHelmDeleteCommandOverried(t *testing.T) {
|
||||
os.Setenv("DRONE_BUILD_EVENT", "deployment")
|
||||
plugin := &Plugin{
|
||||
Config: Config{
|
||||
APIServer: "http://myapiserver",
|
||||
Token: "secret-token",
|
||||
HelmCommand: "delete",
|
||||
Namespace: "default",
|
||||
TillerNs: "default-tiller-ns",
|
||||
SkipTLSVerify: true,
|
||||
Debug: true,
|
||||
DryRun: true,
|
||||
Chart: "./chart/test",
|
||||
Release: "test-release",
|
||||
Values: "image.tag=v.0.1.0,nameOverride=my-over-app",
|
||||
Wait: true,
|
||||
Purge: true,
|
||||
},
|
||||
}
|
||||
setHelmCommand(plugin)
|
||||
res := strings.Join(plugin.command[:], " ")
|
||||
expected := "delete test-release --tiller-namespace default-tiller-ns --dry-run --purge"
|
||||
if res != expected {
|
||||
t.Errorf("Result is %s and we expected %s", res, expected)
|
||||
}
|
||||
}
|
||||
|
||||
func TestResolveSecrets(t *testing.T) {
|
||||
|
||||
// Test resolving secrets from env
|
||||
testEnvs := []struct {
|
||||
prefix string
|
||||
tag string
|
||||
api string
|
||||
token string
|
||||
account string
|
||||
}{
|
||||
{prefix: "PROD", tag: "v0.1.1", api: "http://apiserver", token: "12345", account: "helm"},
|
||||
{prefix: "STAGING", tag: "12345678", api: "http://apiserver", token: "12345", account: "helm"},
|
||||
}
|
||||
for _, env := range testEnvs {
|
||||
envMap := map[string]string{
|
||||
"TAG": env.tag,
|
||||
"API_SERVER": env.api,
|
||||
"KUBERNETES_TOKEN": env.token,
|
||||
"SERVICE_ACCOUNT": env.account,
|
||||
}
|
||||
|
||||
for envKey, envValue := range envMap {
|
||||
os.Setenv(fmt.Sprintf("%s_%s", env.prefix, envKey), envValue)
|
||||
}
|
||||
|
||||
plugin := &Plugin{
|
||||
Config: Config{
|
||||
HelmCommand: "",
|
||||
Namespace: "default",
|
||||
SkipTLSVerify: true,
|
||||
Debug: true,
|
||||
DryRun: true,
|
||||
Chart: "./chart/test",
|
||||
Release: "test-release",
|
||||
Prefix: env.prefix,
|
||||
Values: "image.tag=$TAG,api=${API_SERVER},nameOverride=my-over-app,second.tag=${TAG}",
|
||||
},
|
||||
}
|
||||
|
||||
resolveSecrets(plugin)
|
||||
// test that the subsitution works
|
||||
fmt.Println(plugin.Config.Values)
|
||||
if !strings.Contains(plugin.Config.Values, env.tag) {
|
||||
t.Errorf("env var ${TAG} not resolved %s", env.tag)
|
||||
}
|
||||
if strings.Contains(plugin.Config.Values, "${TAG}") {
|
||||
t.Errorf("env var ${TAG} not resolved %s", env.tag)
|
||||
}
|
||||
|
||||
if plugin.Config.APIServer != env.api {
|
||||
t.Errorf("env var ${API_SERVER} not resolved %s", env.api)
|
||||
}
|
||||
if plugin.Config.Token != env.token {
|
||||
t.Errorf("env var ${KUBERNETES_TOKEN} not resolved %s", env.token)
|
||||
}
|
||||
if plugin.Config.ServiceAccount != env.account {
|
||||
t.Errorf("env var ${SERVICE_ACCOUNT} not resolved %s", env.account)
|
||||
}
|
||||
|
||||
// clean up
|
||||
for envKey := range envMap {
|
||||
os.Unsetenv(fmt.Sprintf("%s_%s", env.prefix, envKey))
|
||||
}
|
||||
}
|
||||
|
||||
// Test resolving provided values
|
||||
testInput := []struct {
|
||||
server string
|
||||
values string
|
||||
token string
|
||||
account string
|
||||
}{
|
||||
{server: "http://apiserver2", token: "123456", account: "helm2", values: "aval=test"},
|
||||
}
|
||||
for _, input := range testInput {
|
||||
plugin := &Plugin{
|
||||
Config: Config{
|
||||
APIServer: input.server,
|
||||
ServiceAccount: input.account,
|
||||
Token: input.token,
|
||||
Values: input.values,
|
||||
},
|
||||
}
|
||||
|
||||
resolveSecrets(plugin)
|
||||
if plugin.Config.APIServer != input.server {
|
||||
t.Errorf("failed to keep APIServer '%s' got '%s'", input.server, plugin.Config.APIServer)
|
||||
}
|
||||
if plugin.Config.ServiceAccount != input.account {
|
||||
t.Errorf("failed to keep ServiceAccount '%s' got '%s'", input.account, plugin.Config.ServiceAccount)
|
||||
}
|
||||
if plugin.Config.Token != input.token {
|
||||
t.Errorf("failed to keep Token '%s' got '%s'", input.token, plugin.Config.Token)
|
||||
}
|
||||
if plugin.Config.Values != input.values {
|
||||
t.Errorf("failed to keep Values '%s' got '%s'", input.values, plugin.Config.Values)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestDetHelmRepoAdd(t *testing.T) {
|
||||
plugin := &Plugin{
|
||||
Config: Config{
|
||||
HelmCommand: "",
|
||||
Namespace: "default",
|
||||
SkipTLSVerify: true,
|
||||
Debug: true,
|
||||
DryRun: true,
|
||||
Chart: "./chart/test",
|
||||
Release: "test-release",
|
||||
Prefix: "MY",
|
||||
Values: "image.tag=$TAG,api=${API_SERVER},nameOverride=my-over-app,second.tag=${TAG}",
|
||||
ClientOnly: true,
|
||||
HelmRepos: []string{
|
||||
`"r1=http://r1.example.com"`, //handle quoted strings
|
||||
`r2=http://r2.example.com`, //and unquoted strings
|
||||
},
|
||||
},
|
||||
}
|
||||
expected := []string{
|
||||
"repo add r1 http://r1.example.com",
|
||||
"repo add r2 http://r2.example.com",
|
||||
}
|
||||
|
||||
for i, r := range plugin.Config.HelmRepos {
|
||||
repos, err := doHelmRepoAdd(r)
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
result := strings.Join(repos, " ")
|
||||
if expected[i] != result {
|
||||
t.Errorf("Helm cannot add remote repositories - expected %q - got %q",
|
||||
expected[i],
|
||||
result,
|
||||
)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestDependencyUpdate(t *testing.T) {
|
||||
chart := "./chart"
|
||||
expected := "dependency update ./chart"
|
||||
result := strings.Join(doDependencyUpdate(chart), " ")
|
||||
|
||||
if expected != result {
|
||||
t.Errorf("Helm cannot update repositories - expected %q - got %q", expected, result)
|
||||
}
|
||||
}
|
||||
|
||||
func TestHelmAddRepositoryError(t *testing.T) {
|
||||
_, err := doHelmRepoAdd("drone-helm=bad://drone-helm.example.com:443/stable")
|
||||
if err == nil {
|
||||
t.Errorf("Expect to see error when repo URL is invalid")
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetEnvVars(t *testing.T) {
|
||||
|
||||
testText := "this should be ${TAG} now"
|
||||
result := getEnvVars(testText)
|
||||
if len(result) == 0 {
|
||||
t.Error("No envvar was found")
|
||||
}
|
||||
envvar := result[0]
|
||||
if !strings.Contains(envvar[2], "TAG") {
|
||||
t.Errorf("envvar not found in %s", testText)
|
||||
}
|
||||
}
|
||||
|
||||
func TestReplaceEnvvars(t *testing.T) {
|
||||
tag := "tagged"
|
||||
os.Setenv("MY_TAG", tag)
|
||||
prefix := "MY"
|
||||
testText := "this should be ${TAG} now ${TAG}"
|
||||
result := getEnvVars(testText)
|
||||
resolved := replaceEnvvars(result, prefix, testText, false)
|
||||
if !strings.Contains(resolved, tag) {
|
||||
t.Errorf("EnvVar MY_TAG no replaced by %s -- %s \n", tag, resolved)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSetHelpCommand(t *testing.T) {
|
||||
plugin := &Plugin{
|
||||
Config: Config{
|
||||
HelmCommand: "",
|
||||
Namespace: "default",
|
||||
SkipTLSVerify: true,
|
||||
Debug: true,
|
||||
DryRun: true,
|
||||
Chart: "./chart/test",
|
||||
Release: "test-release",
|
||||
Prefix: "MY",
|
||||
Values: "image.tag=$TAG,api=${API_SERVER},nameOverride=my-over-app,second.tag=${TAG}",
|
||||
},
|
||||
}
|
||||
setHelpCommand(plugin)
|
||||
if plugin.command == nil {
|
||||
t.Error("Helm help is not displayed")
|
||||
}
|
||||
}
|
||||
|
||||
func TestDetHelmInit(t *testing.T) {
|
||||
plugin := &Plugin{
|
||||
Config: Config{
|
||||
HelmCommand: "",
|
||||
Namespace: "default",
|
||||
SkipTLSVerify: true,
|
||||
Debug: true,
|
||||
DryRun: true,
|
||||
Chart: "./chart/test",
|
||||
Release: "test-release",
|
||||
Prefix: "MY",
|
||||
Values: "image.tag=$TAG,api=${API_SERVER},nameOverride=my-over-app,second.tag=${TAG}",
|
||||
TillerNs: "system-test",
|
||||
},
|
||||
}
|
||||
init := doHelmInit(plugin)
|
||||
result := strings.Join(init, " ")
|
||||
expected := "init --tiller-namespace " + plugin.Config.TillerNs
|
||||
|
||||
if expected != result {
|
||||
t.Error("Tiller not installed in proper namespace")
|
||||
}
|
||||
}
|
||||
|
||||
func TestDetHelmInitClient(t *testing.T) {
|
||||
plugin := &Plugin{
|
||||
Config: Config{
|
||||
HelmCommand: "",
|
||||
Namespace: "default",
|
||||
SkipTLSVerify: true,
|
||||
Debug: true,
|
||||
DryRun: true,
|
||||
Chart: "./chart/test",
|
||||
Release: "test-release",
|
||||
Prefix: "MY",
|
||||
Values: "image.tag=$TAG,api=${API_SERVER},nameOverride=my-over-app,second.tag=${TAG}",
|
||||
ClientOnly: true,
|
||||
},
|
||||
}
|
||||
init := doHelmInit(plugin)
|
||||
result := strings.Join(init, " ")
|
||||
expected := "init "
|
||||
if plugin.Config.ClientOnly {
|
||||
expected = expected + "--client-only"
|
||||
}
|
||||
|
||||
if expected != result {
|
||||
t.Error("Helm cannot init in client only")
|
||||
}
|
||||
}
|
||||
|
||||
func TestDetHelmInitUpgrade(t *testing.T) {
|
||||
plugin := &Plugin{
|
||||
Config: Config{
|
||||
HelmCommand: "",
|
||||
Namespace: "default",
|
||||
SkipTLSVerify: true,
|
||||
Debug: true,
|
||||
DryRun: true,
|
||||
Chart: "./chart/test",
|
||||
Release: "test-release",
|
||||
Prefix: "MY",
|
||||
Values: "image.tag=$TAG,api=${API_SERVER},nameOverride=my-over-app,second.tag=${TAG}",
|
||||
Upgrade: true,
|
||||
},
|
||||
}
|
||||
init := doHelmInit(plugin)
|
||||
result := strings.Join(init, " ")
|
||||
expected := "init "
|
||||
if plugin.Config.Upgrade {
|
||||
expected = expected + "--upgrade"
|
||||
}
|
||||
|
||||
if expected != result {
|
||||
t.Error("Helm cannot init in client only")
|
||||
}
|
||||
}
|
||||
|
||||
func TestDetHelmInitCanary(t *testing.T) {
|
||||
plugin := &Plugin{
|
||||
Config: Config{
|
||||
HelmCommand: "",
|
||||
Namespace: "default",
|
||||
SkipTLSVerify: true,
|
||||
Debug: true,
|
||||
DryRun: true,
|
||||
Chart: "./chart/test",
|
||||
Release: "test-release",
|
||||
Prefix: "MY",
|
||||
Values: "image.tag=$TAG,api=${API_SERVER},nameOverride=my-over-app,second.tag=${TAG}",
|
||||
CanaryImage: true,
|
||||
},
|
||||
}
|
||||
init := doHelmInit(plugin)
|
||||
result := strings.Join(init, " ")
|
||||
expected := "init "
|
||||
if plugin.Config.CanaryImage {
|
||||
expected = expected + "--canary-image"
|
||||
}
|
||||
|
||||
if expected != result {
|
||||
t.Error("Helm cannot init in client only")
|
||||
}
|
||||
}
|
||||
func TestResolveSecretsFallback(t *testing.T) {
|
||||
tag := "v0.1.1"
|
||||
api := "http://apiserver"
|
||||
os.Setenv("MY_TAG", tag)
|
||||
os.Setenv("MY_API_SERVER", api)
|
||||
os.Setenv("MY_TOKEN", "12345")
|
||||
os.Setenv("NOTTOKEN", "99999")
|
||||
|
||||
plugin := &Plugin{
|
||||
Config: Config{
|
||||
HelmCommand: "",
|
||||
Namespace: "default",
|
||||
SkipTLSVerify: true,
|
||||
Debug: true,
|
||||
DryRun: true,
|
||||
Chart: "./chart/test",
|
||||
Release: "test-release",
|
||||
Prefix: "MY",
|
||||
Values: "image.tag=$TAG,api=${API_SERVER},nottoken=${NOTTOKEN},nameOverride=my-over-app,second.tag=${TAG}",
|
||||
},
|
||||
}
|
||||
|
||||
resolveSecrets(plugin)
|
||||
// test that the subsitution works
|
||||
if !strings.Contains(plugin.Config.Values, tag) {
|
||||
t.Errorf("env var ${TAG} not resolved %s", tag)
|
||||
}
|
||||
if strings.Contains(plugin.Config.Values, "${TAG}") {
|
||||
t.Errorf("env var ${TAG} not resolved %s", tag)
|
||||
}
|
||||
|
||||
if plugin.Config.APIServer != api {
|
||||
t.Errorf("env var ${API_SERVER} not resolved %s", api)
|
||||
}
|
||||
if !strings.Contains(plugin.Config.Values, "99999") {
|
||||
t.Errorf("envar ${NOTTOKEN} has not been resolved to 99999, not using prefix")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user