src/helm_test.go

/*
==================================================================================
  Copyright (c) 2019 AT&T Intellectual Property.
  Copyright (c) 2019 Nokia

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.
==================================================================================
*/

package main

import (
    "fmt"
    "os"
    "os/exec"
	"strconv"
    "testing"
    "reflect"
    "github.com/spf13/viper"
    "io/ioutil"
    "path"
)

var helmStatusOutput = `
LAST DEPLOYED: Sat Mar  9 06:50:45 2019
NAMESPACE: default
STATUS: DEPLOYED

RESOURCES:
==> v1/Pod(related)
NAME                        READY  STATUS   RESTARTS  AGE
dummy-xapp-8984fc9fd-bkcbp  1/1    Running  0         55m
dummy-xapp-8984fc9fd-l6xch  1/1    Running  0         55m
dummy-xapp-8984fc9fd-pp4hg  1/1    Running  0         55m

==> v1/Service
NAME                         TYPE       CLUSTER-IP      EXTERNAL-IP  PORT(S)  AGE
dummy-xapp-dummy-xapp-chart  ClusterIP  10.102.184.212  <none>       80/TCP   55m

==> v1beta1/Deployment
NAME        READY  UP-TO-DATE  AVAILABLE  AGE
dummy-xapp  3/3    3           3          55m
`

var helListOutput = `Next: ""
Releases:
- AppVersion: "1.0"
  Chart: dummy-xapp-chart-0.1.0
  Name: dummy-xapp
  Namespace: default
  Revision: 1
  Status: DEPLOYED
  Updated: Mon Mar 11 06:55:05 2019
- AppVersion: "2.0"
  Chart: dummy-xapp-chart-0.1.0
  Name: dummy-xapp2
  Namespace: default
  Revision: 1
  Status: DEPLOYED
  Updated: Mon Mar 11 06:55:05 2019
- AppVersion: "1.0"
  Chart: appmgr-0.0.1
  Name: appmgr
  Namespace: default
  Revision: 1
  Status: DEPLOYED
  Updated: Sun Mar 24 07:17:00 2019`

var mockedExitStatus = 0
var mockedStdout string
var h = Helm{}

func fakeExecCommand(command string, args ...string) *exec.Cmd {
    cs := []string{"-test.run=TestExecCommandHelper", "--", command}
    cs = append(cs, args...)
	
	cmd := exec.Command(os.Args[0], cs...)
    es := strconv.Itoa(mockedExitStatus)
	cmd.Env = []string{"GO_WANT_HELPER_PROCESS=1", "STDOUT=" + mockedStdout, "EXIT_STATUS=" + es}
	
    return cmd
}

func TestExecCommandHelper(t *testing.T) {
    if os.Getenv("GO_WANT_HELPER_PROCESS") != "1" {
        return
    }

    fmt.Fprintf(os.Stdout, os.Getenv("STDOUT"))
    i, _ := strconv.Atoi(os.Getenv("EXIT_STATUS"))
    os.Exit(i)
}

func writeTestCreds() (err error) {

    // Write test entries to helm username and password files
    f, err := os.Create(viper.GetString("helm.helm-username-file"))
    if err != nil {
        return err
    }

    _, err = f.WriteString(viper.GetString("helm.secrets.username"))
    if err != nil {
        f.Close()
        return (err)
    }
    f.Close()

    f, err = os.Create(viper.GetString("helm.helm-password-file"))
    if err != nil {
        return err
    }

    _, err = f.WriteString(viper.GetString("helm.secrets.password"))
    if err != nil {
        f.Close()
        return (err)
    }
    f.Close()
    return
}

func TestHelmInit(t *testing.T) {
	mockedExitStatus = 0
    execCommand = fakeExecCommand
    defer func() { execCommand = exec.Command }()

    if err := writeTestCreds(); err != nil {
        t.Errorf("Writing test entries failed: %s", err)
        return
    }

    out, err := h.Init()
    if err != nil {
        t.Errorf("Helm init failed: %s %s", err, string(out))
    }
}

