Test Software Instance registerComputerPartition

master/bt5/slapos_cloud/TestTemplateItem/testSlapOSSlapTool.py

@@ -829,3 +829,108 @@ class TestSlapOSSlapToolInstanceAccess(TestSlapOSSlapToolMixin):
     self.assertEqual(expected_xml, got_xml,
         '\n'.join([q for q in difflib.unified_diff(expected_xml.split('\n'), got_xml.split('\n'))]))
 
+  def test_registerComputerPartition(self):
+    self._makeComplexComputer()
+    partition_id = self.start_requested_software_instance.getAggregateValue(
+        portal_type='Computer Partition').getReference()
+    created_at = rfc1123_date(DateTime())
+    self.login(self.start_requested_software_instance.getReference())
+    response = self.portal_slap.registerComputerPartition(self.computer_id, partition_id)
+    self.assertEqual(200, response.status)
+    self.assertEqual( 'public, max-age=1, stale-if-error=604800',
+        response.headers.get('cache-control'))
+    self.assertEqual('REMOTE_USER',
+        response.headers.get('vary'))
+    self.assertTrue('last-modified' in response.headers)
+    self.assertEqual('text/xml; charset=utf-8',
+        response.headers.get('content-type'))
+    # check returned XML
+    xml_fp = StringIO.StringIO()
+
+    xml.dom.ext.PrettyPrint(xml.dom.ext.reader.Sax.FromXml(response.body),
+        stream=xml_fp)
+    xml_fp.seek(0)
+    got_xml = xml_fp.read()
+    expected_xml = """\
+<?xml version='1.0' encoding='UTF-8'?>
+<marshal>
+  <object id='i2' module='slapos.slap.slap' class='ComputerPartition'>
+    <tuple>
+      <string>%(computer_id)s</string>
+      <string>partition1</string>
+    </tuple>
+    <dictionary id='i3'>
+      <string>_computer_id</string>
+      <string>%(computer_id)s</string>
+      <string>_connection_dict</string>
+      <dictionary id='i4'>
+        <string>parameter1</string>
+        <string>valueof1</string>
+        <string>parameter2</string>
+        <string>https://niut:pass@example.com:4567/arfarf/oink?m=1#4.5</string>
+      </dictionary>
+      <string>_instance_guid</string>
+      <string>%(instance_guid)s</string>
+      <string>_need_modification</string>
+      <int>1</int>
+      <string>_parameter_dict</string>
+      <dictionary id='i5'>
+        <string>ip_list</string>
+        <list id='i6'>
+          <tuple>
+            <string/>
+            <string>ip_address_1</string>
+          </tuple>
+        </list>
+        <string>param</string>
+        <string>%(param)s</string>
+        <string>slap_computer_id</string>
+        <string>%(computer_id)s</string>
+        <string>slap_computer_partition_id</string>
+        <string>partition1</string>
+        <string>slap_software_release_url</string>
+        <string>%(software_release_url)s</string>
+        <string>slap_software_type</string>
+        <string>%(software_type)s</string>
+        <string>slave_instance_list</string>
+        <list id='i7'/>
+        <string>timestamp</string>
+        <string>%(timestamp)s</string>
+      </dictionary>
+      <string>_partition_id</string>
+      <string>partition1</string>
+      <string>_request_dict</string>
+      <none/>
+      <string>_requested_state</string>
+      <string>started</string>
+      <string>_software_release_document</string>
+      <object id='i8' module='slapos.slap.slap' class='SoftwareRelease'>
+        <tuple>
+          <string>%(software_release_url)s</string>
+          <string>%(computer_id)s</string>
+        </tuple>
+        <dictionary id='i9'>
+          <string>_computer_guid</string>
+          <string>%(computer_id)s</string>
+          <string>_software_instance_list</string>
+          <list id='i10'/>
+          <string>_software_release</string>
+          <string>%(software_release_url)s</string>
+        </dictionary>
+      </object>
+      <string>_synced</string>
+      <bool>1</bool>
+    </dictionary>
+  </object>
+</marshal>
+""" % dict(
+  computer_id=self.computer_id,
+  param=self.start_requested_software_instance.getInstanceXmlAsDict()['param'],
+  software_release_url=self.start_requested_software_instance.getRootSoftwareReleaseUrl(),
+  timestamp=int(self.start_requested_software_instance.getModificationDate()),
+  instance_guid=self.start_requested_software_instance.getReference(),
+  software_type=self.start_requested_software_instance.getSourceReference()
+)
+    self.assertEqual(expected_xml, got_xml,
+        '\n'.join([q for q in difflib.unified_diff(expected_xml.split('\n'), got_xml.split('\n'))]))
+

master/bt5/slapos_cloud/bt/revision

-88
\ No newline at end of file
+89
\ No newline at end of file