msfabricpysdkcore 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

msfabricpysdkcore/adminapi.py

@@ -9,10 +9,12 @@ from msfabricpysdkcore.client import FabricClient
 class FabricClientAdmin(FabricClient):
     """FabricClientAdmin class to interact with Fabric Admin APIs"""
 
-    def __init__(self, tenant_id = None, client_id = None, client_secret = None) -> None:
+    def __init__(self, tenant_id = None, client_id = None, client_secret = None,
+                 username = None, password = None) -> None:
         """Initialize FabricClientAdmin object"""
         super().__init__(scope="https://api.fabric.microsoft.com/.default", 
-                         tenant_id=tenant_id, client_id=client_id, client_secret=client_secret)
+                         tenant_id=tenant_id, client_id=client_id, client_secret=client_secret,
+                         username=username, password=password)
 
 
     def long_running_operation(self, response_headers):

msfabricpysdkcore/auth.py

@@ -3,6 +3,7 @@ from warnings import warn
 import requests
 from abc import abstractmethod
 from azure.identity import AzureCliCredential
+import msal
 from msfabricpysdkcore.util import logger
 import logging
 try:
@@ -98,3 +99,30 @@ class FabricSparkUtilsAuthentication(FabricAuth):
         return token
     
 
+class MSALConfidentialClientApplicationAuthentication(FabricAuth):
+
+    def __init__(self, tenant_id, client_id, client_secret, username, password, scope):
+        super().__init__(scope)
+
+        self._logger.info("Using Microsoft Authentication Library (MSAL) ConfidentialClientApplication for authentication")
+                
+        self.username = username
+        self.password = password
+        self.scopes = [scope]
+
+        authority = f"https://login.microsoftonline.com/{tenant_id}"
+
+        self.app = msal.ConfidentialClientApplication(
+            client_id=client_id,
+            client_credential=client_secret,
+            authority=authority,
+        )
+
+    def get_token(self):
+        result = self.app.acquire_token_for_client(scopes=self.scopes)
+        result = self.app.acquire_token_by_username_password(
+            username=self.username,
+            password=self.password,
+            scopes=self.scopes,
+        )
+        return result["access_token"]

msfabricpysdkcore/client.py

@@ -6,7 +6,7 @@ from time import sleep
 import requests
 import json
 
-from msfabricpysdkcore.auth import FabricAuthClient, FabricServicePrincipal, FabricSparkUtilsAuthentication
+from msfabricpysdkcore.auth import FabricAuthClient, FabricServicePrincipal, FabricSparkUtilsAuthentication, MSALConfidentialClientApplicationAuthentication
 from msfabricpysdkcore.util import logger
 
 class FabricClient():
@@ -14,7 +14,7 @@ class FabricClient():
 
     _logger: logging.Logger
 
-    def __init__(self, scope, tenant_id = None, client_id = None, client_secret = None, silent=None) -> None:
+    def __init__(self, scope, tenant_id = None, client_id = None, client_secret = None, username = None, password = None, silent=None) -> None:
         """Initialize FabricClient object"""
 
         self._logger = logger.getChild(__name__)
@@ -22,8 +22,9 @@ class FabricClient():
         self.tenant_id = tenant_id if tenant_id else os.getenv("FABRIC_TENANT_ID")
         self.client_id = client_id if client_id else os.getenv("FABRIC_CLIENT_ID")
         self.client_secret = client_secret if client_secret else os.getenv("FABRIC_CLIENT_SECRET")
+        self.username = username if username else os.getenv("FABRIC_USERNAME")
+        self.password = password if password else os.getenv("FABRIC_PASSWORD")
         self.scope = scope
-        #self.scope = "https://api.fabric.microsoft.com/.default"
 
         if self.client_id is None or self.client_secret is None or self.tenant_id is None:
             try:
@@ -31,10 +32,18 @@ class FabricClient():
             except:
                 self.auth = FabricAuthClient(self.scope)
         else:
