Fix invoices (customer_uuid) and data_source (system)

chartmogul/__init__.py

@@ -27,7 +27,7 @@
 """
 
 __title__ = 'chartmogul'
-__version__ = '1.0.2'
+__version__ = '1.0.3'
 __build__ = 0x000000
 __author__ = 'ChartMogul Ltd'
 __license__ = 'MIT'

chartmogul/api/data_source.py

@@ -16,6 +16,7 @@ class _Schema(Schema):
         name = fields.String()
         created_at = fields.DateTime()
         status = fields.Str()
+        system = fields.Str()
 
         @post_load
         def make(self, data):

chartmogul/imp/invoice.py

@@ -33,7 +33,7 @@ class Invoice(Resource):
     """
     _path = "/import/customers{/uuid}/invoices"
     _root_key = 'invoices'
-    _many = namedtuple('Invoices', [_root_key, "current_page", "total_pages"])
+    _many = namedtuple('Invoices', [_root_key, "current_page", "total_pages", "customer_uuid"])
     _many.__new__.__defaults__ = (None,) * len(_many._fields)
 
     class _Schema(Schema):

test/imp/test_invoice.py

@@ -154,7 +154,6 @@ class InvoiceTestCase(unittest.TestCase):
 
     @requests_mock.mock()
     def test_create(self, mock_requests):
-
         mock_requests.register_uri(
             'POST',
             "https://api.chartmogul.com/v1/import/customers/UUID/invoices",
@@ -172,3 +171,25 @@ def test_create(self, mock_requests):
         # Struct too complex to do 1:1 comparison
         self.assertTrue(isinstance(result, Invoice._many))
         self.assertEqual(len(result.invoices), 1)
+
+    @requests_mock.mock()
+    def test_list_has_customer_uuid(self, mock_requests):
+        responseData['customer_uuid'] = 'UUID'
+
+        mock_requests.register_uri(
+            'GET',
+            "https://api.chartmogul.com/v1/import/customers/UUID/invoices",
+            request_headers={'Authorization': 'Basic dG9rZW46c2VjcmV0'},
+            status_code=200,
+            json=responseData
+        )
+
+        config = Config("token", "secret")  # is actually checked in mock
+        result = Invoice.all(config, uuid="UUID").get()
+
+        self.assertEqual(mock_requests.call_count, 1, "expected call")
+        self.assertEqual(mock_requests.last_request.qs, {})
+        # Struct too complex to do 1:1 comparison
+        self.assertTrue(isinstance(result, Invoice._many))
+        self.assertEqual(len(result.invoices), 1)
+        self.assertEqual(result.customer_uuid, 'UUID')