Add integration tests coverage

Author: Jonathan Wenger

.github/workflows/python-publish.yml

@@ -14,14 +14,25 @@ jobs:
         uses: actions/setup-python@v2
         with:
           python-version: "3.x"
-      - name: Retrieve bearer token
-        id: get_bearer_token
+
+      - name: Retrieve bearer token for EInvoicing
+        id: get_bearer_token_einvoicing
         run: |
           response=$(curl -X POST ${{secrets.OKTA_ACCESS_TOKEN_URL}}/connect/token \
             -H "Content-Type: application/x-www-form-urlencoded" \
             -d "grant_type=client_credentials&client_id=${{secrets.OKTA_CLIENT_ID}}&client_secret=${{secrets.OKTA_CLIENT_SECRET}}")
           token=$(echo $response | jq -r '.access_token')
-          echo "BEARER_TOKEN=${token}" >> $GITHUB_ENV
+          echo "BEARER_TOKEN_EINVOICING=${token}" >> $GITHUB_ENV
+
+      - name: Retrieve bearer token for A1099
+        id: get_bearer_token_a1099
+        run: |
+          response=$(curl -X POST ${{secrets.AI_SBX_URL}}/connect/token \
+            -H "Content-Type: application/x-www-form-urlencoded" \
+            -d "grant_type=client_credentials&client_id=${{secrets.AI_CLIENT_ID_A1099}}&client_secret=${{secrets.AI_CLIENT_SECRET_A1099}}")
+          token=$(echo $response | jq -r '.access_token')
+          echo "BEARER_TOKEN_A1099=${token}" >> $GITHUB_ENV
+
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip

.github/workflows/test.yml

@@ -22,14 +22,25 @@ jobs:
           pip install build
           pip install -r ./requirements.txt
           pip install -r ./test-requirements.txt
-      - name: Retrieve bearer token
-        id: get_bearer_token
+
+      - name: Retrieve bearer token for EInvoicing
+        id: get_bearer_token_einvoicing
         run: |
           response=$(curl -X POST ${{secrets.OKTA_ACCESS_TOKEN_URL}}/connect/token \
             -H "Content-Type: application/x-www-form-urlencoded" \
             -d "grant_type=client_credentials&client_id=${{secrets.OKTA_CLIENT_ID}}&client_secret=${{secrets.OKTA_CLIENT_SECRET}}")
           token=$(echo $response | jq -r '.access_token')
-          echo "BEARER_TOKEN=${token}" >> $GITHUB_ENV
+          echo "BEARER_TOKEN_EINVOICING=${token}" >> $GITHUB_ENV
+
+      - name: Retrieve bearer token for A1099
+        id: get_bearer_token_a1099
+        run: |
+          response=$(curl -X POST ${{secrets.AI_SBX_URL}}/connect/token \
+            -H "Content-Type: application/x-www-form-urlencoded" \
+            -d "grant_type=client_credentials&client_id=${{secrets.AI_CLIENT_ID_A1099}}&client_secret=${{secrets.AI_CLIENT_SECRET_A1099}}")
+          token=$(echo $response | jq -r '.access_token')
+          echo "BEARER_TOKEN_A1099=${token}" >> $GITHUB_ENV
+
       - name: Build package
         run: python -m build
       - name: Pip Package

Avalara/SDK/configuration.py

@@ -574,6 +574,12 @@ def auth_settings(self, required_scopes):
                 'key': 'Authorization',
                 'value': f'Bearer {self.access_token}',
             }
+            auth['bearer'] = {
+                'type': 'api_key',
+                'in': 'header',
+                'key': 'Authorization',
+                'value': f'Bearer {self.access_token}',
+            }
         elif self.client_id is not None and not self.is_null_or_empty(
             self.client_secret
         ):

test/integration/test_a1099api.py

@@ -0,0 +1,79 @@
+from math import fabs
+import unittest
+import pytest
+import ssl
+import time
+import os
+from dotenv import load_dotenv
+import uuid
+
+# Import Configuration, ApiClient, and exceptions
+from Avalara.SDK.configuration import Configuration
+from Avalara.SDK.api_client import ApiClient
+from Avalara.SDK.exceptions import ApiException
+
+# Import the specific A1099 APIs
+from Avalara.SDK.api.A1099.V2.issuers1099_api import Issuers1099Api
+from Avalara.SDK.api.A1099.V2.companies_w9_api import CompaniesW9Api
+from Avalara.SDK.api.A1099.V2.forms1099_api import Forms1099Api
+
+class TestA1099Api(unittest.TestCase):
+    """A1099Api unit test stubs"""
+
+    def setUp(self):
+        load_dotenv()
+
+        bearer_token = os.getenv('BEARER_TOKEN_A1099')
+        # Configure with the A1099 bearer token
+        configuration = Configuration(
+            environment="sandbox",
+            access_token=bearer_token
+        )
+
+        # Create an API client for all tests
+        self.api_client = ApiClient(configuration)
+        self.issuers_api = Issuers1099Api(self.api_client)
+        self.companies_api = CompaniesW9Api(self.api_client)
+        self.forms_api = Forms1099Api(self.api_client)
+
+    def tearDown(self):
+        pass
+
+    def test_get_issuers(self):
+        """Test the GetIssuers endpoint returns data"""
+        try:
+            correlation_id = str(uuid.uuid4())
+            result = self.issuers_api.get_issuers("2.0", correlation_id)
+            print(result)
+            assert result is not None, "Expected non-null issuers list"
+        except ApiException as e:
+            print(f"Exception when calling Issuers1099Api->get_issuers: {e}\n")
+            assert False
+        print("Completed get_issuers test")
+
+    def test_get_companies(self):
+        """Test the GetCompanies endpoint returns data"""
+        try:
+            correlation_id = str(uuid.uuid4())
+            result = self.companies_api.get_companies("2.0", correlation_id)
+            print(result)
+            assert result is not None, "Expected non-null companies list"
+        except ApiException as e:
+            print(f"Exception when calling CompaniesW9Api->get_companies: {e}\n")
+            assert False
+        print("Completed get_companies test")
+
+    def test_list_1099_forms(self):
+        """Test the List1099Forms endpoint returns data"""
+        try:
+            correlation_id = str(uuid.uuid4())
+            result = self.forms_api.list1099_forms("2.0", correlation_id)
+            print(result)
+            assert result is not None, "Expected non-null 1099 forms list"
+        except ApiException as e:
+            print(f"Exception when calling Forms1099Api->list_1099_forms: {e}\n")
+            assert False
+        print("Completed list_1099_forms test")
+
+if __name__ == '__main__':
+    unittest.main()

test/integration/test_documentsapi.py

@@ -25,7 +25,7 @@ def setUp(self):
         # Use the Configuration from Avalara.SDK
         configuration = Configuration(
             environment="sandbox",
-            access_token=os.getenv('BEARER_TOKEN')
+            access_token=os.getenv('BEARER_TOKEN_EINVOICING')
         )
 
         # Create an API client using that configuration

test/integration/test_mandatesapi.py

@@ -27,7 +27,7 @@ def setUp(self):
         # ssl.SSLContext.verify_mode = ssl.VerifyMode.CERT_OPTIONAL
         configuration = Configuration(
             environment="sandbox",
-            access_token=os.getenv('BEARER_TOKEN')
+            access_token=os.getenv('BEARER_TOKEN_EINVOICING')
         )
         with ApiClient(configuration) as api_client:
             self.api = MandatesApi(api_client)