Finished job creation

[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..4a65dc0 100644 (file)
--- a/dmaap-mediator-producer/internal/config/config_test.go
+++ b/dmaap-mediator-producer/internal/config/config_test.go
@@ -21,31 +21,71 @@
 package config
 
 import (
+       "bytes"
        "os"
        "reflect"
        "testing"
+
+       log "github.com/sirupsen/logrus"
+       "github.com/stretchr/testify/require"
 )
 
 func TestNew_envVarsSetConfigContainSetValues(t *testing.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_SUPERVISION_CALLBACK_HOST", "supervisionCallbackHost")
+       os.Setenv("INFO_PRODUCER_SUPERVISION_CALLBACK_PORT", "8095")
+       os.Setenv("INFO_JOB_CALLBACK_HOST", "jobCallbackHost")
+       os.Setenv("INFO_JOB_CALLBACK_PORT", "8096")
+       os.Setenv("INFO_COORD_ADDR", "infoCoordAddr")
+       t.Cleanup(func() {
+               os.Clearenv()
+       })
+       wantConfig := Config{
+               LogLevel:                            "Debug",
+               InfoProducerSupervisionCallbackHost: "supervisionCallbackHost",
+               InfoProducerSupervisionCallbackPort: 8095,
+               InfoJobCallbackHost:                 "jobCallbackHost",
+               InfoJobCallbackPort:                 8096,
+               InfoCoordinatorAddress:              "infoCoordAddr",
+       }
+       if got := New(); !reflect.DeepEqual(got, &wantConfig) {
+               t.Errorf("New() = %v, want %v", got, &wantConfig)
+       }
+}
+
+func TestNew_faultyIntValueSetConfigContainDefaultValueAndWarnInLog(t *testing.T) {
+       assertions := require.New(t)
+       var buf bytes.Buffer
+       log.SetOutput(&buf)
+
+       os.Setenv("INFO_PRODUCER_SUPERVISION_CALLBACK_PORT", "wrong")
+       t.Cleanup(func() {
+               log.SetOutput(os.Stderr)
+               os.Clearenv()
+       })
        wantConfig := Config{
-               LogLevel:               "Debug",
-               JobResultUri:           "testUrl",
-               InfoCoordinatorAddress: "testAddr",
+               LogLevel:                            "Info",
+               InfoProducerSupervisionCallbackHost: "",
+               InfoProducerSupervisionCallbackPort: 8085,
+               InfoJobCallbackHost:                 "",
+               InfoJobCallbackPort:                 8086,
+               InfoCoordinatorAddress:              "http://enrichmentservice:8083",
        }
        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_SUPERVISION_CALLBACK_PORT. Default value: 8085 will be used")
 }
 
 func TestNew_envVarsNotSetConfigContainDefaultValues(t *testing.T) {
        wantConfig := Config{
-               LogLevel:               "Info",
-               JobResultUri:           "",
-               InfoCoordinatorAddress: "http://enrichmentservice:8083",
+               LogLevel:                            "Info",
+               InfoProducerSupervisionCallbackHost: "",
+               InfoProducerSupervisionCallbackPort: 8085,
+               InfoJobCallbackHost:                 "",
+               InfoJobCallbackPort:                 8086,
+               InfoCoordinatorAddress:              "http://enrichmentservice:8083",
        }
        if got := New(); !reflect.DeepEqual(got, &wantConfig) {
                t.Errorf("New() = %v, want %v", got, &wantConfig)