add helper methods to rspec helper

[sfa.git] / sfa / util / rspecHelper.py

diff --git a/sfa/util/rspecHelper.py b/sfa/util/rspecHelper.py
index 8d00239..a943190 100755 (executable)
--- a/sfa/util/rspecHelper.py
+++ b/sfa/util/rspecHelper.py
@@ -52,6 +52,12 @@ class RSpec:
         result = self.rspec.xpath("./network/site/node/hostname/text()")
         return result
 
+    def get_network_list(self):
+        return self.rspec.xpath("./network[@name]/@name")
+
+    def get_nodes_from_network(self, network):
+        return self.rspec.xpath("./network[@name='%s']/site/node/hostname/text()" % network)
+
     def get_sliver_list(self):
         result = self.rspec.xpath("./network/site/node[sliver]/hostname/text()")
         return result