Extend base class so that it can request SRs and

See merge request nexedi/slapos.core!589

slapos/testing/e2e.py

@@ -30,6 +30,7 @@ class EndToEndTestCase(unittest.TestCase):
     cls._request = staticmethod(local['request'])
     cls.product = staticmethod(local['product'])
     cls._requested = {}
+    cls._supplied = {}
 
   @classmethod
   def createLogger(cls):
@@ -51,6 +52,12 @@ class EndToEndTestCase(unittest.TestCase):
       kw['state'] = 'destroyed'
       cls._request(*args, **kw)
 
+    # remove software releases
+    for k,v in cls._supplied.items():
+      print("Remove %s from %s" %(k, v))
+      cls._supply(k, v, 'destroyed')
+
+
   @classmethod
   def request(cls, *args, **kw):
     instance_name = args[1]
@@ -58,6 +65,11 @@ class EndToEndTestCase(unittest.TestCase):
     partition = cls._request(*args, **kw)
     return cls.unwrapConnectionDict(partition.getConnectionParameterDict())
 
+  @classmethod
+  def supply(cls, software_release, computer_id, state):
+    cls._supplied[software_release] = computer_id
+    cls._supply(software_release, computer_id, state)
+
   @staticmethod
   def unwrapConnectionDict(connection_dict):
     try: