Fix failing collect test

prestodb · Dec 24, 2015 · f5f9e7d · f5f9e7d
1 parent 537d44d
commit f5f9e7d
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/product/test_collect.py b/tests/product/test_collect.py
@@ -200,9 +200,7 @@ def test_collect_system_info_server_stopped(self):
         message = '\nFatal error: [%s] Unable to access node ' \
                   'information. Please check that server is up with ' \
                   'command: server status\n\nAborting.\n'
-        expected = ''
-        for host in self.cluster.all_internal_hosts():
-            expected += message % host
+        expected = message % self.cluster.internal_master
         self.assertEqualIgnoringOrder(actual, expected)
 
     def _add_custom_log_location(self, new_log_location):