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

msfabricpysdkcore/spark_custom_pool.py

@@ -1,13 +1,11 @@
 import json
-from time import sleep
-
-import requests 
+from msfabricpysdkcore.coreapi import FabricClientCore
 
 
 class SparkCustomPool:
     """Class to represent a custom pool in Microsoft Fabric"""
 
-    def __init__(self, id, name, type, node_family, node_size, auto_scale, dynamic_executor_allocation, workspace_id, auth) -> None:
+    def __init__(self, id, name, type, node_family, node_size, auto_scale, dynamic_executor_allocation, workspace_id, core_client: FabricClientCore) -> None:
         
         self.id = id
         self.name = name
@@ -18,7 +16,7 @@ class SparkCustomPool:
         self.dynamic_executor_allocation = dynamic_executor_allocation
         self.workspace_id = workspace_id
         
-        self.auth = auth
+        self.core_client = core_client
 
     def __str__(self) -> str:
         """Return a string representation of the workspace object"""
@@ -37,7 +35,7 @@ class SparkCustomPool:
     def __repr__(self) -> str:
         return self.__str__()
     
-    def from_dict(item_dict, auth):
+    def from_dict(item_dict, core_client):
         """Create Item object from dictionary"""
 
         if 'autoScale' not in item_dict:
@@ -54,55 +52,17 @@ class SparkCustomPool:
         
         return SparkCustomPool(id=item_dict['id'], name=item_dict['name'], type=item_dict['type'], node_family=item_dict['nodeFamily'],
                                 node_size=item_dict['nodeSize'], auto_scale=item_dict['autoScale'], dynamic_executor_allocation=item_dict['dynamicExecutorAllocation'],
-                                workspace_id=item_dict['workspaceId'], auth=auth)
+                                workspace_id=item_dict['workspaceId'], core_client=core_client)
 
 
     def delete(self):
         """Delete the custom pool item"""
-        # DELETE http://api.fabric.microsoft.com/v1/workspaces/{workspaceId}/spark/pools/{poolId}
-
-        url = f"https://api.fabric.microsoft.com/v1/workspaces/{self.workspace_id}/spark/pools/{self.id}"
-        for _ in range(10):
-            response = requests.delete(url=url, headers=self.auth.get_headers())
-            if response.status_code == 429:
-                print("Too many requests, waiting 10 seconds")
-                sleep(10)
-                continue
-            if response.status_code not in (200, 429):
-                raise Exception(f"Error deleting spark pool: {response.status_code}, {response.text}")
-            break
-
-        return response.status_code
+        return self.core_client.delete_workspace_custom_pool(self.workspace_id, self.id)
     
 
     def update(self, name, node_family, node_size, auto_scale, dynamic_executor_allocation):
         """Update the custom pool item"""
-        url = f"https://api.fabric.microsoft.com/v1/workspaces/{self.workspace_id}/spark/pools/{self.id}"
-        body = {}
-
-        if name is not None:
-            body['name'] = name
-        if node_family is not None:
-            body['nodeFamily'] = node_family
-        if node_size is not None:
-            body['nodeSize'] = node_size
-        if auto_scale is not None:
-            body['autoScale'] = auto_scale
-        if dynamic_executor_allocation is not None:
-            body['dynamicExecutorAllocation'] = dynamic_executor_allocation
-
-        if not body:
-            return self
-        for _ in range(10):
-            response = requests.patch(url=url, headers=self.auth.get_headers(), json=body)
-            if response.status_code == 429:
-                print("Too many requests, waiting 10 seconds")
-                sleep(10)
-                continue
-            if response.status_code not in (200, 429):
-                raise Exception(f"Error updating item: {response.status_code}, {response.text}")
-            break
-
+        _ = self.core_client.update_workspace_custom_pool(self.workspace_id, self.id, name, node_family, node_size, auto_scale, dynamic_executor_allocation)
         if name is not None:
             self.name = name
         if node_family is not None:

msfabricpysdkcore/tests/test_admin_apis.py

@@ -25,7 +25,7 @@ class TestFabricClientCore(unittest.TestCase):
 
         self.assertEqual(ws.id, ws_clone.id)
 
