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

msfabricpysdkcore/fabric_azure_client.py

@@ -0,0 +1,232 @@
+from warnings import warn
+
+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/",
+                         tenant_id = tenant_id,
+                         client_id = client_id,
+                         client_secret = client_secret,)
+
+        if silent is not None:
+            warn("The 'silent' parameter is deprecated and will be removed in a future version.", DeprecationWarning, stacklevel=2)
+
+
+    def check_name_availability(self, subscription_id, location, name, type = "Microsoft.Fabric/capacities"):
+        """Check name availability
+        Args:
+            subscription_id (str): The subscription ID
+            location (str): The location
+            name (str): The name
+            type (str): The type
+        Returns:
+            dict: The response
+        """
+
+        url = f"https://management.azure.com/subscriptions/{subscription_id}/providers/Microsoft.Fabric/locations/{location}/checkNameAvailability?api-version=2023-11-01"
+
+        body = {
+            "name": name,
+            "type": type
+        }
+
+        response = self.calling_routine(url=url, operation="POST", body=body, response_codes=[200], return_format="json", error_message="Failed to check name availability")
+        return response
+
+    def create_or_update_capacity(self, subscription_id, resource_group_name, capacity_name, location, properties_administration, sku, tags = None):
+        """Create or update capacity
+        Args:
+            subscription_id (str): The subscription ID
+            resource_group_name (str): The resource group name
+            capacity_name (str): The capacity name
+            location (str): The location
+            properties_administration (dict): The administration properties
+            sku (dict): The sku
+            tags (dict): The tags
+        Returns:
+            FabricAzureCapacity: The capacity
+        """
+        from msfabricpysdkcore.fabric_azure_capacity import FabricAzureCapacity
+
+        if sku and "name" in sku:
+            sku = sku["name"]
+
+        body = {
+            "location": location,
+            "properties": {
+                "administration": properties_administration
+            },
+            "sku": {"name": sku,
+                    "tier": "Fabric"}
+        }
+
+        if tags is not None:
+            body["tags"] = tags
+        url = f"https://management.azure.com/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.Fabric/capacities/{capacity_name}?api-version=2023-11-01"
+
+        response = self.calling_routine(url=url, operation="PUT", body=body, response_codes=[200, 201], return_format="json", error_message="Failed to create or update capacity")
+        response["subscription_id"] = subscription_id
+        response["resource_group_name"] = resource_group_name
+
+        return FabricAzureCapacity.from_dict(response, self)
+
+
+    def delete_capacity(self, subscription_id, resource_group_name, capacity_name):
+        """Delete capacity
+        Args:
+            subscription_id (str): The subscription ID
+            resource_group_name (str): The resource group name
+            capacity_name (str): The capacity name
+        Returns:
+            dict: The response
+        """
+
+        url = f"https://management.azure.com/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.Fabric/capacities/{capacity_name}?api-version=2023-11-01"
+
+        response = self.calling_routine(url=url, operation="DELETE", response_codes=[202], return_format="response", error_message="Failed to delete capacity")
+        return response
+
+
+    def get_capacity(self, subscription_id, resource_group_name, capacity_name):
+        """Get capacity
+        Args:
+            subscription_id (str): The subscription ID
+            resource_group_name (str): The resource group name
+            capacity_name (str): The capacity name
+        Returns:
+            FabricAzureCapacity: The capacity
+        """
+        from msfabricpysdkcore.fabric_azure_capacity import FabricAzureCapacity
+
+        url = f"https://management.azure.com/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.Fabric/capacities/{capacity_name}?api-version=2023-11-01"
+
+        response = self.calling_routine(url=url, operation="GET", response_codes=[200], return_format="json", error_message="Failed to get capacity")
+        response["subscription_id"] = subscription_id
+        response["resource_group_name"] = resource_group_name
+
+        return FabricAzureCapacity.from_dict(response, self)
+
+    def list_by_resource_group(self, subscription_id, resource_group_name):
+        """List capacities by resource group
+        Args:
+            subscription_id (str): The subscription ID
+            resource_group_name (str): The resource group name
+        Returns:
+            dict: The response
+        """
+
+        url = f"https://management.azure.com/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.Fabric/capacities?api-version=2023-11-01"
+
+        response = self.calling_routine(url=url, operation="GET", response_codes=[200], return_format="value_json", error_message="Failed to list capacities by resource group")
+        return response
+
+
+    def list_by_subscription(self, subscription_id):
+        """List capacities by subscription
+        Args:
+            subscription_id (str): The subscription ID
+        Returns:
+            dict: The response
+        """
+
+        url = f"https://management.azure.com/subscriptions/{subscription_id}/providers/Microsoft.Fabric/capacities?api-version=2023-11-01"
+
+        response = self.calling_routine(url=url, operation="GET", response_codes=[200], return_format="value_json", error_message="Failed to list capacities by subscription")
+        return response
+
+    def list_skus(self, subscription_id):
+        """List skus
+        Args:
+            subscription_id (str): The subscription ID
+        Returns:
+            dict: The response
+        """
+
+        url = f"https://management.azure.com/subscriptions/{subscription_id}/providers/Microsoft.Fabric/skus?api-version=2023-11-01"
+
+        response = self.calling_routine(url=url, operation="GET", response_codes=[200], return_format="value_json", error_message="Failed to list skus")
+        return response
+    
+    def list_skus_for_capacity(self, subscription_id, resource_group_name, capacity_name):
+        """List skus for capacity
+        Args:
+            subscription_id (str): The subscription ID
+            resource_group_name (str): The resource group name
+            capacity_name (str): The capacity name
+        Returns:
+            dict: The response
+        """
+
+        url = f"https://management.azure.com/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.Fabric/capacities/{capacity_name}/skus?api-version=2023-11-01"
+
+        response = self.calling_routine(url=url, operation="GET", response_codes=[200], return_format="value_json", error_message="Failed to list skus for capacity")
+        return response
+
+    def resume_capacity(self, subscription_id, resource_group_name, capacity_name):
+        """Resume capacity
+        Args:
+            subscription_id (str): The subscription ID
+            resource_group_name (str): The resource group name
+            capacity_name (str): The capacity name
+        Returns:
+            dict: The response
+        """
+
+        url = f"https://management.azure.com/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.Fabric/capacities/{capacity_name}/resume?api-version=2023-11-01"
+
+        response = self.calling_routine(url=url, operation="POST", response_codes=[202], return_format="response", error_message="Failed to resume capacity")
+        return response
+
+    def suspend_capacity(self, subscription_id, resource_group_name, capacity_name):
+        """Suspend capacity
+        Args:
+            subscription_id (str): The subscription ID
+            resource_group_name (str): The resource group name
+            capacity_name (str): The capacity name
+        Returns:
+            dict: The response
+        """
+
+        url = f"https://management.azure.com/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.Fabric/capacities/{capacity_name}/suspend?api-version=2023-11-01"
+
+        response = self.calling_routine(url=url, operation="POST", response_codes=[202], return_format="response", error_message="Failed to suspend capacity")
+        return response
+    
+    def update_capacity(self, subscription_id, resource_group_name, capacity_name, properties_administration = None, sku = None, tags = None):
+        """Update capacity
+        Args:
+            subscription_id (str): The subscription ID
+            resource_group_name (str): The resource group name
+            capacity_name (str): The capacity name
+            body (dict): The body of the request
+        Returns:
+            FabricAzureCapacity: The capacity
+        """
+        from msfabricpysdkcore.fabric_azure_capacity import FabricAzureCapacity
+
+        body = {}
+        if sku and "name" in sku:
+            sku = sku["name"]
+
+        if properties_administration is not None:
+            body["properties"] = {}
+            body["properties"]["administration"] = properties_administration
+
+        if sku is not None:
+            body["sku"] = {"name": sku,
+                           "tier": "Fabric"}
+        
+        if tags is not None:
+            body["tags"] = tags
+
+        url = f"https://management.azure.com/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.Fabric/capacities/{capacity_name}?api-version=2023-11-01"
+
+        response = self.calling_routine(url=url, operation="PATCH", body=body, response_codes=[200, 202],
+                                        return_format="json", error_message="Failed to update capacity")
+        response["subscription_id"] = subscription_id
+        response["resource_group_name"] = resource_group_name
+
+        return FabricAzureCapacity.from_dict(response, self)

