Fix: import stomp in the function scope (elastic#14703)

mtojek · Nov 26, 2019 · 80b20e2 · 80b20e2
1 parent 1343b0e
commit 80b20e2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/x-pack/metricbeat/tests/system/test_activemq.py b/x-pack/metricbeat/tests/system/test_activemq.py
@@ -2,7 +2,6 @@
 import string
 import sys
 
-import stomp
 import unittest
 
 from xpack_metricbeat import XPackTest
@@ -39,12 +38,14 @@ def destination_metrics_collected(self, destination_type, destination_name):
         return False
 
     def verify_destination_metrics_collection(self, destination_type):
+        from stomp import Connection
+
         self.render_config_template(modules=[self.get_activemq_module_config(destination_type)])
         proc = self.start_beat(home=self.beat_path)
 
         destination_name = ''.join(random.choice(string.ascii_lowercase) for i in range(10))
 
-        conn = stomp.Connection([self.get_stomp_host_port()])
+        conn = Connection([self.get_stomp_host_port()])
         conn.start()
         conn.connect(wait=True)
         conn.send('/{}/{}'.format(destination_type, destination_name), 'first message')