Changeset 12143 for main/waeup.ikoba/trunk/src/waeup/ikoba/customers/tests/test_export.py

Timestamp:

4 Dec 2014, 12:22:03 (10 years ago)

Author:

Henrik Bettermann

Message:

Add contract exporter.

File:

• main/waeup.ikoba/trunk/src/waeup/ikoba/customers/tests/test_export.py (modified) (4 diffs)

main/waeup.ikoba/trunk/src/waeup/ikoba/customers/tests/test_export.py

@@ -31 +31 @@
 from waeup.ikoba.customers.catalog import CustomersQuery
 from waeup.ikoba.customers.export import (
-    CustomerExporter, CustomerDocumentExporter, get_customers)
+    CustomerExporter, CustomerDocumentExporter, ContractExporter,
+    get_customers)
 from waeup.ikoba.customers.interfaces import ICSVCustomerExporter
 from waeup.ikoba.customers.customer import Customer
 from waeup.ikoba.customers.documents import CustomerSampleDocument
+from waeup.ikoba.customers.contracts import SampleContract
 from waeup.ikoba.customers.tests.test_batching import CustomerImportExportSetup
 from waeup.ikoba.testing import FunctionalLayer
@@ -88 +90 @@
         exporter.export([self.customer], self.outfile)
         result = open(self.outfile, 'rb').read()
-        self.assertTrue(
-            'customer_id,email,firstname,lastname,middlename,phone,'
-            'reg_number,sex,suspended,suspended_comment,password,state,'
-            'history\r\n'
-            'A111111,anna@sample.com,Anna,Tester,M.,+234-123-12345#,'
-            '123,,0,,,created'
-            in result
+        self.assertMatches(
+            'customer_id,email,firstname,lastname,middlename,phone,reg_number,'
+            'sex,suspended,suspended_comment,password,state,history\r\n'
+            'A111111,anna@sample.com,Anna,Tester,M.,+234-123-12345#,123,,0,,,'
+            'created,[u\'2014-12-04 11:25:35 UTC - Customer record created '
+            'by system\']\r\n',
+            result
             )
         return
@@ -202 +204 @@
         return
 
-    def test_export_customer(self):
+    def test_export_customer_document(self):
         # we can really export all documents of a certain customer
         # set values we can expect in export file
@@ -219 +221 @@
             )
         return
+
+
+class ContractExporterTest(CustomerImportExportSetup):
+
+    layer = FunctionalLayer
+
+    def setUp(self):
+        super(ContractExporterTest, self).setUp()
+        self.setup_for_export()
+        return
+
+    def test_ifaces(self):
+        # make sure we fullfill interface contracts
+        obj = ContractExporter()
+        verifyObject(ICSVCustomerExporter, obj)
+        verifyClass(ICSVCustomerExporter, ContractExporter)
+        return
+
+    def test_get_as_utility(self):
+        # we can get a contracts exporter as utility
+        result = queryUtility(ICSVExporter, name="contracts")
+        self.assertTrue(result is not None)
+        return
+
+    def test_export_empty(self):
+        # we can export a nearly empty contract
+        contract = SampleContract()
+        exporter = ContractExporter()
+        exporter.export([contract], self.outfile)
+        result = open(self.outfile, 'rb').read()
+        self.assertEqual(
+            result,
+            'class_name,contract_category,contract_id,document_object,'
+            'history,is_editable,last_product_id,last_transition_date,'
+            'product_object,state,title\r\n'
+
+            'SampleContract,sample,c101,,[],1,,,,,\r\n'
+            )
+        return
+
+    def test_export(self):
+        # we can really export customer contracts.
+        # set values we can expect in export file
+        self.setup_customer(self.customer)
+        exporter = ContractExporter()
+        exporter.export([self.contract], self.outfile)
+        result = open(self.outfile, 'rb').read()
+        self.assertMatches(
+            'class_name,contract_category,contract_id,document_object,'
+            'history,is_editable,last_product_id,last_transition_date,'
+            'product_object,state,title\r\n'
+
+            'SampleContract,sample,c101,d101,[u\'2014-12-04 12:10:46 UTC - '
+            'Contract record created by system\'],0,,'
+            '2014-12-04 12:10:46.333445#,SAM,created,My Contract\r\n',
+            result
+            )
+        return
+
+    def test_export_all(self):
+        # we can really export all contracts
+        # set values we can expect in export file
+        self.setup_customer(self.customer)
+        exporter = ContractExporter()
+        exporter.export_all(self.app, self.outfile)
+        result = open(self.outfile, 'rb').read()
+        self.assertMatches(
+            'class_name,contract_category,contract_id,document_object,'
+            'history,is_editable,last_product_id,last_transition_date,'
+            'product_object,state,title\r\n'
+
+            'SampleContract,sample,c101,d101,[u\'2014-12-04 12:10:46 UTC - '
+            'Contract record created by system\'],0,,'
+            '2014-12-04 12:10:46.333445#,SAM,created,My Contract\r\n',
+            result
+            )
+        return
+
+    def test_export_contract(self):
+        # we can really export all contracts of a certain customer
+        # set values we can expect in export file
+        self.setup_customer(self.customer)
+        exporter = ContractExporter()
+        exporter.export_customer(self.customer, self.outfile)
+        result = open(self.outfile, 'rb').read()
+        self.assertMatches(
+            'class_name,contract_category,contract_id,document_object,'
+            'history,is_editable,last_product_id,last_transition_date,'
+            'product_object,state,title\r\n'
+
+            'SampleContract,sample,c101,d101,[u\'2014-12-04 12:10:46 UTC - '
+            'Contract record created by system\'],0,,'
+            '2014-12-04 12:10:46.333445#,SAM,created,My Contract\r\n',
+            result
+            )
+        return
+