msfabricpysdkcore/long_running_operation.py

@@ -1,16 +1,23 @@
+import logging
 from time import sleep, time
 from msfabricpysdkcore.coreapi import FabricClientCore
+from msfabricpysdkcore.util import logger
+
 
 class LongRunningOperation:
     """Class to represent a workspace in Microsoft Fabric"""
 
+    _logger: logging.Logger
+
     def __init__(self, operation_id, core_client: FabricClientCore) -> None:
+        """Initialize the LongRunningOperation object"""
+
+        self._logger = logger.getChild(__name__)
         self.operation_id = operation_id
         self.core_client = core_client
 
         self.state = self.get_operation_state()["status"]
 
-
     def get_operation_results(self):
         return self.core_client.get_operation_results(operation_id=self.operation_id)
     
@@ -26,22 +33,24 @@ class LongRunningOperation:
             if duration > 60:
                 
                 if self.state == "Running":
-                    print(f"Operation did not complete after {duration} seconds")
+                    self._logger.info(f"Operation did not complete after {duration} seconds")
                     return "Running"
-                raise Exception(f"Operation did not complete after {duration} seconds")
+                raise TimeoutError(f"Operation did not complete after {duration} seconds")
             sleep(3)
         return self.state
     
 
 def check_long_running_operation(headers, core_client):
