producteca 2.0.15__py3-none-any.whl → 2.0.17__py3-none-any.whl

producteca/sales_orders/search_sale_orders.py

@@ -1,4 +1,4 @@
-from typing import List, Optional
+from typing import List, Optional, Union
 from pydantic import BaseModel, Field
 import logging
 
@@ -9,7 +9,7 @@ class SalesOrderProduct(BaseModel):
     id: int
     name: str
     code: str
-    brand: str
+    brand: Optional[str] = None
 
 
 class SalesOrderVariationAttribute(BaseModel):
@@ -51,18 +51,18 @@ class SalesOrderPayment(BaseModel):
     coupon_amount: float = Field(alias="couponAmount")
     status: str
     method: str
-    integration: SalesOrderPaymentIntegration
+    integration: Optional[SalesOrderPaymentIntegration] = None
     transaction_fee: float = Field(alias="transactionFee")
     installments: int
-    card: SalesOrderCard
-    notes: str
+    card: Optional[SalesOrderCard] = None
+    notes: Optional[str] = None
     has_cancelable_status: bool = Field(alias="hasCancelableStatus")
     id: int
 
 
 class SalesOrderIntegration(BaseModel):
-    alternate_id: str = Field(alias="alternateId")
-    integration_id: int = Field(alias="integrationId")
+    alternate_id: Optional[str] = Field(default=None, alias="alternateId")
+    integration_id: Union[str, int] = Field(alias="integrationId")
     app: int
 
 
@@ -73,13 +73,13 @@ class SalesOrderShipmentProduct(BaseModel):
 
 
 class SalesOrderShipmentMethod(BaseModel):
-    tracking_number: str = Field(alias="trackingNumber")
+    tracking_number: Optional[str] = Field(alias="trackingNumber")
     tracking_url: str = Field(alias="trackingUrl")
     courier: str
-    mode: str
+    mode: Optional[str] = None
     cost: float
     type: str
-    eta: str
+    eta: Optional[Union[int, str]] = Field(None)
     status: str
 
 
@@ -94,12 +94,12 @@ class SalesOrderShipment(BaseModel):
     date: str
     products: List[SalesOrderShipmentProduct]
     method: SalesOrderShipmentMethod
-    integration: SalesOrderShipmentIntegration
+    integration: Optional[SalesOrderShipmentIntegration] = None
 
 
 class SalesOrderResultItem(BaseModel):
     codes: List[str]
-    contact_id: int = Field(alias="contactId")
+    contact_id: Optional[int] = Field(default=None, alias="contactId")
     currency: str
     date: str
     delivery_method: str = Field(alias="deliveryMethod")
@@ -107,33 +107,33 @@ class SalesOrderResultItem(BaseModel):
     id: str
     integration_ids: List[str] = Field(alias="integrationIds")
     integrations: List[SalesOrderIntegration]
-    invoice_integration_app: int = Field(alias="invoiceIntegrationApp")
-    invoice_integration_id: str = Field(alias="invoiceIntegrationId")
+    invoice_integration_app: Optional[int] = Field(default=None, alias="invoiceIntegrationApp")
+    invoice_integration_id: Optional[str] = Field(default=None, alias="invoiceIntegrationId")
     lines: List[SalesOrderLine]
     payments: List[SalesOrderPayment]
     payment_status: str = Field(alias="paymentStatus")
     payment_term: str = Field(alias="paymentTerm")
     product_names: List[str] = Field(alias="productNames")
-    reserving_product_ids: str = Field(alias="reservingProductIds")
+    reserving_product_ids: Union[str, List[str]] = Field(alias="reservingProductIds")
     sales_channel: int = Field(alias="salesChannel")
     shipments: List[SalesOrderShipment]
-    tracking_number: str = Field(alias="trackingNumber")
+    tracking_number: Optional[str] = Field(alias="trackingNumber")
     skus: List[str]
     status: str
     tags: List[str]
     warehouse: str
     company_id: int = Field(alias="companyId")
     shipping_cost: float = Field(alias="shippingCost")
-    contact_phone: str = Field(alias="contactPhone")
+    contact_phone: Optional[str] = Field(default=None, alias="contactPhone")
     brands: List[str]
     courier: str
     order_id: int = Field(alias="orderId")
     updated_at: str = Field(alias="updatedAt")
