From 0cfd0c2eb17a9fdb11d637457481948d599b7c9e Mon Sep 17 00:00:00 2001
From: John Busby <johncbusby@gmail.com>
Date: Sat, 29 Mar 2025 07:25:35 -0700
Subject: [PATCH] use updated schema for udate transaction expense data

---
 extend/resources/transactions.py | 14 +++-----------
 tests/test_client.py             | 14 ++++----------
 2 files changed, 7 insertions(+), 21 deletions(-)

diff --git a/extend/resources/transactions.py b/extend/resources/transactions.py
index 5ef2002..16854f9 100644
--- a/extend/resources/transactions.py
+++ b/extend/resources/transactions.py
@@ -84,20 +84,12 @@ async def update_transaction_expense_data(self, transaction_id: str, data: Dict)
             data (Dict): A dictionary representing the expense data to update, should match
                          the schema:
                          {
-                             "supplier": {
-                                 "name": "Some Supplier",
-                                 "id": "supplier-id"
-                             },
-                             "expenseCategories": [
+                             "expenseDetails": [
                                  {
-                                     "categoryCode": "COMPCODE",
-                                     "labelCode": "ABC123"
+                                     "categoryId": "ec_1234",
+                                     "labelId": "ecl_1234"
                                  }
                              ],
-                             "customer": {
-                                 "name": "Some Customer",
-                                 "id": "customer-id"
-                             }
                          }
 
         Returns:
diff --git a/tests/test_client.py b/tests/test_client.py
index a41175a..3cfcd14 100644
--- a/tests/test_client.py
+++ b/tests/test_client.py
@@ -389,10 +389,8 @@ async def test_update_transaction_expense_data_success(extend, mocker, mock_tran
     mock_response = {
         "transaction": {
             **mock_transaction,
-            "supplier": {"name": "Acme Inc", "id": "sup_123"},
-            "customer": {"name": "Client A", "id": "cust_456"},
-            "expenseCategories": [
-                {"categoryCode": "TRAVEL", "labelCode": "TAXI"}
+            "expenseDetails": [
+                {"categoryId": "ec_1234", "labelId": "ecl_1234"}
             ]
         }
     }
@@ -406,15 +404,11 @@ async def test_update_transaction_expense_data_success(extend, mocker, mock_tran
     response = await extend.transactions.update_transaction_expense_data(
         transaction_id=mock_transaction["id"],
         data={
-            "supplier": {"name": "Acme Inc", "id": "sup_123"},
-            "customer": {"name": "Client A", "id": "cust_456"},
-            "expenseCategories": [{"categoryCode": "TRAVEL", "labelCode": "TAXI"}]
+            "expenseDetails": [{"categoryId": "ec_1234", "labelId": "ecl_1234"}]
         }
     )
 
-    assert response["transaction"]["supplier"]["name"] == "Acme Inc"
-    assert response["transaction"]["customer"]["id"] == "cust_456"
-    assert response["transaction"]["expenseCategories"][0]["categoryCode"] == "TRAVEL"
+    assert response["transaction"]["expenseDetails"][0]["categoryId"] == "ec_1234"
 
 
 @pytest.mark.asyncio