Merge "Update helm yaml file to adapt the changes of mrstub"

[nonrtric.git] / dmaap-mediator-producer / internal / config / config_test.go

diff --git a/dmaap-mediator-producer/internal/config/config_test.go b/dmaap-mediator-producer/internal/config/config_test.go
index f0106d0..90d3c03 100644 (file)
--- a/dmaap-mediator-producer/internal/config/config_test.go
+++ b/dmaap-mediator-producer/internal/config/config_test.go
@@ -21,33 +21,91 @@
 package config
 
 import (
+       "bytes"
        "os"
        "reflect"
        "testing"
+
+       log "github.com/sirupsen/logrus"
+       "github.com/stretchr/testify/require"
 )
 
 func TestNew_envVarsSetConfigContainSetValues(t *testing.T) {
+       assertions := require.New(t)
        os.Setenv("LOG_LEVEL", "Debug")
-       os.Setenv("JOB_RESULT_URI", "testUrl")
-       os.Setenv("INFO_COORD_ADDR", "testAddr")
-       defer os.Clearenv()
+       os.Setenv("INFO_PRODUCER_HOST", "producerHost")
+       os.Setenv("INFO_PRODUCER_PORT", "8095")
+       os.Setenv("INFO_COORD_ADDR", "infoCoordAddr")
+       os.Setenv("DMAAP_MR_ADDR", "mrHost:3908")
+       os.Setenv("PRODUCER_CERT_PATH", "cert")
+       os.Setenv("PRODUCER_KEY_PATH", "key")
+       t.Cleanup(func() {
+               os.Clearenv()
+       })
        wantConfig := Config{
-               LogLevel:               "Debug",
-               JobResultUri:           "testUrl",
-               InfoCoordinatorAddress: "testAddr",
-       }
-       if got := New(); !reflect.DeepEqual(got, &wantConfig) {
-               t.Errorf("New() = %v, want %v", got, &wantConfig)
+               LogLevel:               log.DebugLevel,
+               InfoProducerHost:       "producerHost",
+               InfoProducerPort:       8095,
+               InfoCoordinatorAddress: "infoCoordAddr",
+               DMaaPMRAddress:         "mrHost:3908",
+               ProducerCertPath:       "cert",
+               ProducerKeyPath:        "key",
        }
+       got := New()
+
+       assertions.Equal(&wantConfig, got)
 }
 
-func TestNew_envVarsNotSetConfigContainDefaultValues(t *testing.T) {
+func TestNew_faultyIntValueSetConfigContainDefaultValueAndWarnInLog(t *testing.T) {
+       assertions := require.New(t)
+       var buf bytes.Buffer
+       log.SetOutput(&buf)
+
+       os.Setenv("INFO_PRODUCER_PORT", "wrong")
+       t.Cleanup(func() {
+               log.SetOutput(os.Stderr)
+               os.Clearenv()
+       })
        wantConfig := Config{
-               LogLevel:               "Info",
-               JobResultUri:           "",
-               InfoCoordinatorAddress: "http://enrichmentservice:8083",
+               LogLevel:               log.InfoLevel,
+               InfoProducerHost:       "",
+               InfoProducerPort:       8085,
+               InfoCoordinatorAddress: "https://enrichmentservice:8434",
+               DMaaPMRAddress:         "https://message-router.onap:3905",
+               ProducerCertPath:       "security/producer.crt",
+               ProducerKeyPath:        "security/producer.key",
        }
        if got := New(); !reflect.DeepEqual(got, &wantConfig) {
                t.Errorf("New() = %v, want %v", got, &wantConfig)
        }
+       logString := buf.String()
+       assertions.Contains(logString, "Invalid int value: wrong for variable: INFO_PRODUCER_PORT. Default value: 8085 will be used")
+}
+
+func TestNew_envFaultyLogLevelConfigContainDefaultValues(t *testing.T) {
+       assertions := require.New(t)
+       var buf bytes.Buffer
+       log.SetOutput(&buf)
+
+       os.Setenv("LOG_LEVEL", "wrong")
+       t.Cleanup(func() {
+               log.SetOutput(os.Stderr)
+               os.Clearenv()
+       })
+
+       wantConfig := Config{
+               LogLevel:               log.InfoLevel,
+               InfoProducerHost:       "",
+               InfoProducerPort:       8085,
+               InfoCoordinatorAddress: "https://enrichmentservice:8434",
+               DMaaPMRAddress:         "https://message-router.onap:3905",
+               ProducerCertPath:       "security/producer.crt",
+               ProducerKeyPath:        "security/producer.key",
+       }
+
+       got := New()
+
+       assertions.Equal(&wantConfig, got)
+       logString := buf.String()
+       assertions.Contains(logString, "Invalid log level: wrong. Log level will be Info!")
 }