-            self.auth = FabricServicePrincipal(scope= self.scope,
-                                               tenant_id = self.tenant_id,
-                                               client_id = self.client_id, 
-                                               client_secret = self.client_secret)
+            if username and password:
+                self.auth = MSALConfidentialClientApplicationAuthentication(tenant_id = self.tenant_id,
+                                                                            client_id = self.client_id,
+                                                                            client_secret = self.client_secret,
+                                                                            username = self.username,
+                                                                            password = self.password,
+                                                                            scope = self.scope)
+            else:
+                self.auth = FabricServicePrincipal(scope= self.scope,
+                                                   tenant_id = self.tenant_id,
+                                                   client_id = self.client_id, 
+                                                   client_secret = self.client_secret)
 
         if silent is not None:
             warn("The 'silent' parameter is deprecated and will be removed in a future version.", DeprecationWarning, stacklevel=2)

msfabricpysdkcore/coreapi.py

@@ -9,12 +9,15 @@ from msfabricpysdkcore.util import logger
 class FabricClientCore(FabricClient):
     """FabricClientCore class to interact with Fabric Core APIs"""
 
-    def __init__(self, tenant_id = None, client_id = None, client_secret = None, silent=None) -> None:
+    def __init__(self, tenant_id = None, client_id = None, client_secret = None,
+                 username = None, password = None, silent=None) -> None:
         """Initialize FabricClientCore object"""
         super().__init__(scope="https://api.fabric.microsoft.com/.default", 
                          tenant_id=tenant_id,
                          client_id=client_id,
-                         client_secret=client_secret)
+                         client_secret=client_secret,
+                         username=username,
+                         password=password)
         if silent is not None:
             warn("The 'silent' parameter is deprecated and will be removed in a future version.", DeprecationWarning, stacklevel=2)
 

msfabricpysdkcore/fabric_azure_client.py

@@ -5,11 +5,14 @@ from msfabricpysdkcore.client import FabricClient
 
 class FabricAzureClient(FabricClient):
 
-    def __init__(self, tenant_id=None, client_id=None, client_secret=None, silent=None) -> None:
-        super().__init__(scope = "https://management.azure.com/",
+    def __init__(self, tenant_id=None, client_id=None, client_secret=None, 
+                 username = None, password = None, silent=None) -> None:
+        super().__init__(scope = "https://management.azure.com/.default",
                          tenant_id = tenant_id,
                          client_id = client_id,
-                         client_secret = client_secret,)
+                         client_secret = client_secret,
+                         username = username,
+                         password = password)
 
         if silent is not None:
             warn("The 'silent' parameter is deprecated and will be removed in a future version.", DeprecationWarning, stacklevel=2)

