A new test to build a tunnel between two local nodes. Simpler.

protogeni/test/loctuntest.py

+#! /usr/bin/env python
+#
+# GENIPUBLIC-COPYRIGHT
+# Copyright (c) 2008-2009 University of Utah and the Flux Group.
+# All rights reserved.
+# 
+# Permission to use, copy, modify and distribute this software is hereby
+# granted provided that (1) source code retains these copyright, permission,
+# and disclaimer notices, and (2) redistributions including binaries
+# reproduce the notices in supporting documentation.
+#
+# THE UNIVERSITY OF UTAH ALLOWS FREE USE OF THIS SOFTWARE IN ITS "AS IS"
+# CONDITION.  THE UNIVERSITY OF UTAH DISCLAIMS ANY LIABILITY OF ANY KIND
+# FOR ANY DAMAGES WHATSOEVER RESULTING FROM THE USE OF THIS SOFTWARE.
+#
+
+#
+#
+import sys
+import pwd
+import getopt
+import os
+import re
+import xmlrpclib
+import urllib
+from xml.sax.handler import ContentHandler
+import xml.sax
+import string
+from M2Crypto import X509
+
+ACCEPTSLICENAME=1
+
+execfile( "test-common.py" )
+
+class findElement(ContentHandler):
+    name       = None
+    value      = None
+    string     = None
+    attributes = None
+    data       = None
+    
+    def __init__(self, name, stuff):
+        self.name = name
+        xml.sax.parseString(stuff, self)
+        pass
+    def startElement(self, name, attrs):
+        if self.name == name:
+            self.data = []
+            self.attributes = attrs
+        elif self.data != None:
+            self.data.append("<" + name + ">")
+            pass
+        pass
+    def characters(self, content):
+        if self.data != None:
+            self.data.append(content)
+            pass
+        pass
+    def endElement(self, name):
+        if self.name == name:
+            self.value  = string.join(self.data, "");
+            self.string = "<" + name + ">" + self.value + "</" + name + ">"
+            self.data   = None;
+        elif self.data != None:
+            self.data.append("</" + name + ">")
+            pass
+        pass
+    pass
+
+#
+# Get a credential for myself, that allows me to do things at the SA.
+#
+mycredential = get_self_credential()
+print "Got my SA credential"
+
+#
+# Lookup slice.
+#
+params = {}
+params["credential"] = mycredential
+params["type"]       = "Slice"
+params["hrn"]        = SLICENAME
+rval,response = do_method("sa", "Resolve", params)
+if rval:
+    #
+    # Create a slice. 
+    #
+    print "Creating new slice called " + SLICENAME
+    params = {}
+    params["credential"] = mycredential
+    params["type"]       = "Slice"
+    params["hrn"]        = SLICENAME
+    rval,response = do_method("sa", "Register", params)
+    if rval:
+        Fatal("Could not create new slice")
+        pass
+    myslice = response["value"]
+    print "New slice created"
+    pass
+else:
+    #
+    # Get the slice credential.
+    #
+    print "Asking for slice credential for " + SLICENAME
+    myslice = response["value"]
+    myslice = get_slice_credential( myslice, mycredential )
+    print "Got the slice credential"
+    pass
+
+#
+# Get a ticket from the local CM.
+#
+rspec = "<rspec xmlns=\"http://protogeni.net/resources/rspec/0.1\"> " +\
+        " <node uuid=\"*\" " +\
+        "       nickname=\"geni1\" "+\
+        "       virtualization_type=\"emulab-vnode\"> " +\
+        " </node>" +\
+        " <node uuid=\"*\" " +\
+        "       nickname=\"geni2\" "+\
+        "       virtualization_type=\"emulab-vnode\"> " +\
+        " </node>" +\
+        " <link name=\"link0\" nickname=\"link0\" link_type=\"tunnel\"> " +\
+        "  <linkendpoints nickname=\"destination_interface\" " +\
+        "            tunnel_ip=\"192.168.1.1\" " +\
+        "            node_nickname=\"geni1\" " +\
+        "            node_uuid=\"*\" /> " +\
+        "  <linkendpoints nickname=\"source_interface\" " +\
+        "            tunnel_ip=\"192.168.1.2\" " +\
+        "            node_nickname=\"geni2\" " +\
+        "            node_uuid=\"*\" /> " +\
+        "</rspec>"
+
+params = {}
+params["credential"] = myslice
+params["rspec"]      = rspec
+rval,response = do_method("cm", "GetTicket", params)
+if rval:
+    Fatal("Could not get ticket")
+    pass
+ticket = response["value"]
+print "Got a ticket from CM"
+
+#
+# Create the sliver
+#
+params = {}
+params["ticket"]   = ticket
+rval,response = do_method("cm", "RedeemTicket", params)
+if rval:
+    Fatal("Could not redeem ticket on CM")
+    pass
+sliver = response["value"]
+print "Created a sliver CM"
+
+#
+# Start the slivers
+#
+params = {}
+params["credential"] = sliver
+rval,response = do_method("cm", "StartSliver", params)
+if rval:
+    Fatal("Could not start sliver on CM")
+    pass
+
+print "Slivers have been started, waiting for input to delete it"
+print "You should be able to log into the sliver after a little bit"
+sys.stdin.readline();
+
+#
+# Delete the sliver.
+#
+print "Deleting sliver now"
+params = {}
+params["credential"] = sliver
+rval,response = do_method("cm", "DeleteSliver", params)
+if rval:
+    Fatal("Could not delete sliver on CM")
+    pass
+print "Sliver has been deleted"