cmd/vesmgr/vesmgr_test.go

   1 /*
   2  *  Copyright (c) 2019 AT&T Intellectual Property.
   3  *  Copyright (c) 2018-2019 Nokia.
   4  *
   5  *  Licensed under the Apache License, Version 2.0 (the "License");
   6  *  you may not use this file except in compliance with the License.
   7  *  You may obtain a copy of the License at
   8  *
   9  *     http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  *  Unless required by applicable law or agreed to in writing, software
  12  *  distributed under the License is distributed on an "AS IS" BASIS,
  13  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  *  See the License for the specific language governing permissions and
  15  *  limitations under the License.
  16  */
  17
  18 package main
  19
  20 import (
  21         "os"
  22         "testing"
  23
  24         "github.com/stretchr/testify/assert"
  25 )
  26
  27 func init() {
  28         vesagent.name = "echo" // no need to run real ves-agent
  29         logger.MdcAdd("Testvesmgr", "0.0.1")
  30         os.Setenv("VESMGR_HB_INTERVAL", "30s")
  31         os.Setenv("VESMGR_MEAS_INTERVAL", "30s")
  32         os.Setenv("VESMGR_PRICOLLECTOR_ADDR", "127.1.1.1")
  33         os.Setenv("VESMGR_PRICOLLECTOR_PORT", "8443")
  34         os.Setenv("VESMGR_PROMETHEUS_ADDR", "http://localhost:9090")
  35 }
  36
  37 func TestStartVesagent(t *testing.T) {
  38         assert.Equal(t, 0, vesagent.Pid)
  39         ch := startVesagent()
  40         assert.NotEqual(t, 0, vesagent.Pid)
  41         t.Logf("VES agent pid = %d", vesagent.Pid)
  42         vesagent.Pid = 0
  43         err := <-ch
  44         assert.Nil(t, err)
  45 }
  46
  47 func TestStartVesagentFails(t *testing.T) {
  48
  49         vesagent.name = "Not-ves-agent"
  50         assert.Equal(t, 0, vesagent.Pid)
  51         ch := startVesagent()
  52         err := <-ch
  53         assert.NotNil(t, err)
  54         assert.Equal(t, 0, vesagent.Pid)
  55         vesagent.name = "ves-agent"
  56 }