diff --git a/locust_influxdb_listener/__init__.py b/locust_influxdb_listener/__init__.py index 81d1a07..ed26a6e 100644 --- a/locust_influxdb_listener/__init__.py +++ b/locust_influxdb_listener/__init__.py @@ -47,8 +47,8 @@ def __init__( :param verify_ssl: Verify SSL certificates (only if SSL is enabled). :param additional_tags: Additional tags to include in globally for all data points. """ - self.host = host if host else influx_host # Renamed from influx_host - self.port = port if port else influx_port # Renamed from influx_port + self.host = influx_host if influx_host else host # Renamed from influx_host + self.port = influx_port if influx_port else port # Renamed from influx_port self.user = user self.pwd = pwd self.database = database @@ -208,7 +208,7 @@ def __listen_for_locust_errors(self, node_id, user_instance, exception: Exceptio """ time = datetime.utcnow() - tags = {**{'exception_tag': repr(exception)}, **self.tags} + tags = {**{'exception_tag': repr(exception)}, **self.additional_tags} fields = { 'node_id': node_id, 'user_instance': repr(user_instance), diff --git a/tests/test_initializer.py b/tests/test_initializer.py index 718f4a8..b31e1d5 100644 --- a/tests/test_initializer.py +++ b/tests/test_initializer.py @@ -35,8 +35,8 @@ def test_deprecated_attributes(self): influx_port=8000, ) - self.assertEqual(influx_settings.host, 'localhost') - self.assertEqual(influx_settings.port, 8086) + self.assertEqual(influx_settings.host, 'this_host') + self.assertEqual(influx_settings.port, 8000) if __name__ == '__main__':