From a5ba9f5866d4d39e6f7400ad102cdc5702697e48 Mon Sep 17 00:00:00 2001
From: Daniel Palomar <danypr92@gmail.com>
Date: Fri, 8 Apr 2022 11:08:07 +0200
Subject: [PATCH] Add sponsor data to Partner resource

---
 odoo_somconnexio_python_client/resources/partner.py         | 6 ++++++
 .../test_partner/PartnerTests.test_get_with_ref.yaml        | 2 +-
 .../test_partner/PartnerTests.test_search_by_vat.yaml       | 2 +-
 3 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/odoo_somconnexio_python_client/resources/partner.py b/odoo_somconnexio_python_client/resources/partner.py
index 24a0d1e..9672b82 100644
--- a/odoo_somconnexio_python_client/resources/partner.py
+++ b/odoo_somconnexio_python_client/resources/partner.py
@@ -26,6 +26,9 @@ class Partner:
         sponsor_id,
         coop_candidate,
         member,
+        sponsorship_code,
+        sponsees_max,
+        sponsees_number,
         **kwargs
     ):
         self.id = id
@@ -45,6 +48,9 @@ class Partner:
         self.coop_agreement_code = coop_agreement_code
         self.coop_candidate = coop_candidate
         self.member = member
+        self.sponsorship_code = sponsorship_code
+        self.sponsees_max = sponsees_max
+        self.sponsees_number = sponsees_number
 
     @classmethod
     def get(cls, ref):
diff --git a/tests/resources/cassettes/test_partner/PartnerTests.test_get_with_ref.yaml b/tests/resources/cassettes/test_partner/PartnerTests.test_get_with_ref.yaml
index a923725..2995c24 100644
--- a/tests/resources/cassettes/test_partner/PartnerTests.test_get_with_ref.yaml
+++ b/tests/resources/cassettes/test_partner/PartnerTests.test_get_with_ref.yaml
@@ -21,7 +21,7 @@ interactions:
         "vat": "ES55642302N", "type": "representative", "email": "felipdata@demo.net",
         "phone": "", "mobile": "", "cooperator_register_number": 855, "cooperator_end_date":
         "", "sponsor_id": 0, "coop_agreement_code": "", "coop_candidate": false, "member":
-        true}'
+        true, "sponsorship_code": "12345", "sponsees_max": 5, "sponsees_number": 2}'
     headers:
       access-control-allow-methods:
       - GET
diff --git a/tests/resources/cassettes/test_partner/PartnerTests.test_search_by_vat.yaml b/tests/resources/cassettes/test_partner/PartnerTests.test_search_by_vat.yaml
index 160c76d..c9625cb 100644
--- a/tests/resources/cassettes/test_partner/PartnerTests.test_search_by_vat.yaml
+++ b/tests/resources/cassettes/test_partner/PartnerTests.test_search_by_vat.yaml
@@ -21,7 +21,7 @@ interactions:
         "vat": "ES55642302N", "type": "representative", "email": "felipdata@demo.net",
         "phone": "", "mobile": "", "cooperator_register_number": 855, "cooperator_end_date":
         "", "sponsor_id": 0, "coop_agreement_code": "", "coop_candidate": false, "member":
-        true}'
+        true, "sponsorship_code": "12345", "sponsees_max": 5, "sponsees_number": 2}'
     headers:
       access-control-allow-methods:
       - GET
-- 
GitLab