-        # Get workspace access details
+        # List workspace access details
 
         ws_access = fca.list_workspace_access_details(ws.id)
         principials = ws_access["accessDetails"]
@@ -77,3 +77,23 @@ class TestFabricClientCore(unittest.TestCase):
         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 = fca.list_workspaces(name="testworkspace")[0]
+
+        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]
+
+        self.assertEqual(data_shares[0]['status'], 'Revoked')

msfabricpysdkcore/tests/test_datapipelines.py

@@ -17,32 +17,29 @@ class TestFabricClientCore(unittest.TestCase):
 
         fc = self.fc
         workspace_id = 'd8a5abe0-9eed-406d-ab46-343bc57ddbe5'
-        
+        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("pipeline1", dp_names)
-
-        dp = fc.get_data_pipeline(workspace_id, data_pipeline_name="pipeline1")
-        self.assertIsNotNone(dp.id)
-        self.assertIsNotNone(dp.definition)
-        self.assertEqual(dp.display_name, "pipeline1")
-
-        dp_new = fc.create_data_pipeline(workspace_id, display_name="pipeline_new", description="asda")
-        dp_new.update_definition(dp.definition)
-        
-        self.assertEqual(dp_new.display_name, "pipeline_new")
+        self.assertIn(pipeline_name, dp_names)
 
-        dp2 = fc.update_data_pipeline(workspace_id, dp.id, display_name="pipeline2")
+        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, "pipeline2")
+        self.assertEqual(dp.display_name, pipeline_name2)
         self.assertEqual(dp.id, dp2.id)
 
-        dp2 = fc.update_data_pipeline(workspace_id, dp.id, display_name="pipeline1")
+        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, "pipeline1")
+        self.assertEqual(dp.display_name, pipeline_name)
         self.assertEqual(dp.id, dp2.id)
-        status_code = fc.delete_data_pipeline(workspace_id, dp_new.id)
+        status_code = fc.delete_data_pipeline(workspace_id, dp.id)
         self.assertEqual(status_code, 200)

msfabricpysdkcore/tests/test_deployment_pipeline.py

@@ -32,7 +32,7 @@ class TestFabricClientCore(unittest.TestCase):
         self.assertEqual(pipe.display_name, 'sdkpipe')
         self.assertEqual(pipe.id, pipe_id)
 
-        stages = fc.get_deployment_pipeline_stages(pipe_id)
+        stages = fc.list_deployment_pipeline_stages(pipe_id)
 
         self.assertGreater(len(stages), 0)
         names = [stage.display_name for stage in stages]
@@ -42,12 +42,12 @@ class TestFabricClientCore(unittest.TestCase):
         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.get_deployment_pipeline_stages_items(pipeline_id=pipe_id, stage_id=dev_stage.id)
+        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.get_items() if item["itemDisplayName"] == 'cicdlakehouse']
+        items = [item for item in dev_stage.list_items() if item["itemDisplayName"] == 'cicdlakehouse']
         item = items[0]
         item = {"sourceItemId": item["itemId"],
                 "itemType": item["itemType"]}

msfabricpysdkcore/tests/test_domains.py

@@ -1,5 +1,6 @@
 import unittest
 from dotenv import load_dotenv
+from datetime import datetime
 from msfabricpysdkcore import FabricClientCore, FabricClientAdmin
 
 
@@ -32,7 +33,7 @@ class TestFabricClientCore(unittest.TestCase):
             pass
 
         # Create domain
-        domain_name = "sdktestdomains"
+        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)
@@ -54,8 +55,8 @@ class TestFabricClientCore(unittest.TestCase):
         self.assertIn(domain.id, domains_ids)
 
         # Update domain
-        domain_new_name = "sdktestdomains2"
-        domain_clone = fca.update_domain(domain.id, display_name=domain_new_name)
+        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

msfabricpysdkcore/tests/test_environments.py

@@ -30,7 +30,7 @@ class TestFabricClientCore(unittest.TestCase):
         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)
+        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)
@@ -62,4 +62,53 @@ class TestFabricClientCore(unittest.TestCase):
         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)
