cmd/vesmgr/config_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  package main
  18
  19  import (
  20         "testing"
  21         "time"
  22         "bytes"
  23         "github.com/stretchr/testify/assert"
  24         "gopkg.in/yaml.v2"
  25  )
  26
  27  func testBaseConf(t *testing.T, vesconf VESAgentConfiguration) {
  28         assert.Equal(t, "/tmp/data", vesconf.DataDir)
  29         assert.False(t, vesconf.Debug)
  30         assert.Equal(t, vesconf.Event.MaxMissed, 2)
  31         assert.Equal(t, vesconf.Event.RetryInterval, time.Second*5)
  32         assert.Equal(t, vesconf.Measurement.Prometheus.KeepAlive, time.Second*30)
  33  }
  34
  35  func TestBasicConfigContainsCorrectValues(t *testing.T) {
  36         vesconf := basicVespaConf()
  37         testBaseConf(t, vesconf)
  38  }
  39
  40  func TestYamlGeneratio(t *testing.T) {
  41         buffer := new(bytes.Buffer)
  42         createVespaConfig(buffer)
  43         var vesconf VESAgentConfiguration
  44         err := yaml.Unmarshal(buffer.Bytes(), &vesconf)
  45         assert.Nil(t, err)
  46         testBaseConf(t, vesconf)
  47  }