-    invoice_integration_created_at: str = Field(alias="invoiceIntegrationCreatedAt")
-    invoice_integration_document_url: str = Field(alias="invoiceIntegrationDocumentUrl")
+    invoice_integration_created_at: Optional[str] = Field(default=None, alias="invoiceIntegrationCreatedAt")
+    invoice_integration_document_url: Optional[str] = Field(default=None, alias="invoiceIntegrationDocumentUrl")
     has_document_url: bool = Field(alias="hasDocumentUrl")
-    integration_alternate_ids: str = Field(alias="integrationAlternateIds")
-    cart_id: str = Field(alias="cartId")
+    integration_alternate_ids: Union[str, List[str]] = Field(alias="integrationAlternateIds")
+    cart_id: Optional[str] = Field(default=None, alias="cartId")
     amount: float
     has_any_shipments: bool = Field(alias="hasAnyShipments")
 

producteca/sales_orders/tests/test_sales_orders.py

@@ -1,6 +1,6 @@
 import unittest
 from unittest.mock import patch, Mock
-from producteca.sales_orders.sales_orders import SaleOrder, SaleOrderInvoiceIntegration
+from producteca.sales_orders.sales_orders import SaleOrder
 from producteca.client import ProductecaClient
 
 
@@ -12,7 +12,14 @@ class TestSaleOrder(unittest.TestCase):
         self.mock_response = {
             "id": self.sale_order_id,
             "contact": {"id": 1, "name": "Test Contact"},
-            "lines": []
+            "lines": [],
+            "invoiceIntegration": {
+                'id': 1,
+                'integrationId': 'test-integration',
+                'app': 1,
+                'createdAt': '2023-01-01',
+                'decreaseStock': True
+            }
         }
 
     @patch('requests.get')
@@ -34,7 +41,15 @@ class TestSaleOrder(unittest.TestCase):
             json=lambda: mock_labels
         )
 
-        labels = self.client.SalesOrder(id=1234).get_shipping_labels()
+        labels = self.client.SalesOrder(id=1234, invoiceIntegration={
+                'id': 1,
+                'integrationId': 'test-integration',
+                'app': 1,
+                'createdAt': '2023-01-01',
+                'decreaseStock': True,
+                "documentUrl": "https://aallala.copm",
+                "xmlUrl": "https://aallala.copm",
+            }).get_shipping_labels()
         self.assertEqual(labels, mock_labels)
         mock_get.assert_called_once()
 
@@ -44,7 +59,15 @@ class TestSaleOrder(unittest.TestCase):
             status_code=200
         )
 
-        self.client.SalesOrder(id=1234).close()
+        self.client.SalesOrder(id=1234, invoiceIntegration={
+                'id': 1,
+                'integrationId': 'test-integration',
+                'app': 1,
+                'createdAt': '2023-01-01',
+                'decreaseStock': True,
+                "documentUrl": "https://aallala.copm",
+                "xmlUrl": "https://aallala.copm",
+            }).close()
         mock_post.assert_called_once()
 
     @patch('requests.post')
@@ -54,7 +77,15 @@ class TestSaleOrder(unittest.TestCase):
             json=lambda: {"status": "cancelled"}
         )
 
-        self.client.SalesOrder(id=1234).cancel()
+        self.client.SalesOrder(id=1234, invoiceIntegration={
+                'id': 1,
+                'integrationId': 'test-integration',
+                'app': 1,
+                'createdAt': '2023-01-01',
+                'decreaseStock': True,
+                "documentUrl": "https://aallala.copm",
+                "xmlUrl": "https://aallala.copm",
+            }).cancel()
         mock_post.assert_called_once()
 
     @patch('requests.post')
@@ -65,31 +96,30 @@ class TestSaleOrder(unittest.TestCase):
             json=lambda: self.mock_response
         )
 
-        response = self.client.SalesOrder.synchronize(sale_order)
+        response = self.client.SalesOrder(**sale_order.model_dump(by_alias=True)).synchronize()
         self.assertEqual(response.id, self.sale_order_id)
         mock_post.assert_called_once()
 
     @patch('requests.put')
     def test_invoice_integration(self, mock_put):
         invoice_data = {
-            "id": 1,
-            "integrationId": "test123",
-            "app": 1
-        }
-        invoice_integration = SaleOrderInvoiceIntegration(**invoice_data)
-        sale_order = SaleOrder(id=self.sale_order_id, invoiceIntegration=invoice_integration)
+                'id': 1,
+                'integrationId': 'test-integration',
+                'app': 1,
+                'createdAt': '2023-01-01',
+                'decreaseStock': True,
+                "documentUrl": "https://aallala.copm",
+                "xmlUrl": "https://aallala.copm",
+            }
 
         mock_put.return_value = Mock(
             status_code=200,
-            json=lambda: {
-                "id": 1,
-                "integrationId": "test123",
-                "app": 1
-                }
+            json=lambda: invoice_data,
+            ok=True
         )
 
