pkg/helm/helm_test.go

   1 /*
   2 ==================================================================================
   3   Copyright (c) 2019 AT&T Intellectual Property.
   4   Copyright (c) 2019 Nokia
   5
   6    Licensed under the Apache License, Version 2.0 (the "License");
   7    you may not use this file except in compliance with the License.
   8    You may obtain a copy of the License at
   9
  10        http://www.apache.org/licenses/LICENSE-2.0
  11
  12    Unless required by applicable law or agreed to in writing, software
  13    distributed under the License is distributed on an "AS IS" BASIS,
  14    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15    See the License for the specific language governing permissions and
  16    limitations under the License.
  17 ==================================================================================
  18 */
  19
  20 package helm
  21
  22 import (
  23         "os"
  24         "reflect"
  25         "strconv"
  26         "testing"
  27
  28         "gerrit.oran-osc.org/r/ric-plt/appmgr/pkg/appmgr"
  29         "gerrit.oran-osc.org/r/ric-plt/appmgr/pkg/models"
  30         "gerrit.oran-osc.org/r/ric-plt/appmgr/pkg/util"
  31 )
  32
  33 var helmStatusOutput = `
  34 LAST DEPLOYED: Sat Mar  9 06:50:45 2019
  35 NAMESPACE: default
  36 STATUS: DEPLOYED
  37
  38 RESOURCES:
  39 ==> v1/Pod(related)
  40 NAME                        READY  STATUS   RESTARTS  AGE
  41 dummy-xapp-8984fc9fd-bkcbp  1/1    Running  0         55m
  42 dummy-xapp-8984fc9fd-l6xch  1/1    Running  0         55m
  43 dummy-xapp-8984fc9fd-pp4hg  1/1    Running  0         55m
  44
  45 ==> v1/Service
  46 NAME                         TYPE       CLUSTER-IP      EXTERNAL-IP  PORT(S)  AGE
  47 dummy-xapp-dummy-xapp-chart  ClusterIP  10.102.184.212  <none>       80/TCP   55m
  48
  49 ==> v1beta1/Deployment
  50 NAME        READY  UP-TO-DATE  AVAILABLE  AGE
  51 dummy-xapp  3/3    3           3          55m
  52 `
  53
  54 var helListOutput = `Next: ""
  55 Releases:
  56 - AppVersion: "1.0"
  57   Chart: dummy-xapp-chart-0.1.0
  58   Name: dummy-xapp
  59   Namespace: default
  60   Revision: 1
  61   Status: DEPLOYED
  62   Updated: Mon Mar 11 06:55:05 2019
  63 - AppVersion: "2.0"
  64   Chart: dummy-xapp-chart-0.1.0
  65   Name: dummy-xapp2
  66   Namespace: default
  67   Revision: 1
  68   Status: DEPLOYED
  69   Updated: Mon Mar 11 06:55:05 2019
  70 - AppVersion: "1.0"
  71   Chart: appmgr-0.0.1
  72   Name: appmgr
  73   Namespace: default
  74   Revision: 1
  75   Status: DEPLOYED
  76   Updated: Sun Mar 24 07:17:00 2019`
  77
  78 // Test cases
  79 func TestMain(m *testing.M) {
  80         appmgr.Init()
  81         appmgr.Logger.SetLevel(0)
  82
  83         code := m.Run()
  84         os.Exit(code)
  85 }
  86
  87 func TestHelmStatus(t *testing.T) {
  88         //NewHelm().SetCM(&ConfigMap{})
  89         util.KubectlExec = func(args string) (out []byte, err error) {
  90                 return []byte("10.102.184.212"), nil
  91         }
  92         xapp, err := NewHelm().ParseStatus("dummy-xapp", helmStatusOutput)
  93         if err != nil {
  94                 t.Errorf("Helm install failed: %v", err)
  95         }
  96         x := getXappData()
  97         xapp.Version = "1.0"
  98
  99         if *x.Name != *xapp.Name || x.Status != xapp.Status || x.Version != xapp.Version {
 100                 t.Errorf("\n%v \n%v", *xapp.Name, *x.Name)
 101         }
 102
 103         if *x.Instances[0].Name != *xapp.Instances[0].Name || x.Instances[0].Status != xapp.Instances[0].Status {
 104                 t.Errorf("\n1:%v 2:%v", *x.Instances[0].Name, *xapp.Instances[0].Name)
 105         }
 106
 107         if x.Instances[0].IP != xapp.Instances[0].IP || x.Instances[0].Port != xapp.Instances[0].Port {
 108                 t.Errorf("\n1:%v 2:%v", x.Instances[0].IP, xapp.Instances[0].IP)
 109         }
 110 }
 111
 112 func TestHelmLists(t *testing.T) {
 113         names, err := NewHelm().GetNames(helListOutput)
 114         if err != nil {
 115                 t.Errorf("Helm status failed: %v", err)
 116         }
 117
 118         if !reflect.DeepEqual(names, []string{"dummy-xapp", "dummy-xapp2"}) {
 119                 t.Errorf("Helm status failed: %v", err)
 120         }
 121 }
 122
 123 func TestAddTillerEnv(t *testing.T) {
 124         if NewHelm().AddTillerEnv() != nil {
 125                 t.Errorf("TestAddTillerEnv failed!")
 126         }
 127 }
 128
 129 func TestGetInstallArgs(t *testing.T) {
 130         name := "dummy-xapp"
 131         x := models.XappDescriptor{XappName: &name, Namespace: "ricxapp"}
 132
 133         expectedArgs := "install helm-repo/dummy-xapp  --namespace=ricxapp --name=dummy-xapp"
 134         if args := NewHelm().GetInstallArgs(x, false); args != expectedArgs {
 135                 t.Errorf("TestGetInstallArgs failed: expected %v, got %v", expectedArgs, args)
 136         }
 137
 138         x.HelmVersion = "1.2.3"
 139         expectedArgs = "install helm-repo/dummy-xapp  --namespace=ricxapp --version=1.2.3 --name=dummy-xapp"
 140         if args := NewHelm().GetInstallArgs(x, false); args != expectedArgs {
 141                 t.Errorf("TestGetInstallArgs failed: expected %v, got %v", expectedArgs, args)
 142         }
 143
 144         x.ReleaseName = "ueec-xapp"
 145         expectedArgs = "install helm-repo/dummy-xapp  --namespace=ricxapp --version=1.2.3 --name=ueec-xapp"
 146         if args := NewHelm().GetInstallArgs(x, false); args != expectedArgs {
 147                 t.Errorf("TestGetInstallArgs failed: expected %v, got %v", expectedArgs, args)
 148         }
 149 }
 150
 151 func getXappData() (x models.Xapp) {
 152         //name1 := "dummy-xapp-8984fc9fd-l6xch"
 153         //name2 := "dummy-xapp-8984fc9fd-pp4hg"
 154         x = generateXapp("dummy-xapp", "deployed", "1.0", "dummy-xapp-8984fc9fd-bkcbp", "running", "service-ricxapp-dummy-xapp-rmr.ricxapp", "4560")
 155         //x.Instances = append(x.Instances, x.Instances[0])
 156         //x.Instances = append(x.Instances, x.Instances[0])
 157         //x.Instances[1].Name = &name1
 158         //x.Instances[2].Name = &name2
 159
 160         return x
 161 }
 162
 163 func generateXapp(name, status, ver, iname, istatus, ip, port string) (x models.Xapp) {
 164         x.Name = &name
 165         x.Status = status
 166         x.Version = ver
 167         p, _ := strconv.Atoi(port)
 168         var msgs appmgr.MessageTypes
 169
 170         instance := &models.XappInstance{
 171                 Name:       &iname,
 172                 Status:     istatus,
 173                 IP:         ip,
 174                 Port:       int64(p),
 175                 TxMessages: msgs.TxMessages,
 176                 RxMessages: msgs.RxMessages,
 177         }
 178         x.Instances = append(x.Instances, instance)
 179
 180         return
 181 }