{msfabricpysdkcore-0.1.3.dist-info → msfabricpysdkcore-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: msfabricpysdkcore
-Version: 0.1.3
+Version: 0.1.5
 Summary: A Python SDK for Microsoft Fabric
 Author: Andreas Rederer
 Project-URL: Homepage, https://github.com/DaSenf1860/ms-fabric-sdk-core
@@ -25,7 +25,11 @@ They are designed to automate your Fabric processes.
 
 This SDK helps to interact with the Fabric APIs in a more Pythonic way.
 Additionally it brings some extra features like:
-- Authentication is handled for you (currently Azure CLI Authentication, Authentication from a Microsoft Fabric notebook and Service Principal Authentication are supported)
+- Authentication is handled for you, the following is supported:
+  - Azure CLI Authentication
+  - Authentication from a Microsoft Fabric notebook
+  - Service Principal Authentication
+  - MSALConfidentialClientApplicationAuthentication
 - Waiting for completion of long running operations
 - Retry logic when hitting the API rate limits
 - Referencing objects by name instead of ID

msfabricpysdkcore-0.1.5.dist-info/RECORD

@@ -0,0 +1,28 @@
+msfabricpysdkcore/__init__.py,sha256=ObRW5Q8IMqrvA6VH6zXSv3n01AzRCGjn5RojJXAR6VE,208
+msfabricpysdkcore/admin_item.py,sha256=9L09Kb7kn7JW0bqVu1qPOMT6oHLduX4Q5_qhVJZTLxQ,3290
+msfabricpysdkcore/admin_workspace.py,sha256=umNnIF8sf5Y8BDgfQdfG6sOZrQQc7RJINL9j8DPnm3c,3180
+msfabricpysdkcore/adminapi.py,sha256=90OcFVXYhv6wf3bQCwspbw-NRxVugrgACkyru-h79MU,26188
+msfabricpysdkcore/auth.py,sha256=CwagXOYpGlfOh9zGA8nP8g-L602ujp0aqBWJuZNYt98,4332
+msfabricpysdkcore/capacity.py,sha256=Q_2-XrZtdf9F67fY0qU3D0ocEOGQq4KtIXAv9dXjQhI,1761
+msfabricpysdkcore/client.py,sha256=KpHREbfzQRRighyJzyES3COZP3raK0C1-WhxGd07LHU,7890
+msfabricpysdkcore/coreapi.py,sha256=X4gj7fJBVdkypliZFig5rlc9qaCFRD1TJPsVCiO2fK0,132261
+msfabricpysdkcore/deployment_pipeline.py,sha256=2d7BqRPgfBiAlQOP4UfYLmFMM8xonSjPkJKhevIKGEY,5756
+msfabricpysdkcore/domain.py,sha256=92IVvZ3jXHIT1a0zlTPD7uoQX6TcBBE_Y_b4dScxHSU,6949
+msfabricpysdkcore/environment.py,sha256=4k2Le1mAQIrfcpNc3n1DbgdCzAldGTSTCbiDQGk0DlA,2784
+msfabricpysdkcore/fabric_azure_capacity.py,sha256=7JxMp9weiKG_mDjlRK-88oIXr0kdG0pzv-Ouycf3fus,2808
+msfabricpysdkcore/fabric_azure_client.py,sha256=I7Z0Y8Xy_esQcPaXgPL7EAkQmoQkAklJO4hxk_90dno,10722
+msfabricpysdkcore/item.py,sha256=3ixgRDTmrKXw82xl1MkaaOQ2986s2_DFEJQeUkKPcEU,6376
+msfabricpysdkcore/job_instance.py,sha256=HXSUVoE4XaG2Ic5RYZ1Mx3ymiIarHDAnyjaXXY4Aj74,2381
+msfabricpysdkcore/lakehouse.py,sha256=yIrzatWM9emPn-Y54Cg_ZdAydIWjxrpK65jIQ4SClgE,1703
+msfabricpysdkcore/long_running_operation.py,sha256=XTlsueSZKVFICxhx99geEQ6btZFlFb8-lssigmQ9c6Y,2133
+msfabricpysdkcore/onelakeshortcut.py,sha256=H02wR6Z86qTEJOwVRMKz1Ou3K88Y9pfJa91vizjprvo,1661
+msfabricpysdkcore/otheritems.py,sha256=rkIbF-LxUUZh0ZDqFiWeutMeIsBXw_R8UUCglIifWBQ,12024
+msfabricpysdkcore/spark_custom_pool.py,sha256=YsEULaQG-FO507aPIb-4kk93ZWBmDZj6fbOEHYoyxHE,3188
+msfabricpysdkcore/workspace.py,sha256=7ZOt8ovlUOfusQ-avl4OOgMueN1XAp7AXdfpzAhXh3s,40428
+msfabricpysdkcore/util/__init__.py,sha256=p-1dC4AurfKdIUppNVlRCIcmwHBTURqfgIwxcie3fEA,50
+msfabricpysdkcore/util/logger.py,sha256=XXdeaqI-BhXZgacChKrWP88TPU4FTWvS8L0YWjCT54A,1166
+msfabricpysdkcore-0.1.5.dist-info/LICENSE,sha256=1NrGuF-zOmzbwzk3iI6lsP9koyDeKO1B0-8OD_tTvOQ,1156
+msfabricpysdkcore-0.1.5.dist-info/METADATA,sha256=t6LBXzBORqvYmBULkAZ8AONvTY1wlJfqkWZhAHQV_FY,23352
+msfabricpysdkcore-0.1.5.dist-info/WHEEL,sha256=Rp8gFpivVLXx-k3U95ozHnQw8yDcPxmhOpn_Gx8d5nc,91
+msfabricpysdkcore-0.1.5.dist-info/top_level.txt,sha256=3iRonu6ptDGQN4Yl6G76XGM7xbFNsskiEHW-P2gMQGY,18
+msfabricpysdkcore-0.1.5.dist-info/RECORD,,

{msfabricpysdkcore-0.1.3.dist-info → msfabricpysdkcore-0.1.5.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (71.1.0)
+Generator: setuptools (72.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

msfabricpysdkcore/tests/test_admin_apis.py

@@ -1,98 +0,0 @@
-import unittest
-from dotenv import load_dotenv
-from msfabricpysdkcore import FabricClientAdmin
-
-load_dotenv()
-
-class TestFabricClientCore(unittest.TestCase):
-
-    def __init__(self, *args, **kwargs):
-        super(TestFabricClientCore, self).__init__(*args, **kwargs)
-        self.fca = FabricClientAdmin()
-                  
-    def test_admin_api(self):
-        fca = self.fca
-
-        user_id = '1dc64c6e-7a10-4ea9-8488-85d0739a377d'
-
-        # List workspaces
-        ws = fca.list_workspaces(name="testitems")[0]
-
-        self.assertEqual(ws.name, "testitems")
-
-        # Get workspace
-        ws_clone = fca.get_workspace(workspace_id=ws.id)
-
-        self.assertEqual(ws.id, ws_clone.id)
-
-        # List workspace access details
-
-        ws_access = fca.list_workspace_access_details(ws.id)
-        principials = ws_access["accessDetails"]
-        principials_ids = [p["principal"]["id"] for p in principials]
-        self.assertIn(user_id, principials_ids)
-
-        # Get access entities
-
-        access_entities = fca.list_access_entities(user_id, type="Notebook")
-        self.assertGreater(len(access_entities), 0)
-
-        # Get tenant settings
-
-        tenant_settings = fca.list_tenant_settings()
-        self.assertGreater(len(tenant_settings["tenantSettings"]), 0)
-
-        # Get capacity tenant settings overrides
-
-        overrides = fca.list_capacities_tenant_settings_overrides()
-        self.assertGreater(len(overrides), -1)
-
-        # List items
-
-        item_list = fca.list_items(workspace_id=ws.id)
-        self.assertGreater(len(item_list), 0)
-
-        # Get item
-
-        item = fca.get_item(workspace_id=ws.id, item_id=item_list[0].id)
-        self.assertEqual(item.id, item_list[0].id)
-
-        # Get item access details
-
-        item_access = fca.list_item_access_details(workspace_id=ws.id, item_id=item_list[0].id)
-        principials = item_access["accessDetails"]
-
-        principials_ids = [p["principal"]["id"] for p in principials]
-
-        self.assertIn(user_id, principials_ids)
-
-
-    def test_labels(self):
-
-        fca = self.fca
-
-        items = [{"id": "9cdd3192-bcd0-4cbe-b945-29f5964e7ab7", "type": "Lakehouse"}]
-        label_id = "defa4170-0d19-0005-0007-bc88714345d2"
-        resp = fca.bulk_set_labels(items=items, label_id=label_id)
-        self.assertEqual(resp["itemsChangeLabelStatus"][0]["status"], "Succeeded")
-        resp = fca.bulk_remove_labels(items=items)
-        self.assertEqual(resp["itemsChangeLabelStatus"][0]["status"], "Succeeded")
-
-    def test_admin_external_data_shares(self):
-
-        fca = self.fca
-
-        data_shares = fca.list_external_data_shares()
-        ws_id = "63aa9e13-4912-4abe-9156-8a56e565b7a3"
-
-        data_shares = [d for d in data_shares if d['workspaceId'] == ws_id]
-
-        self.assertGreater(len(data_shares), 0)
-        fca.revoke_external_data_share(external_data_share_id = data_shares[0]['id'], 
-                                       item_id = data_shares[0]['itemId'], 
-                                       workspace_id = data_shares[0]['workspaceId'])
-        data_shares = fca.list_external_data_shares()
-
-        data_shares = [d for d in data_shares if d['workspaceId'] == ws_id]
-
-        self.assertEqual(data_shares[0]['status'], 'Revoked')

msfabricpysdkcore/tests/test_datapipelines.py

@@ -1,45 +0,0 @@
-import unittest
-from datetime import datetime
-from dotenv import load_dotenv
-from time import sleep
-from msfabricpysdkcore.coreapi import FabricClientCore
-
-load_dotenv()
-
-class TestFabricClientCore(unittest.TestCase):
-
-    def __init__(self, *args, **kwargs):
-        super(TestFabricClientCore, self).__init__(*args, **kwargs)
-        #load_dotenv()
-        self.fc = FabricClientCore()
-
-    def test_data_pipelines(self):
-
-        fc = self.fc
-        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
-        datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
-        pipeline_name = f"pipeline_{datetime_str}"
-
-        dp = fc.create_data_pipeline(workspace_id, display_name=pipeline_name, description="asda")
-        dp.update_definition(dp.definition)
-
-        dps = fc.list_data_pipelines(workspace_id)
-        dp_names = [dp.display_name for dp in dps]
-        self.assertGreater(len(dps), 0)
-        self.assertIn(pipeline_name, dp_names)
-
-        self.assertEqual(dp.display_name, pipeline_name)
-        pipeline_name2 = f"pipeline_{datetime_str}_2"
-        dp2 = fc.update_data_pipeline(workspace_id, dp.id, display_name=pipeline_name2, return_item=True)
-
-        dp = fc.get_data_pipeline(workspace_id, data_pipeline_id=dp.id)
-        self.assertEqual(dp.display_name, pipeline_name2)
-        self.assertEqual(dp.id, dp2.id)
-
-        dp2 = fc.update_data_pipeline(workspace_id, dp.id, display_name=pipeline_name, return_item=True)
-
-        dp = fc.get_data_pipeline(workspace_id, data_pipeline_id=dp.id)
-        self.assertEqual(dp.display_name, pipeline_name)
-        self.assertEqual(dp.id, dp2.id)
-        status_code = fc.delete_data_pipeline(workspace_id, dp.id)
-        self.assertEqual(status_code, 200)

msfabricpysdkcore/tests/test_deployment_pipeline.py

@@ -1,63 +0,0 @@
-import unittest
-from msfabricpysdkcore.coreapi import FabricClientCore
-from dotenv import load_dotenv
-
-load_dotenv()
-
-
-class TestFabricClientCore(unittest.TestCase):
-
-    def __init__(self, *args, **kwargs):
-        super(TestFabricClientCore, self).__init__(*args, **kwargs)
-        self.fc = FabricClientCore()
-        
-
-
-    def test_spark_workspace_custom_pools(self):
-        fc = self.fc
-
-        dep_pipes = fc.list_deployment_pipelines()
-
-        self.assertGreater(len(dep_pipes), 0)
-
-        self.assertIn("sdkpipe", [pipe.display_name for pipe in dep_pipes])
-
-        for pipe in dep_pipes:
-            if pipe.display_name == 'sdkpipe':
-                pipe_id = pipe.id
-                break
-
-        pipe = fc.get_deployment_pipeline(pipe_id)
-
-        self.assertEqual(pipe.display_name, 'sdkpipe')
-        self.assertEqual(pipe.id, pipe_id)
-
-        stages = fc.list_deployment_pipeline_stages(pipe_id)
-
-        self.assertGreater(len(stages), 0)
-        names = [stage.display_name for stage in stages]
-        self.assertIn("Development", names)
-        self.assertIn("Production", names)
-
-        dev_stage = [stage for stage in stages if stage.display_name == "Development"][0]
-        prod_stage = [stage for stage in stages if stage.display_name == "Production"][0]
-
-        items = fc.list_deployment_pipeline_stages_items(deployment_pipeline_id=pipe_id, stage_id=dev_stage.id)
-        
-        self.assertGreater(len(items), 0)
-        self.assertIn("cicdlakehouse", [item["itemDisplayName"] for item in items])
-       
-        items = [item for item in dev_stage.list_items() if item["itemDisplayName"] == 'cicdlakehouse']
-        item = items[0]
-        item = {"sourceItemId": item["itemId"],
-                "itemType": item["itemType"]}
-        items = [item]
-
-        response = fc.deploy_stage_content(deployment_pipeline_id=pipe_id, source_stage_id=dev_stage.id,target_stage_id=prod_stage.id, items=items)
-
-        self.assertEqual(response["status"], "Succeeded")
-
-        
-
-
-

msfabricpysdkcore/tests/test_domains.py

@@ -1,126 +0,0 @@
-import unittest
-from dotenv import load_dotenv
-from datetime import datetime
-from msfabricpysdkcore import FabricClientCore, FabricClientAdmin
-
-
-load_dotenv()
-
-class TestFabricClientCore(unittest.TestCase):
-
-    def __init__(self, *args, **kwargs):
-        super(TestFabricClientCore, self).__init__(*args, **kwargs)
-
-                  
-    def test_domains(self):
-        fcc = FabricClientCore()
-        fca = FabricClientAdmin()
-
-        ws = fcc.get_workspace_by_name("sdktestdomains")
-        cap = fcc.get_capacity(capacity_id=ws.capacity_id)
-        principal = {'id': '1dc64c6e-7a10-4ea9-8488-85d0739a377d',  'type': 'User'}
-
-        # Delete if exists
-        try:
-            domain = fca.get_domain_by_name("sdktestdomains")
-            domain.delete()
-        except:
-            pass
-        try:
-            domain = fca.get_domain_by_name("sdktestdomains2")
-            domain.delete()
-        except:
-            pass
-
-        # Create domain
-        domain_name = "sdktestdomains" + datetime.now().strftime("%Y%m%d%H%M%S")
-        domain = fca.create_domain(display_name=domain_name)
-        self.assertIsNotNone(domain.id)
-        self.assertEqual(domain.display_name, domain_name)
-
-        # Get domain by name
-        domain_clone = fca.get_domain_by_name(domain_name)
-        self.assertIsNotNone(domain_clone.id)
-        self.assertEqual(domain_clone.display_name, domain_name)
-
-        # Get domain by id
-        domain_clone = fca.get_domain_by_id(domain.id)
-        self.assertIsNotNone(domain_clone.id)
-        self.assertEqual(domain_clone.display_name, domain_name)
-
-        # List domains
-        domains = fca.list_domains()
-        self.assertGreater(len(domains), 0)
-        domains_ids = [d.id for d in domains]
-        self.assertIn(domain.id, domains_ids)
-
-        # Update domain
-        domain_new_name = f"{domain_name}2"
-        domain_clone = fca.update_domain(domain.id, display_name=domain_new_name, return_item=True)
-        self.assertEqual(domain_clone.display_name, domain_new_name)
-
-        # Assign domain workspaces by Ids
-        status_code = fca.assign_domain_workspaces_by_ids(domain.id, [ws.id])
-        self.assertEqual(status_code, 200)
-
-        # List domain workspaces
-        workspaces = fca.list_domain_workspaces(domain.id, workspace_objects=True)
-        self.assertGreater(len(workspaces), 0)
-        workspaces_ids = [w.id for w in workspaces]
-        self.assertIn(ws.id, workspaces_ids)
-
-        # Unassign domain workspaces by ids
-        status_code = fca.unassign_domain_workspaces_by_ids(domain.id, [ws.id])
-        self.assertEqual(status_code, 200)
-
-        workspaces = fca.list_domain_workspaces(domain.id)
-        self.assertEqual(len(workspaces), 0)
-
-        # Assign domain workspaces by capacities
-        status_code = fca.assign_domain_workspaces_by_capacities(domain.id, [cap.id])
-        self.assertEqual(status_code, 202)
-
-        workspaces = fca.list_domain_workspaces(domain.id, workspace_objects=True)
-        self.assertGreater(len(workspaces), 0)
-        workspaces_ids = [w.id for w in workspaces]
-        self.assertIn(ws.id, workspaces_ids)
-
-        # Unassign all domain workspaces
-        status_code = fca.unassign_all_domain_workspaces(domain.id)
-        self.assertEqual(status_code, 200)
-
-        workspaces = fca.list_domain_workspaces(domain.id)
-        self.assertEqual(len(workspaces), 0)
-
-        # Assign domain workspaces by principals
-        status_code = fca.assign_domains_workspaces_by_principals(domain.id, [principal], wait_for_completion=True)
-
-        self.assertEqual(status_code, 202)
-
-        workspaces = fca.list_domain_workspaces(domain.id, workspace_objects=True)
-        self.assertGreater(len(workspaces), 0)
-        workspaces_ids = [w.id for w in workspaces]
-        self.assertIn(ws.id, workspaces_ids)
-
-        # Role assignments bulk assign
-        
-        principal_2 = {'id': 'e0505016-ef55-4ca7-b106-e085cc201823', 'type': 'User'}
-        principals = [principal, principal_2]
-
-        status_code = fca.role_assignments_bulk_assign(domain.id, "Contributors", principals)
-
-        self.assertEqual(status_code, 200)
-
-        # Role assignments bulk unassign
-        status_code = fca.role_assignments_bulk_unassign(domain.id, "Contributors", [principal_2])
-
-        self.assertEqual(status_code, 200)
-
-        # Delete domain
-        status_code = fca.delete_domain(domain.id)
-
-        self.assertEqual(status_code, 200)
-
-        domains = fca.list_domains()
-        domains_ids = [d.id for d in domains]
-        self.assertNotIn(domain.id, domains_ids)

msfabricpysdkcore/tests/test_environments.py

@@ -1,114 +0,0 @@
-import unittest
-from datetime import datetime
-from dotenv import load_dotenv
-from time import sleep
-from msfabricpysdkcore.coreapi import FabricClientCore
-
-load_dotenv()
-
-class TestFabricClientCore(unittest.TestCase):
-
-    def __init__(self, *args, **kwargs):
-        super(TestFabricClientCore, self).__init__(*args, **kwargs)
-        self.fc = FabricClientCore()
-
-    def test_environments_crudl(self):
-        fc = self.fc
-        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
-        datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
-
-        env_name = "env" + datetime_str
-        environment1 = fc.create_environment(workspace_id, display_name=env_name)
-        self.assertEqual(environment1.display_name, env_name)
-
-        environments = fc.list_environments(workspace_id)
-        environment_names = [env.display_name for env in environments]
-        self.assertGreater(len(environments), 0)
-        self.assertIn(env_name, environment_names)
-
-        env = fc.get_environment(workspace_id, environment_name=env_name)
-        self.assertIsNotNone(env.id)
-        self.assertEqual(env.display_name, env_name)
-        new_name = env_name + "2"
-        env2 = fc.update_environment(workspace_id, env.id, display_name=new_name, return_item=True)
-
-        env = fc.get_environment(workspace_id, environment_id=env.id)
-        self.assertEqual(env.display_name, new_name)
-        self.assertEqual(env.id, env2.id)
-
-        status_code = fc.delete_environment(workspace_id, env.id)
-        self.assertEqual(status_code, 200)
-
-    def test_environment_details(self):
-        fc = FabricClientCore()
-        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
-        environment_id = '5648be67-28fa-48b6-9d1f-3c87c3704d3c'
-        published_settings = fc.get_published_settings(workspace_id=workspace_id, environment_id=environment_id)
-        self.assertIsNotNone(published_settings)
-        self.assertIn("instancePool", published_settings)
-        self.assertIn("dynamicExecutorAllocation", published_settings)
-        staging_settings = fc.get_staging_settings(workspace_id=workspace_id, environment_id=environment_id)
-        self.assertIsNotNone(staging_settings)
-        self.assertIn("instancePool", staging_settings)
-        self.assertIn("dynamicExecutorAllocation", staging_settings)
-        if staging_settings["driverCores"] == 8:
-            driver_cores = 4
-        else:
-            driver_cores = 8
-        updated_settings = fc.update_staging_settings(workspace_id=workspace_id, environment_id=environment_id, driver_cores=driver_cores)
-        self.assertIn("instancePool", updated_settings)
-        self.assertIn("dynamicExecutorAllocation", updated_settings)
-        self.assertEqual(updated_settings["driverCores"], driver_cores)
-        updated_settings = fc.get_staging_settings(workspace_id=workspace_id, environment_id=environment_id)
-        self.assertIn("instancePool", updated_settings)
-        self.assertIn("dynamicExecutorAllocation", updated_settings)
-        self.assertEqual(updated_settings["driverCores"], driver_cores)
-
-
-    def test_environment_spark_libraries(self):
-        fc = self.fc
-        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
-        environment_id = '5648be67-28fa-48b6-9d1f-3c87c3704d3c'
-
-        resp = fc.get_published_libraries(workspace_id, environment_id)
-        self.assertIn('customLibraries', resp)
-        self.assertIn('wheelFiles', resp['customLibraries'])
-        self.assertIn('msfabricpysdkcore-0.1.1-py3-none-any.whl', resp['customLibraries']['wheelFiles'])
-
-        resp = fc.upload_staging_library(workspace_id, environment_id, 'dummy.whl')
-        self.assertEqual(resp.status_code, 200)
-        
-        resp = fc.get_staging_libraries(workspace_id, environment_id)
-
-        self.assertIn('customLibraries', resp)
-        self.assertIn('wheelFiles', resp['customLibraries'])
-        self.assertIn('dummy.whl', resp['customLibraries']['wheelFiles'])
-        self.assertIn('staging.whl', resp['customLibraries']['wheelFiles'])
-
-        
-        resp = fc.publish_environment(workspace_id, environment_id)
-        self.assertIn('publishDetails', resp)
-        self.assertIn('state', resp['publishDetails'])
-        self.assertEqual(resp['publishDetails']['state'], 'running')
-
-        
-        resp = fc.cancel_publish(workspace_id, environment_id)
-        self.assertIn('publishDetails', resp)
-        self.assertIn('state', resp['publishDetails'])
-        self.assertEqual(resp['publishDetails']['state'], 'cancelled')
-
-        resp = fc.delete_staging_library(workspace_id, environment_id, 'dummy.whl')
-        self.assertEqual(resp.status_code, 200)
-
-        resp = fc.get_staging_libraries(workspace_id, environment_id)
-
-        self.assertIn('customLibraries', resp)
-        self.assertIn('wheelFiles', resp['customLibraries'])
-        self.assertNotIn('dummy.whl', resp['customLibraries']['wheelFiles'])
-        self.assertIn('staging.whl', resp['customLibraries']['wheelFiles'])
-
-
-        
-
-
-