-    """Check the status of a long running operation"""
+    """Check the status of a long-running operation"""
     location = headers.get('Location', None)
     operation_id = headers.get('x-ms-operation-id', None)
     if location:
         operation_id = location.split("/")[-1]
     
+    logger_ = logger.getChild(__name__)
+
     if not operation_id:
-        print("Operation initiated, no operation id found")
+        logger_.info("Operation initiated, no operation id found")
         return None
     lro = LongRunningOperation(operation_id=operation_id, core_client=core_client)
     lro.wait_for_completion()

msfabricpysdkcore/tests/test_admin_apis.py

@@ -13,12 +13,12 @@ class TestFabricClientCore(unittest.TestCase):
     def test_admin_api(self):
         fca = self.fca
 
-        user_id = 'b4f4e299-e6e1-4667-886c-57e4a8dde1c2'
+        user_id = '1dc64c6e-7a10-4ea9-8488-85d0739a377d'
 
         # List workspaces
-        ws = fca.list_workspaces(name="testworkspace")[0]
+        ws = fca.list_workspaces(name="testitems")[0]
 
-        self.assertEqual(ws.name, "testworkspace")
+        self.assertEqual(ws.name, "testitems")
 
         # Get workspace
         ws_clone = fca.get_workspace(workspace_id=ws.id)
@@ -71,8 +71,8 @@ class TestFabricClientCore(unittest.TestCase):
 
         fca = self.fca
 
-        items = [{"id": "d417b834-d381-454c-9cf0-c491f69508de", "type": "Lakehouse"}]
-        label_id = "defa4170-0d19-0005-000a-bc88714345d2"
+        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)
@@ -83,17 +83,16 @@ class TestFabricClientCore(unittest.TestCase):
         fca = self.fca
 
         data_shares = fca.list_external_data_shares()
-        ws = fca.list_workspaces(name="testworkspace")[0]
+        ws_id = "63aa9e13-4912-4abe-9156-8a56e565b7a3"
 
-        data_shares = [d for d in data_shares if d['workspaceId'] == ws.id]
+        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()
-        ws = fca.list_workspaces(name="testworkspace")[0]
 
-        data_shares = [d for d in data_shares if d['workspaceId'] == ws.id]
+        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

@@ -16,7 +16,7 @@ class TestFabricClientCore(unittest.TestCase):
     def test_data_pipelines(self):
 
         fc = self.fc
-        workspace_id = 'd8a5abe0-9eed-406d-ab46-343bc57ddbe5'
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
         datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
         pipeline_name = f"pipeline_{datetime_str}"
 

msfabricpysdkcore/tests/test_domains.py

@@ -18,7 +18,7 @@ class TestFabricClientCore(unittest.TestCase):
 
         ws = fcc.get_workspace_by_name("sdktestdomains")
         cap = fcc.get_capacity(capacity_id=ws.capacity_id)
-        principal = {'id': 'b4f4e299-e6e1-4667-886c-57e4a8dde1c2',  'type': 'User'}
+        principal = {'id': '1dc64c6e-7a10-4ea9-8488-85d0739a377d',  'type': 'User'}
 
         # Delete if exists
         try:
@@ -104,7 +104,7 @@ class TestFabricClientCore(unittest.TestCase):
 
         # Role assignments bulk assign
         
