Changeset 14219

Timestamp:

20 Oct 2016, 12:48:21 (8 years ago)

Author:

Henrik Bettermann

Message:

Customize RIAAContractExporter.mangle_value.

Location:

main/ikobacustom.uniben/trunk/src/ikobacustom/uniben/customers

Files:

• export.py (modified) (1 diff)
• tests/test_browser.py (modified) (1 diff)

main/ikobacustom.uniben/trunk/src/ikobacustom/uniben/customers/export.py

@@ -58 +58 @@
     title = _(u'RIAA Customer Contracts')
     class_name = 'RIAAContract'
+
+    def mangle_value(self, value, name, context=None):
+
+        if name == 'history':
+            return value.messages
+        mangled_value = getattr(value, 'document_id', None)
+        if mangled_value:
+            return mangled_value
+        mangled_value = getattr(value, 'product_id', None)
+        if mangled_value:
+            return mangled_value
+        if name == 'product_options' and value is not None:
+            return [eval(entry.to_string()) for entry in value]
+        return super(
+            ContractExporterBase, self).mangle_value(
+            value, name, context=context)

main/ikobacustom.uniben/trunk/src/ikobacustom/uniben/customers/tests/test_browser.py

@@ -286 +286 @@
         self.contract.tc_dict = {'en': u'<strong>Hello world</strong>'}
         self.contract.title = u'Contract Title'
-        self.browser.open(self.customer_path + '/contracts/CON1')
+        self.browser.open(self.customer_path + '/contracts/11111222223333344444555556666677')
         self.browser.getLink("Download contract slip").click()
         self.assertEqual(self.browser.headers['Status'], '200 Ok')