+func TestCollectorConfiguration(t *testing.T) {
+ os.Setenv("VESMGR_PRICOLLECTOR_USER", "user123")
+ os.Setenv("VESMGR_PRICOLLECTOR_PASSWORD", "pass123")
+ os.Setenv("VESMGR_PRICOLLECTOR_PASSPHRASE", "phrase123")
+ os.Setenv("VESMGR_PRICOLLECTOR_ADDR", "1.2.3.4")
+ os.Setenv("VESMGR_PRICOLLECTOR_PORT", "1234")
+ os.Setenv("VESMGR_PRICOLLECTOR_SERVERROOT", "vescollector")
+ os.Setenv("VESMGR_PRICOLLECTOR_TOPIC", "sometopic")
+ os.Setenv("VESMGR_PRICOLLECTOR_SECURE", "true")
+
+ vesconf := basicVespaConf()
+ getCollectorConfiguration(&vesconf)
+
+ assert.Equal(t, "user123", vesconf.PrimaryCollector.User)
+ assert.Equal(t, "pass123", vesconf.PrimaryCollector.Password)
+ assert.Equal(t, "phrase123", vesconf.PrimaryCollector.PassPhrase)
+ assert.Equal(t, "1.2.3.4", vesconf.PrimaryCollector.FQDN)
+ assert.Equal(t, 1234, vesconf.PrimaryCollector.Port)
+ assert.Equal(t, "vescollector", vesconf.PrimaryCollector.ServerRoot)
+ assert.Equal(t, "sometopic", vesconf.PrimaryCollector.Topic)
+ assert.Equal(t, true, vesconf.PrimaryCollector.Secure)
+}
+
+func TestCollectorConfigurationWhenEnvironmentVariablesAreNotDefined(t *testing.T) {
+ os.Unsetenv("VESMGR_PRICOLLECTOR_USER")
+ os.Unsetenv("VESMGR_PRICOLLECTOR_PASSWORD")
+ os.Unsetenv("VESMGR_PRICOLLECTOR_PASSPHRASE")
+ os.Unsetenv("VESMGR_PRICOLLECTOR_ADDR")
+ os.Unsetenv("VESMGR_PRICOLLECTOR_PORT")
+ os.Unsetenv("VESMGR_PRICOLLECTOR_SERVERROOT")
+ os.Unsetenv("VESMGR_PRICOLLECTOR_TOPIC")
+ os.Unsetenv("VESMGR_PRICOLLECTOR_SECURE")
+
+ vesconf := basicVespaConf()
+ getCollectorConfiguration(&vesconf)
+
+ assert.Equal(t, "", vesconf.PrimaryCollector.User)
+ assert.Equal(t, "", vesconf.PrimaryCollector.Password)
+ assert.Equal(t, "", vesconf.PrimaryCollector.PassPhrase)
+ assert.Equal(t, "", vesconf.PrimaryCollector.FQDN)
+ assert.Equal(t, 8443, vesconf.PrimaryCollector.Port)
+ assert.Equal(t, "", vesconf.PrimaryCollector.ServerRoot)
+ assert.Equal(t, "", vesconf.PrimaryCollector.Topic)
+ assert.Equal(t, false, vesconf.PrimaryCollector.Secure)
+}
+
+func TestCollectorConfigurationWhenPrimaryCollectorPortIsNotInteger(t *testing.T) {
+ os.Setenv("VESMGR_PRICOLLECTOR_PORT", "abcd")
+ vesconf := basicVespaConf()
+ getCollectorConfiguration(&vesconf)
+ assert.Equal(t, 0, vesconf.PrimaryCollector.Port)
+}
+
+func TestCollectorConfigurationWhenPrimaryCollectorSecureIsNotTrueOrFalse(t *testing.T) {
+ os.Setenv("VESMGR_PRICOLLECTOR_SECURE", "foo")
+ vesconf := basicVespaConf()
+ getCollectorConfiguration(&vesconf)
+ assert.Equal(t, false, vesconf.PrimaryCollector.Secure)
+}
+
+func TestYamlGeneration(t *testing.T) {