-        principal_2 = {'id': '6edbc444-16cd-43f4-ae48-cbe734b89657', 'type': 'User'}
+        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)

msfabricpysdkcore/tests/test_environments.py

@@ -14,7 +14,7 @@ class TestFabricClientCore(unittest.TestCase):
 
     def test_environments_crudl(self):
         fc = self.fc
-        workspace_id = 'd8a5abe0-9eed-406d-ab46-343bc57ddbe5'
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
         datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
 
         env_name = "env" + datetime_str
@@ -41,8 +41,8 @@ class TestFabricClientCore(unittest.TestCase):
 
     def test_environment_details(self):
         fc = FabricClientCore()
-        workspace_id = 'd8a5abe0-9eed-406d-ab46-343bc57ddbe5'
-        environment_id = 'fae6d1a7-d671-4091-89b1-f42626deb56f'
+        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)
@@ -67,13 +67,13 @@ class TestFabricClientCore(unittest.TestCase):
 
     def test_environment_spark_libraries(self):
         fc = self.fc
-        workspace_id = 'd8a5abe0-9eed-406d-ab46-343bc57ddbe5'
-        environment_id = 'fae6d1a7-d671-4091-89b1-f42626deb56f'
+        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.0.13-py3-none-any.whl', resp['customLibraries']['wheelFiles'])
+        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)

msfabricpysdkcore/tests/test_evenhouses.py

@@ -12,7 +12,6 @@ class TestFabricClientCore(unittest.TestCase):
         super(TestFabricClientCore, self).__init__(*args, **kwargs)
         #load_dotenv()
         self.fc = FabricClientCore()
-        self.workspace_id = "c3352d34-0b54-40f0-b204-cc964b1beb8d"
 
         datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
         self.item_name = "testitem" + datetime_str
@@ -23,7 +22,7 @@ class TestFabricClientCore(unittest.TestCase):
     def test_eventhouses(self):
             
         fc = self.fc
-        workspace_id = 'd8a5abe0-9eed-406d-ab46-343bc57ddbe5'
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
         datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
         eventhouse_name = "evh" + datetime_str
         eventhouse1 = fc.create_eventhouse(workspace_id, display_name=eventhouse_name)

msfabricpysdkcore/tests/test_evenstreams.py

@@ -15,30 +15,30 @@ class TestFabricClientCore(unittest.TestCase):
 
     def test_eventstreams(self):
 
-            fc = self.fc
-            workspace_id = 'd8a5abe0-9eed-406d-ab46-343bc57ddbe5'
+        fc = self.fc
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
 
-            datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
-            es_name = "es" + datetime_str
+        datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
+        es_name = "es" + datetime_str
 
-            eventstream = fc.create_eventstream(workspace_id, display_name=es_name)
-            self.assertEqual(eventstream.display_name, es_name)
+        eventstream = fc.create_eventstream(workspace_id, display_name=es_name)
+        self.assertEqual(eventstream.display_name, es_name)
 
-            eventstreams = fc.list_eventstreams(workspace_id)
-            eventstream_names = [es.display_name for es in eventstreams]
-            self.assertGreater(len(eventstreams), 0)
-            self.assertIn(es_name, eventstream_names)
+        eventstreams = fc.list_eventstreams(workspace_id)
+        eventstream_names = [es.display_name for es in eventstreams]
+        self.assertGreater(len(eventstreams), 0)
+        self.assertIn(es_name, eventstream_names)
 
-            
-            es = fc.get_eventstream(workspace_id, eventstream_name=es_name)
-            self.assertIsNotNone(es.id)
-            self.assertEqual(es.display_name, es_name)
+        
+        es = fc.get_eventstream(workspace_id, eventstream_name=es_name)
+        self.assertIsNotNone(es.id)
+        self.assertEqual(es.display_name, es_name)
 
-            es2 = fc.update_eventstream(workspace_id, es.id, display_name=f"{es_name}2", return_item=True)
+        es2 = fc.update_eventstream(workspace_id, es.id, display_name=f"{es_name}2", return_item=True)
 
-            es = fc.get_eventstream(workspace_id, eventstream_id=es.id)
-            self.assertEqual(es.display_name, f"{es_name}2")
-            self.assertEqual(es.id, es2.id)
+        es = fc.get_eventstream(workspace_id, eventstream_id=es.id)
+        self.assertEqual(es.display_name, f"{es_name}2")
+        self.assertEqual(es.id, es2.id)
 
-            status_code = fc.delete_eventstream(workspace_id, es.id)
-            self.assertEqual(status_code, 200)
+        status_code = fc.delete_eventstream(workspace_id, es.id)
+        self.assertEqual(status_code, 200)

msfabricpysdkcore/tests/test_external_data_shares.py

@@ -14,13 +14,13 @@ class TestFabricClientCore(unittest.TestCase):
         
         fc = self.fc
 
-        workspace_id = 'c3352d34-0b54-40f0-b204-cc964b1beb8d'
-        item_id = 'e2c09c89-bf97-4f71-bdeb-36338795ec36'
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
+        item_id = "148ef579-4a5d-4048-8a48-0a703c5e3a1a"
 
         recipient = {
             "userPrincipalName": "lisa4@fabrikam.com"
         }
-        paths=["Files/external"]
+        paths=["Files/to_share"]
 
         resp = fc.create_external_data_share(workspace_id, item_id, paths, recipient)
         self.assertIsNotNone(resp)

msfabricpysdkcore/tests/test_fabric_azure_client.py

@@ -0,0 +1,78 @@
+import unittest
+from dotenv import load_dotenv
+from datetime import datetime
+from time import sleep
+from msfabricpysdkcore import FabricAzureClient
+
+load_dotenv()
+
+
+class TestFabricClientCore(unittest.TestCase):
+
+    def __init__(self, *args, **kwargs):
+        super(TestFabricClientCore, self).__init__(*args, **kwargs)
+        #load_dotenv()
+        self.fac = FabricAzureClient()
+
+    def test_azure_capacity(self):
+
+        fac = self.fac
+
+        subscription_id = "c77cc8fc-43bb-4d44-bdc5-6e20511ed2a8"
+        resource_group_name = "fabricdemo"
+        capacity_name = "westeuropeajrederer"
+        capacity_name_new = "westeuropeajrederer" + datetime.now().strftime("%Y%m%d%H%M%S")
+
+        resp = fac.check_name_availability(subscription_id, "westeurope", capacity_name_new)
+        self.assertIn('nameAvailable', resp)
+        self.assertEqual(resp['nameAvailable'], True)
+
+        resp = fac.create_or_update_capacity(subscription_id, resource_group_name, capacity_name_new, 
+                                            location="westeurope",
+                                            properties_administration={"members": ['admin@MngEnvMCAP065039.onmicrosoft.com']},
+                                            sku = "F2")
+        self.assertIsNotNone(resp.name)
+        self.assertEqual(resp.name, capacity_name_new)
+        
+        resp = fac.get_capacity(subscription_id, resource_group_name, capacity_name_new)
+        self.assertIsNotNone(resp.name)
+        self.assertEqual(resp.name, capacity_name_new)
+
+        sku = resp.sku['name']
+        
+        resp = fac.delete_capacity(subscription_id, resource_group_name, capacity_name_new)
+        self.assertEqual(resp.status_code, 202)
+
+
+        resp = fac.list_by_resource_group(subscription_id, resource_group_name)
+        cap_names = [cap["name"] for cap in resp]
+        self.assertIn(capacity_name, cap_names)
+
+        
+
+        resp = fac.list_by_subscription(subscription_id)
+        cap_names = [cap["name"] for cap in resp]
+        self.assertIn(capacity_name, cap_names)
+
+
+        resp = fac.list_skus(subscription_id)
+        self.assertGreater(len(resp), 0, msg=f"No SKUs found: {resp}")
+
+
+        resp = fac.list_skus_for_capacity(subscription_id, resource_group_name, capacity_name)
+        self.assertGreater(len(resp), 0, msg=f"No SKUs found: {resp}")
+
+        resp = fac.resume_capacity(subscription_id, resource_group_name, capacity_name)
+        self.assertEqual(resp.status_code, 202)
+
+        sleep(30)
+        resp = fac.suspend_capacity(subscription_id, resource_group_name, capacity_name)
+        self.assertEqual(resp.status_code, 202)
+        sleep(30)
+
+        if sku != "F4":
+            resp = fac.update_capacity(subscription_id, resource_group_name, capacity_name, sku="F4")
+            self.assertEqual(resp.sku["name"], "F4")
+        else:
+            resp = fac.update_capacity(subscription_id, resource_group_name, capacity_name, sku="F2")
+            self.assertEqual(resp.sku["name"], "F2")