+        self.assertEqual(updated_settings["driverCores"], driver_cores)
+
+
+    def test_environment_spark_libraries(self):
+        fc = self.fc
+        workspace_id = 'd8a5abe0-9eed-406d-ab46-343bc57ddbe5'
+        environment_id = 'fae6d1a7-d671-4091-89b1-f42626deb56f'
+
+        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'])
+
+        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'])
+
+
+        
+
+
+

msfabricpysdkcore/tests/test_evenhouses.py

@@ -0,0 +1,48 @@
+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()
+        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
+        self.item_type = "Notebook"
+
+
+
+    def test_eventhouses(self):
+            
+        fc = self.fc
+        workspace_id = 'd8a5abe0-9eed-406d-ab46-343bc57ddbe5'
+        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)
+        self.assertEqual(eventhouse1.display_name, eventhouse_name)
+        
+        eventhouses = fc.list_eventhouses(workspace_id)
+        eventhouse_names = [eh.display_name for eh in eventhouses]
+        self.assertGreater(len(eventhouses), 0)
+        self.assertIn(eventhouse_name, eventhouse_names)
+
+        eh = fc.get_eventhouse(workspace_id, eventhouse_name=eventhouse_name)
+        self.assertIsNotNone(eh.id)
+        self.assertEqual(eh.display_name, eventhouse_name)
+        new_display_name = eventhouse_name + "2"
+        eh2 = fc.update_eventhouse(workspace_id, eh.id, display_name=new_display_name, return_item=True)
+
+        eh = fc.get_eventhouse(workspace_id, eventhouse_id=eh.id)
+        self.assertEqual(eh.display_name, new_display_name)
+        self.assertEqual(eh.id, eh2.id)
+
+        status_code = fc.delete_eventhouse(workspace_id, eh.id)
+        self.assertEqual(status_code, 200)

msfabricpysdkcore/tests/test_evenstreams.py

@@ -34,7 +34,7 @@ class TestFabricClientCore(unittest.TestCase):
             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")
+            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")

msfabricpysdkcore/tests/test_external_data_shares.py

@@ -0,0 +1,51 @@
+import unittest
+from dotenv import load_dotenv
+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_external_data_shares(self):
+        
+        fc = self.fc
+
+        workspace_id = 'c3352d34-0b54-40f0-b204-cc964b1beb8d'
+        item_id = 'e2c09c89-bf97-4f71-bdeb-36338795ec36'
+
+        recipient = {
+            "userPrincipalName": "lisa4@fabrikam.com"
+        }
+        paths=["Files/external"]
+
+        resp = fc.create_external_data_share(workspace_id, item_id, paths, recipient)
+        self.assertIsNotNone(resp)
+        self.assertIn('id', resp)
+
+
+        get = fc.get_external_data_share(workspace_id, item_id, resp['id'])
+        self.assertIsNotNone(get)
+        self.assertEqual(get['id'], resp['id'])
+
+
+        resp = fc.list_external_data_shares_in_item(workspace_id, item_id)
+        self.assertGreater(len(resp), 0)
+
+        data_share_ids = [ds['id'] for ds in resp]
+        self.assertIn(get['id'], data_share_ids)
+
+
+        resp = fc.revoke_external_data_share(workspace_id, item_id, get['id'])
+        self.assertEqual(resp, 200)
+
+        get2 = fc.get_external_data_share(workspace_id, item_id, get['id'])
+        self.assertIsNotNone(get2)
+        
+        self.assertEqual(get['id'], get2['id'])
+        self.assertEqual(get2['status'], 'Revoked')
+
+

msfabricpysdkcore/tests/test_items.py

@@ -0,0 +1,80 @@
+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()
+        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
+        self.item_type = "Notebook"
+    
+    def test_item_end_to_end(self):
+
+        item = self.fc.create_item(display_name=self.item_name, type=self.item_type, workspace_id=self.workspace_id) 
+        self.assertEqual(item.display_name, self.item_name)
+        self.assertEqual(item.type, self.item_type)
+        self.assertEqual(item.workspace_id, self.workspace_id)
+        self.assertEqual(item.description, "")
+
+        item = self.fc.get_item(workspace_id=self.workspace_id, item_id=item.id)
+        item_ = self.fc.get_item(workspace_id=self.workspace_id,
+                                  item_name=self.item_name, item_type=self.item_type)
+        self.assertEqual(item.id, item_.id)
+        self.assertEqual(item.display_name, self.item_name)
+        self.assertEqual(item.type, self.item_type)
+        self.assertEqual(item.workspace_id, self.workspace_id)
+        self.assertEqual(item.description, "")
+
+        item_list = self.fc.list_items(workspace_id=self.workspace_id)
+        self.assertTrue(len(item_list) > 0)
+
+        item_ids = [item_.id for item_ in item_list]
+        self.assertIn(item.id, item_ids)
+
+        self.fc.update_item(workspace_id=self.workspace_id, item_id=item.id, display_name=f"u{self.item_name}", return_item=True)
+        item = self.fc.get_item(workspace_id=self.workspace_id, item_id=item.id)
+        self.assertEqual(item.display_name, f"u{self.item_name}")
+
+        status_code = self.fc.delete_item(workspace_id=self.workspace_id, item_id=item.id)
+
+        self.assertAlmostEqual(status_code, 200)
+
+    def test_item_definition(self):
+
+        sjd = self.fc.get_item(workspace_id=self.workspace_id, item_name="blubb", item_type="SparkJobDefinition")
+        self.assertIsNotNone(sjd.definition)
+        datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
+        blubb2 = "blubb2" + datetime_str
+        blubb3 = "blubb3" + datetime_str
+        blubb2 = self.fc.create_item(display_name=blubb2, type="SparkJobDefinition", workspace_id=self.workspace_id,
+                                   definition=sjd.definition)
+
+        blubb3 =  self.fc.create_item(display_name=blubb3, type="SparkJobDefinition", workspace_id=self.workspace_id)
+
+        response = self.fc.update_item_definition(workspace_id=self.workspace_id,
+                                                item_id=blubb3.id, definition=sjd.definition)
+        
+        self.assertEqual(response.status_code, 200)
+        blubb3 = self.fc.get_item(workspace_id=self.workspace_id, item_id=blubb3.id)
+        self.assertEqual(blubb3.definition, sjd.definition)
+        
+        self.assertNotEqual(blubb2.id, sjd.id)
+        self.assertEqual(blubb2.definition, sjd.definition)
+        self.assertNotEqual(blubb2.id, blubb3.id)
+        
+        blubb2.delete()
+        blubb3.delete()
+
+
+if __name__ == "__main__":
+    unittest.main()

msfabricpysdkcore/tests/test_kql_queryset.py

@@ -0,0 +1,50 @@
+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()
+        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
+        self.item_type = "Notebook"
+    
+    def test_kql_querysets(self):
+
+        fc = self.fc
+        workspace_id = 'd8a5abe0-9eed-406d-ab46-343bc57ddbe5'
+
+        kql_queryset_name = "kqlqueryset1"
+
+        kql_querysets = fc.list_kql_querysets(workspace_id)
+        kql_queryset_names = [kqlq.display_name for kqlq in kql_querysets]
+        self.assertGreater(len(kql_querysets), 0)
+        self.assertIn(kql_queryset_name, kql_queryset_names)
+
+        kqlq = fc.get_kql_queryset(workspace_id, kql_queryset_name=kql_queryset_name)
+        self.assertIsNotNone(kqlq.id)
+        self.assertEqual(kqlq.display_name, kql_queryset_name)
+
+        kqlq2 = fc.update_kql_queryset(workspace_id, kqlq.id, display_name=f"{kql_queryset_name}2", return_item=True)
+
+        kqlq = fc.get_kql_queryset(workspace_id, kql_queryset_id=kqlq.id)
+        self.assertEqual(kqlq.display_name, f"{kql_queryset_name}2")
+        self.assertEqual(kqlq.id, kqlq2.id)
+
+        kqlq2 = fc.update_kql_queryset(workspace_id, kqlq.id, display_name=kql_queryset_name, return_item=True)
+
+        kqlq = fc.get_kql_queryset(workspace_id, kql_queryset_id=kqlq.id)
+        self.assertEqual(kqlq.display_name, kql_queryset_name)
+        self.assertEqual(kqlq.id, kqlq2.id)
+
+        # status_code = fc.delete_kql_queryset(workspace_id, kqlq.id)
+        # self.assertEqual(status_code, 200)

