Add new parameters to the tests

influxdata · Jun 12, 2024 · c13996f · c13996f
1 parent 76e8e83
commit c13996f
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/plugins/inputs/mqtt_consumer/mqtt_consumer_test.go b/plugins/inputs/mqtt_consumer/mqtt_consumer_test.go
@@ -619,8 +619,10 @@ func TestIntegration(t *testing.T) {
 	plugin := &MQTTConsumer{
 		Servers:                []string{url},
 		Topics:                 []string{topic},
-		ConnectionTimeout:      config.Duration(5 * time.Second),
 		MaxUndeliveredMessages: defaultMaxUndeliveredMessages,
+		ConnectionTimeout:      config.Duration(5 * time.Second),
+		KeepAliveInterval:      config.Duration(1 * time.Second),
+		PingTimeout:            config.Duration(100 * time.Millisecond),
 		Log:                    testutil.Logger{Name: "mqtt-integration-test"},
 		clientFactory:          factory,
 	}
@@ -706,8 +708,10 @@ func TestStartupErrorBehaviorErrorIntegration(t *testing.T) {
 	plugin := &MQTTConsumer{
 		Servers:                []string{url},
 		Topics:                 []string{topic},
-		ConnectionTimeout:      config.Duration(5 * time.Second),
 		MaxUndeliveredMessages: defaultMaxUndeliveredMessages,
+		ConnectionTimeout:      config.Duration(5 * time.Second),
+		KeepAliveInterval:      config.Duration(1 * time.Second),
+		PingTimeout:            config.Duration(100 * time.Millisecond),
 		Log:                    testutil.Logger{Name: "mqtt-integration-test"},
 		clientFactory:          factory,
 	}
@@ -763,8 +767,10 @@ func TestStartupErrorBehaviorIgnoreIntegration(t *testing.T) {
 	plugin := &MQTTConsumer{
 		Servers:                []string{url},
 		Topics:                 []string{topic},
-		ConnectionTimeout:      config.Duration(5 * time.Second),
 		MaxUndeliveredMessages: defaultMaxUndeliveredMessages,
+		ConnectionTimeout:      config.Duration(5 * time.Second),
+		KeepAliveInterval:      config.Duration(1 * time.Second),
+		PingTimeout:            config.Duration(100 * time.Millisecond),
 		Log:                    testutil.Logger{Name: "mqtt-integration-test"},
 		clientFactory:          factory,
 	}
@@ -826,8 +832,10 @@ func TestStartupErrorBehaviorRetryIntegration(t *testing.T) {
 	plugin := &MQTTConsumer{
 		Servers:                []string{url},
 		Topics:                 []string{topic},
-		ConnectionTimeout:      config.Duration(5 * time.Second),
 		MaxUndeliveredMessages: defaultMaxUndeliveredMessages,
+		ConnectionTimeout:      config.Duration(5 * time.Second),
+		KeepAliveInterval:      config.Duration(1 * time.Second),
+		PingTimeout:            config.Duration(100 * time.Millisecond),
 		Log:                    testutil.Logger{Name: "mqtt-integration-test"},
 		clientFactory:          factory,
 	}