discovering
diff --git a/nodes/fuel.py b/nodes/fuel.py
index 76dedcd..c828793 100644
--- a/nodes/fuel.py
+++ b/nodes/fuel.py
@@ -1,10 +1,17 @@
import node
import fuel_rest_api
+import logging
-def discover_fuel_nodes(root_url, credentials):
+logger = logging.getLogger("io-perf-tool")
+
+
+def discover_fuel_nodes(root_url, credentials, roles):
"""Discover Fuel nodes"""
connection = fuel_rest_api.KeystoneAuth(root_url, credentials)
fi = fuel_rest_api.FuelInfo(connection)
-
- return [node.Node(n.ip, n.get_roles()) for n in fi.nodes]
\ No newline at end of file
+ nodes = []
+ for role in roles:
+ nodes.extend(getattr(fi.nodes, role))
+ logger.debug("Found %s fuel nodes" % len(fi.nodes))
+ return [node.Node(n.ip, n.get_roles()) for n in nodes]
\ No newline at end of file