msfabricpysdkcore/tests/test_git.py

@@ -11,7 +11,6 @@ class TestFabricClientCore(unittest.TestCase):
         super(TestFabricClientCore, self).__init__(*args, **kwargs)
         #load_dotenv()
         self.fc = FabricClientCore()
-        self.workspace_id = "fd3ba978-0b94-43e2-9f23-f65761e9ff34"
         
     def test_git(self):
 
@@ -19,14 +18,14 @@ class TestFabricClientCore(unittest.TestCase):
         ws2_name = "git" + datetime_str
         self.fc.create_workspace(display_name=ws2_name)
         ws2 = self.fc.get_workspace_by_name(name=ws2_name)
-        self.fc.assign_to_capacity(workspace_id=ws2.id, capacity_id="a089a991-221c-49be-a41e-9020198cf7af")
-
-        git_provider_details = {'organizationName': 'dasenf1860',
-                        'projectName': 'fabrictest',
-                        'gitProviderType': 'AzureDevOps',
-                        'repositoryName': 'fabrictest',
-                        'branchName': 'main',
-                        'directoryName': '/folder1'}
+        self.fc.assign_to_capacity(workspace_id=ws2.id, capacity_id="840a6c1e-5289-4094-bbc8-716daabaeeba")
+
+        git_provider_details = {'organizationName': 'MngEnvMCAP065039',
+                                'projectName': 'fabricdevops',
+                                'gitProviderType': 'AzureDevOps',
+                                'repositoryName': 'fabricdevops',
+                                'branchName': 'main',
+                                'directoryName': '/sdkdemoRTI'}
 
         status_code = self.fc.git_connect(workspace_id=ws2.id, git_provider_details=git_provider_details)
 

msfabricpysdkcore/tests/test_items.py

@@ -12,7 +12,8 @@ class TestFabricClientCore(unittest.TestCase):
         super(TestFabricClientCore, self).__init__(*args, **kwargs)
         #load_dotenv()
         self.fc = FabricClientCore()
-        self.workspace_id = "c3352d34-0b54-40f0-b204-cc964b1beb8d"
+        self.workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
+
 
         datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
         self.item_name = "testitem" + datetime_str
@@ -51,7 +52,7 @@ class TestFabricClientCore(unittest.TestCase):
 
     def test_item_definition(self):
 
-        sjd = self.fc.get_item(workspace_id=self.workspace_id, item_name="blubb", item_type="SparkJobDefinition")
+        sjd = self.fc.get_item(workspace_id=self.workspace_id, item_name="helloworld", item_type="SparkJobDefinition")
         self.assertIsNotNone(sjd.definition)
         datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
         blubb2 = "blubb2" + datetime_str

msfabricpysdkcore/tests/test_jobs.py

@@ -11,8 +11,8 @@ class TestFabricClientCore(unittest.TestCase):
         super(TestFabricClientCore, self).__init__(*args, **kwargs)
         #load_dotenv()
         self.fc = FabricClientCore()
-        self.workspace_id = "c3352d34-0b54-40f0-b204-cc964b1beb8d"
-        self.item_id = "7e38f344-81df-4805-83b6-b9cc16830500"
+        self.workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
+        self.item_id = "38a1c15f-8a9e-49c5-8d05-a27cf9ce8b18"
 
 
     def test_jobs_end_to_end(self):

msfabricpysdkcore/tests/test_kql_queryset.py

@@ -12,7 +12,6 @@ class TestFabricClientCore(unittest.TestCase):
         super(TestFabricClientCore, self).__init__(*args, **kwargs)
         #load_dotenv()
         self.fc = FabricClientCore()
-        self.workspace_id = "c3352d34-0b54-40f0-b204-cc964b1beb8d"
 
         datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
         self.item_name = "testitem" + datetime_str
@@ -21,7 +20,7 @@ class TestFabricClientCore(unittest.TestCase):
     def test_kql_querysets(self):
 
         fc = self.fc
-        workspace_id = 'd8a5abe0-9eed-406d-ab46-343bc57ddbe5'
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
 
         kql_queryset_name = "kqlqueryset1"