msfabricpysdkcore/tests/test_kqldatabases.py

@@ -38,7 +38,7 @@ class TestFabricClientCore(unittest.TestCase):
         self.assertEqual(kqldb.display_name, kqldb_name)
         
         new_name = kqldb_name+"2"
-        kqldb2 = fc.update_kql_database(workspace_id, kqldb.id, display_name=new_name)
+        kqldb2 = fc.update_kql_database(workspace_id, kqldb.id, display_name=new_name, return_item=True)
 
         kqldb = fc.get_kql_database(workspace_id, kql_database_id=kqldb.id)
         self.assertEqual(kqldb.display_name, new_name)

msfabricpysdkcore/tests/test_lakehouse.py

@@ -0,0 +1,86 @@
+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()
+        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
+        self.item_type = "Notebook"
+
+    def test_lakehouse(self):
+
+        lakehouse = self.fc.get_item(workspace_id=self.workspace_id, item_name="lakehouse1", item_type="Lakehouse")
+        self.assertIsNotNone(lakehouse.properties)
+        lakehouse_id = lakehouse.id
+        workspace_id = self.workspace_id
+        date_str = datetime.now().strftime("%Y%m%d%H%M%S")
+        table_name = f"table{date_str}"
+
+
+        status_code = self.fc.load_table(workspace_id=self.workspace_id, lakehouse_id=lakehouse_id, table_name=table_name, 
+                                         path_type="File", relative_path="Files/folder1/titanic.csv")
+
+        self.assertEqual(status_code, 202)
+
+        # Run on demand table maintenance
+        table_name_maintenance = "table20240515114529"
+
+        execution_data = {
+            "tableName": table_name_maintenance,
+            "optimizeSettings": {
+            "vOrder": True,
+            "zOrderBy": [
+                "tipAmount"
+            ]
+            },
+            "vacuumSettings": {
+            "retentionPeriod": "7:01:00:00"
+            }
+        }
+        
+        response = self.fc.run_on_demand_table_maintenance(workspace_id=workspace_id, lakehouse_id=lakehouse_id, 
+                                                           execution_data = execution_data,
+                                                           job_type = "TableMaintenance", wait_for_completion = False)
+        self.assertIn(response.status_code, [200, 202])
+
+        table_list = self.fc.list_tables(workspace_id=self.workspace_id, lakehouse_id=lakehouse_id)
+        table_names = [table["name"] for table in table_list]
+
+        self.assertIn(table_name, table_names)
+
+        fc = self.fc
+        workspace_id = 'd8a5abe0-9eed-406d-ab46-343bc57ddbe5'
+
+        lakehouse = fc.create_lakehouse(workspace_id=workspace_id, display_name="lakehouse2")
+        self.assertIsNotNone(lakehouse.id)
+
+        lakehouses = fc.list_lakehouses(workspace_id)
+        lakehouse_names = [lh.display_name for lh in lakehouses]
+        self.assertGreater(len(lakehouse_names), 0)
+        self.assertIn("lakehouse2", lakehouse_names)
+
+        lakehouse2 = fc.get_lakehouse(workspace_id=workspace_id, lakehouse_id=lakehouse.id)
+        self.assertEqual(lakehouse.id, lakehouse2.id)
+
+        sleep(20)
+        lakehouse2 = fc.update_lakehouse(workspace_id=workspace_id, lakehouse_id=lakehouse.id, display_name="lakehouse3", return_item=True)
+        self.assertEqual(lakehouse2.display_name, "lakehouse3")
+
+        id = lakehouse2.id
+
+        lakehouse2 = fc.get_lakehouse(workspace_id=workspace_id, lakehouse_name="lakehouse3")
+        self.assertEqual(lakehouse2.id, id)
+
+        status_code = fc.delete_lakehouse(workspace_id=workspace_id, lakehouse_id=lakehouse.id)
+        self.assertEqual(status_code, 200)