-        response = self.client.SalesOrder(**sale_order.model_dump()).invoice_integration()
-        self.assertIsInstance(response, SaleOrder)
+        response = self.client.SalesOrder(id=self.sale_order_id, invoiceIntegration=invoice_data).invoice_integration()
+        self.assertTrue(response)
         mock_put.assert_called_once()
 
 

producteca/sales_orders/tests/test_search_so.py

@@ -26,8 +26,7 @@ class TestSearchSalesOrder(unittest.TestCase):
             mock_get.return_value = mock_response
 
         response = self.client.SalesOrder.search(self.params)
-        
-        # Validate response
+
         self.assertEqual(response.count, 0)
         self.assertEqual(len(response.results), 1)
         self.assertEqual(response.results[0].id, "string")
@@ -35,14 +34,13 @@ class TestSearchSalesOrder(unittest.TestCase):
 
     @patch('requests.get')
     def test_search_saleorder_error(self, mock_get):
-        # Mock error response
         mock_response = Mock()
         mock_response.json.return_value = {"error": "Invalid request"}
         mock_response.status_code = 400
         mock_get.return_value = mock_response
         with self.assertRaises(Exception):
             self.client.SalesOrder.search(self.params)
-       
+
 
 if __name__ == '__main__':
     unittest.main()

producteca/shipments/tests/test_shipment.py

@@ -14,14 +14,20 @@ class TestShipment(unittest.TestCase):
         products = [ShipmentProduct(product=1, variation=2, quantity=3)]
         method = ShipmentMethod(trackingNumber="TN123", trackingUrl="http://track.url", courier="DHL", mode="air", cost=10.5, type="express", eta=5, status="shipped")
         integration = ShipmentIntegration(id=1, integrationId="int123", app=10, status="active")
-        payload = Shipment(date="2023-01-01", products=products, method=method, integration=integration)
+        payload = Shipment(date="2023-01-01", products=products, method=method, integration=integration).model_dump(by_alias=True)
 
         mock_response = MagicMock()
         mock_response.status_code = 201
-        mock_response.json.return_value = {'success': True}
+        mock_response.json.return_value = payload
         mock_post.return_value = mock_response
         # Act
-        shipment = self.client.SalesOrder(id=1234).add_shipment(payload)
+        shipment = self.client.SalesOrder(id=1234, invoiceIntegration={
+            'id': 1,
+            'integrationId': 'test-integration',
+            'app': 1,
+            'createdAt': '2023-01-01',
+            'decreaseStock': True
+        }).add_shipment(payload)
 
         self.assertIsInstance(shipment, Shipment)
         mock_post.assert_called_once()
@@ -33,14 +39,20 @@ class TestShipment(unittest.TestCase):
         products = [ShipmentProduct(product=4, quantity=7)]
         method = ShipmentMethod(courier="FedEx", cost=15.0)
         integration = ShipmentIntegration(status="pending")
-        payload = Shipment(date="2023-02-02", products=products, method=method, integration=integration)
+        payload = Shipment(date="2023-02-02", products=products, method=method, integration=integration).model_dump(by_alias=True)
 
         mock_response = MagicMock()
         mock_response.status_code = 200
-        mock_response.json.return_value = {'updated': True}
+        mock_response.json.return_value = payload
         mock_put.return_value = mock_response
-
-        shipment = self.client.SalesOrder(id=1234).update_shipment(shipment_id, payload)
+        # Act
+        shipment = self.client.SalesOrder(id=1234, invoiceIntegration={
+            'id': 1,
+            'integrationId': 'test-integration',
+            'app': 1,
+            'createdAt': '2023-01-01',
+            'decreaseStock': True
+        }).update_shipment(shipment_id, payload)
 
         self.assertIsInstance(shipment, Shipment)
         mock_put.assert_called_once()