func TestHelmInstall(t *testing.T) {
    copyFile(t)
    mockedExitStatus = 0
	execCommand = fakeExecCommand
	mockedStdout = helmStatusOutput
    defer func() { execCommand = exec.Command }()

    xapp, err := h.Install("dummy-xapp")
    if err != nil {
        t.Errorf("Helm install failed: %v", err)
	}

    x := getXappData()
    xapp.Version = "1.0"

    if !reflect.DeepEqual(xapp, x) {
        t.Errorf("%v \n%v", xapp, x)
    }
}

func TestHelmStatus(t *testing.T) {
    copyFile(t)
    mockedExitStatus = 0
    mockedStdout = helmStatusOutput
    execCommand = fakeExecCommand
    defer func() { execCommand = exec.Command }()

    xapp, err := h.Status("dummy-xapp")
    if err != nil {
        t.Errorf("Helm status failed: %v", err)
	}

    x := getXappData()
    xapp.Version = "1.0"

	if !reflect.DeepEqual(xapp, x) {
        t.Errorf("%v \n%v", xapp, x)
    }
}

func TestHelmStatusAll(t *testing.T) {
    copyFile(t)
    mockedExitStatus = 0
    mockedStdout = helListOutput
    execCommand = fakeExecCommand
    defer func() { execCommand = exec.Command }()

    xapp, err := h.StatusAll()
    if err != nil {
        t.Errorf("Helm StatusAll failed: %v - %v", err, xapp)
	}

    // Todo: check the content
}

func TestHelmParseAllStatus(t *testing.T) {
    copyFile(t)
    mockedExitStatus = 0
    mockedStdout = helListOutput
    execCommand = fakeExecCommand
    defer func() { execCommand = exec.Command }()

    xapp, err := h.parseAllStatus([]string{"dummy-xapp", "dummy-xapp2"})
    if err != nil {
        t.Errorf("Helm parseAllStatus failed: %v - %v", err, xapp)
	}

    // Todo: check the content
}

func TestHelmDelete(t *testing.T) {
    copyFile(t)
    mockedExitStatus = 0
    mockedStdout = helListOutput
    execCommand = fakeExecCommand
    defer func() { execCommand = exec.Command }()

    xapp, err := h.Delete("dummy-xapp")
    if err != nil {
        t.Errorf("Helm delete failed: %v - %v", err, xapp)
	}

    // Todo: check the content
}

func TestHelmLists(t *testing.T) {
    mockedExitStatus = 0
    mockedStdout = helListOutput
    execCommand = fakeExecCommand
    defer func() { execCommand = exec.Command }()

    names, err := h.List()
    if err != nil {
        t.Errorf("Helm status failed: %v", err)
	}

    if !reflect.DeepEqual(names, []string{"dummy-xapp", "dummy-xapp2"}) {
        t.Errorf("Helm status failed: %v", err)
    }
}

func getXappData() (x Xapp) {
    x = generateXapp("dummy-xapp", "deployed", "1.0", "dummy-xapp-8984fc9fd-bkcbp", "running", "10.102.184.212", "80")
    x.Instances = append(x.Instances, x.Instances[0])
    x.Instances = append(x.Instances, x.Instances[0])
    x.Instances[1].Name = "dummy-xapp-8984fc9fd-l6xch"
    x.Instances[2].Name = "dummy-xapp-8984fc9fd-pp4hg"

    return x
}


func copyFile(t *testing.T) {
    tarDir := path.Join(viper.GetString("xapp.tarDir"), "dummy-xapp")
    err := os.MkdirAll(tarDir, 0777)
    if err != nil {
         t.Errorf("%v", err)
    }

    data, err := ioutil.ReadFile("../config/msg_type.yaml")
    if err != nil {
         t.Errorf("%v", err)
    }

    _ = ioutil.WriteFile(path.Join(tarDir, "msg_type.yaml"), data, 0644)
    if err != nil {
         t.Errorf("%v", err)
    }
}