msfabricpysdkcore 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

msfabricpysdkcore/tests/test_notebooks.py

@@ -0,0 +1,57 @@
+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()
+
+        datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
+        self.item_name = "testitem" + datetime_str
+        self.item_type = "Notebook"
+
+    
+    def test_notebooks(self):
+            
+        fc = self.fc
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
+
+        notebook_name = "notebook" + datetime.now().strftime("%Y%m%d%H%M%S")
+
+        notebook_w_content = fc.get_notebook(workspace_id, notebook_name="HelloWorld")
+
+        definition = fc.get_notebook_definition(workspace_id, notebook_w_content.id)
+        
+        self.assertIsNotNone(definition)
+        self.assertIn("definition", definition)
+        definition = definition["definition"]
+        notebook = fc.create_notebook(workspace_id, definition=definition, display_name=notebook_name)
+        fc.update_notebook_definition(workspace_id, notebook.id, definition=definition)
+        notebook = fc.get_notebook(workspace_id, notebook_id=notebook.id)
+        self.assertEqual(notebook.display_name, notebook_name)
+        self.assertIsNotNone(notebook.definition)
+        
+        notebooks = fc.list_notebooks(workspace_id)
+        notebook_names = [nb.display_name for nb in notebooks]
+        self.assertGreater(len(notebooks), 0)
+        self.assertIn(notebook_name, notebook_names)
+
+        nb = fc.get_notebook(workspace_id, notebook_name=notebook_name)
+        self.assertIsNotNone(nb.id)
+        self.assertEqual(nb.display_name, notebook_name)
+
+        nb2 = fc.update_notebook(workspace_id, notebook_id=nb.id, display_name=f"{notebook_name}2", return_item=True)
+
+        nb = fc.get_notebook(workspace_id, notebook_id=nb.id)
+        self.assertEqual(nb.display_name, f"{notebook_name}2")
+        self.assertEqual(nb.id, nb2.id)
+
+        status_code = fc.delete_notebook(workspace_id, nb.id)
+        self.assertEqual(status_code, 200)

msfabricpysdkcore/tests/test_one_lake_data_access_security.py

@@ -0,0 +1,63 @@
+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_one_lake_data_access_security(self):
+
+        fc = self.fc
+
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
+        item_id = "148ef579-4a5d-4048-8a48-0a703c5e3a1a"
+
+        resp = fc.list_data_access_roles(workspace_id=workspace_id, item_id=item_id)
+        self.assertEqual(len(resp), 2)
+
+        roles = resp[0]
+        etag = resp[1]
+
+        role1 = roles[1]
+
+        self.assertIn('members', role1)
+        self.assertIn('fabricItemMembers', role1['members'])
+        self.assertGreater(len(role1['members']['fabricItemMembers']), 0)
+        self.assertIn('itemAccess', role1['members']['fabricItemMembers'][0])
+
+        item_access = role1["members"]["fabricItemMembers"][0]['itemAccess']
+        item_access_old = list(item_access)
+        if 'ReadAll' in item_access:
+            item_access = ['Read', 'Write', 'Execute']
+        else:
+            item_access.append('ReadAll')
+
+        role1["members"]["fabricItemMembers"][0]['itemAccess'] = item_access
+        roles[1] = role1
+
+        resp = fc.create_or_update_data_access_roles(workspace_id=workspace_id, 
+                                                     item_id=item_id, 
+                                                     data_access_roles=roles, 
+                                                     etag_match={"If-Match":etag})
+
+
+        resp = fc.list_data_access_roles(workspace_id=workspace_id, item_id=item_id)
+        self.assertEqual(len(resp), 2)
+
+        roles = resp[0]
+        etag = resp[1]
+
+        role1 = roles[1]
+
+        self.assertIn('members', role1)
+        self.assertIn('fabricItemMembers', role1['members'])
+        self.assertGreater(len(role1['members']['fabricItemMembers']), 0)
+        self.assertIn('itemAccess', role1['members']['fabricItemMembers'][0])
+
+        item_access = role1["members"]["fabricItemMembers"][0]['itemAccess']
+        self.assertNotEqual(item_access, item_access_old)

msfabricpysdkcore/tests/test_other_items.py

@@ -0,0 +1,45 @@
+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()
+
+        datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
+        self.item_name = "testitem" + datetime_str
+        self.item_type = "Notebook"
+    
+    def test_list_other_items(self):
+
+        fc = self.fc
+
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
+
+        list_dashboards = fc.list_dashboards(workspace_id)
+        dashboard_names = [dashboard.display_name for dashboard in list_dashboards]
+        self.assertGreater(len(list_dashboards), 0)
+        self.assertIn("dashboard1", dashboard_names)
+
+        # list_datamarts = fc.list_datamarts(workspace_id)
+        # datamart_names = [datamart.display_name for datamart in list_datamarts]
+        # self.assertGreater(len(list_datamarts), 0)
+        # self.assertIn("datamart1", datamart_names)
+
+        list_sql_endpoints = fc.list_sql_endpoints(workspace_id)
+        sqlendpoint_names = [sqlendpoint.display_name for sqlendpoint in list_sql_endpoints]
+        self.assertGreater(len(list_sql_endpoints), 0)
+        self.assertIn("lakelhousewlabels", sqlendpoint_names)
+
+        # list_mirrored_warehouses = fc.list_mirrored_warehouses(self.workspace_id)
+        # self.assertGreater(len(list_mirrored_warehouses), 0)
+
+        # list_paginated_reports = fc.list_paginated_reports(self.workspace_id)
+        # self.assertGreater(len(list_paginated_reports), 0)

msfabricpysdkcore/tests/test_reflex.py

@@ -0,0 +1,57 @@
+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):
+
+class TestFabricClientCore(unittest.TestCase):
+
+    def __init__(self, *args, **kwargs):
+        super(TestFabricClientCore, self).__init__(*args, **kwargs)
+        #load_dotenv()
+        self.fc = FabricClientCore()
+
+   
+    def test_reflex(self):
+
+        fc = self.fc
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
+
+        datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
+
+        reflex_name = "reflex" + datetime_str
+
+        reflex = fc.create_reflex(workspace_id, display_name=reflex_name)
+        self.assertEqual(reflex.display_name, reflex_name)
+
+        reflexes = fc.list_reflexes(workspace_id)
+        reflex_names = [ref.display_name for ref in reflexes]
+        self.assertGreater(len(reflexes), 0)
+        self.assertIn(reflex_name, reflex_names)
+
+
+        ref = fc.get_reflex(workspace_id, reflex_name=reflex_name)
+        self.assertIsNotNone(ref.id)
+        self.assertEqual(ref.display_name, reflex_name)
+
+        ref2 = fc.update_reflex(workspace_id, ref.id, display_name=f"{reflex_name}2", return_item=True)
+
+        ref = fc.get_reflex(workspace_id, reflex_id=ref.id)
+        self.assertEqual(ref.display_name, f"{reflex_name}2")
+        self.assertEqual(ref.id, ref2.id)
+
+        response = fc.update_reflex_definition(workspace_id, reflex_id=ref.id, definition=ref.definition)
+        self.assertIn(response.status_code, [200, 202])
+
+        definition = fc.get_reflex_definition(workspace_id, reflex_id=ref.id)
+        self.assertIn("definition", definition)
+        self.assertIn("parts", definition["definition"])
+        self.assertGreaterEqual(len(definition["definition"]["parts"]), 2)
+
+        status_code = fc.delete_reflex(workspace_id, ref.id)
+        self.assertEqual(status_code, 200)
+

msfabricpysdkcore/tests/test_reports.py

@@ -0,0 +1,56 @@
+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()
+
+        datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
+        self.item_name = "testitem" + datetime_str
+        self.item_type = "Notebook"
+
+
+    def test_reports(self):
+                
+        fc = self.fc
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
+
+        report_name = "report1234"
+
+        report_w_content = fc.get_report(workspace_id, report_name="HelloWorldReport")
+
+        definition = fc.get_report_definition(workspace_id, report_w_content.id)
+        
+        self.assertIsNotNone(definition)
+        self.assertIn("definition", definition)
+        definition = definition["definition"]
+
+        report = fc.create_report(workspace_id, display_name=report_name, definition=definition)
+        fc.update_report_definition(workspace_id, report.id, definition=definition)
+        report = fc.get_report(workspace_id, report_id=report.id)
+        self.assertEqual(report.display_name, report_name)
+        self.assertIsNotNone(report.definition)
+        
+        reports = fc.list_reports(workspace_id)
+        report_names = [r.display_name for r in reports]
+        self.assertGreater(len(reports), 0)
+        self.assertIn(report_name, report_names)
+
+        r = fc.get_report(workspace_id, report_name=report_name)
+        self.assertIsNotNone(r.id)
+        self.assertEqual(r.display_name, report_name)
+
+        r2 = fc.update_report(workspace_id, r.id, display_name=f"{report_name}2", return_item=True)
+        self.assertEqual(r2.display_name, f"{report_name}2")
+        self.assertEqual(r2.id, r.id)
+
+        status_code = fc.delete_report(workspace_id, r.id)
+        self.assertEqual(status_code, 200)

msfabricpysdkcore/tests/test_semantic_model.py

@@ -0,0 +1,56 @@
+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()
+
+        datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
+        self.item_name = "testitem" + datetime_str
+        self.item_type = "Notebook"
+
+    def test_semantic_models(self):
+                    
+        fc = self.fc
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
+        datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
+        semantic_model_name = "semanticmodel" + datetime_str
+
+        semantic_model_w_content = fc.get_semantic_model(workspace_id, semantic_model_name="Table")
+
+        definition = fc.get_semantic_model_definition(workspace_id, semantic_model_w_content.id)
+
+        self.assertIsNotNone(definition)
+        self.assertIn("definition", definition)
+        definition = definition["definition"]
+        semantic_model = fc.create_semantic_model(workspace_id, display_name=semantic_model_name, definition=definition)
+        fc.update_semantic_model_definition(workspace_id, semantic_model.id, definition=definition)
+        semantic_model = fc.get_semantic_model(workspace_id, semantic_model_id=semantic_model.id)
+        self.assertEqual(semantic_model.display_name, semantic_model_name)
+        self.assertIsNotNone(semantic_model.definition)
+        
+        semantic_models = fc.list_semantic_models(workspace_id)
+        semantic_model_names = [sm.display_name for sm in semantic_models]
+        self.assertGreater(len(semantic_models), 0)
+        self.assertIn(semantic_model_name, semantic_model_names)
+
+        sm = fc.get_semantic_model(workspace_id, semantic_model_name=semantic_model_name)
+        self.assertIsNotNone(sm.id)
+        self.assertEqual(sm.display_name, semantic_model_name)
+
+        sm2 = fc.update_semantic_model(workspace_id, sm.id, display_name=f"u{semantic_model_name}", return_item=True)
+
+        sm2 = fc.get_semantic_model(workspace_id, semantic_model_id=sm2.id)
+
+        self.assertEqual(sm2.display_name, f"u{semantic_model_name}")
+
+        status_code = fc.delete_semantic_model(workspace_id, sm.id)
+        self.assertEqual(status_code, 200)

msfabricpysdkcore/tests/test_shortcuts.py

@@ -0,0 +1,60 @@
+import unittest
+from msfabricpysdkcore.coreapi import FabricClientCore
+from datetime import datetime
+from dotenv import load_dotenv
+
+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 = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
+
+        self.lakehouse_target = "148ef579-4a5d-4048-8a48-0a703c5e3a1a"
+        self.lakehouse_shortcut = "9d1e8b07-2fea-4fc5-872c-fb586d085149"
+
+        datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
+        self.shortcutname = "shortcut" + datetime_str
+        self.path_target = "Files/to_share"
+        self.path_shortcut = "Files/shortcutfolder"
+
+        self.target = {'oneLake': {'itemId': self.lakehouse_target,
+                                   'path': self.path_target,        
+                                   'workspaceId': self.workspace_id}}
+
+    def test_shortcut_end_to_end(self):
+
+        item = self.fc.create_shortcut(workspace_id=self.workspace_id,
+                                       item_id=self.lakehouse_shortcut,
+                                       path=self.path_shortcut,
+                                       name=self.shortcutname,
+                                       target=self.target)
+        self.assertEqual(item.name, self.shortcutname)
+        self.assertEqual(item.path, self.path_shortcut)
+        self.assertIn("oneLake", item.target)
+
+        item = self.fc.get_shortcut(workspace_id=self.workspace_id,
+                                    item_id=self.lakehouse_shortcut,
+                                    path=self.path_shortcut,
+                                    name=self.shortcutname)
+        self.assertEqual(item.name, self.shortcutname)
+        self.assertEqual(item.path, self.path_shortcut)
+        self.assertIn("oneLake", item.target)
+
+        shortcuts = self.fc.list_shortcuts(workspace_id=self.workspace_id,
+                                          item_id=self.lakehouse_shortcut, parent_path="Files")
+        self.assertGreater(len(shortcuts), 0)
+
+
+        status_code = self.fc.delete_shortcut(workspace_id=self.workspace_id,
+                                             item_id=self.lakehouse_shortcut,
+                                             path=self.path_shortcut,
+                                             name=self.shortcutname)
+        
+        self.assertAlmostEqual(status_code, 200)
+
+if __name__ == "__main__":
+    unittest.main()

msfabricpysdkcore/tests/test_spark.py

@@ -0,0 +1,91 @@
+import unittest
+from msfabricpysdkcore.coreapi import FabricClientCore
+from dotenv import load_dotenv
+from datetime import datetime
+
+load_dotenv()
+
+
+class TestFabricClientCore(unittest.TestCase):
+
+    def __init__(self, *args, **kwargs):
+        super(TestFabricClientCore, self).__init__(*args, **kwargs)
+        #load_dotenv()
+        self.fc = FabricClientCore()
+
+
+    def test_spark_workspace_custom_pools(self):
+        fc = self.fc
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
+        pool_name = "pool" + datetime.now().strftime("%Y%m%d%H%M%S")
+        # List 
+
+        pools = fc.list_workspace_custom_pools(workspace_id=workspace_id)
+        self.assertGreater(len(pools), 0)
+
+        pools = fc.list_workspace_custom_pools(workspace_id=workspace_id)
+
+        self.assertIn("pool1", [p.name for p in pools])
+        pool1 = [p for p in pools if p.name == "pool1"][0]
+
+        # Get
+
+        pool1_clone = fc.get_workspace_custom_pool(workspace_id=workspace_id, pool_id=pool1.id)
+        self.assertEqual(pool1_clone.id, pool1.id)
+        # Create
+
+        pool2 = fc.create_workspace_custom_pool(workspace_id=workspace_id, 
+                                        name=pool_name, 
+                                        node_family="MemoryOptimized",
+                                        node_size="Small", 
+                                        auto_scale = {"enabled": True, "minNodeCount": 1, "maxNodeCount": 2},
+                                        dynamic_executor_allocation = {"enabled": True, "minExecutors": 1, "maxExecutors": 1})
+
+        self.assertEqual(pool2.name, pool_name)
+        self.assertEqual(pool2.node_family, "MemoryOptimized")
+
+        # Update
+
+        pool2 = fc.update_workspace_custom_pool(workspace_id=workspace_id, pool_id=pool2.id,
+                                        auto_scale = {"enabled": True, "minNodeCount": 1, "maxNodeCount": 7},
+                                        return_item=True)
+
+        self.assertEqual(pool2.auto_scale["maxNodeCount"], 7)
+        pool2_clone = fc.get_workspace_custom_pool(workspace_id=workspace_id, pool_id=pool2.id)
+        self.assertEqual(pool2_clone.auto_scale["maxNodeCount"], 7)
+
+        # Delete
+        status_code = fc.delete_workspace_custom_pool(workspace_id=workspace_id, pool_id=pool2.id)
+        self.assertEqual(status_code, 200)
+
+        pools = fc.list_workspace_custom_pools(workspace_id=workspace_id)
+        self.assertNotIn(pool_name, [p.name for p in pools])
+
+    def test_workspace_settings(self):
+        fc = self.fc
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
+
+        # Get
+
+        settings = fc.get_spark_settings(workspace_id)
+        self.assertIn("automaticLog", settings)
+        
+        
+        orig_setting = settings["automaticLog"]["enabled"]
+        settings["automaticLog"]["enabled"] = not settings["automaticLog"]["enabled"]
+
+        # Update
+        settings = fc.update_spark_settings(workspace_id, automatic_log=settings["automaticLog"])
+        new_setting = settings["automaticLog"]["enabled"]
+        self.assertNotEqual(orig_setting, new_setting)
+        self.assertTrue(orig_setting or new_setting)
+        self.assertFalse(orig_setting and new_setting)
+
+        settings = fc.get_spark_settings(workspace_id)
+        checked_setting = settings["automaticLog"]["enabled"]
+        self.assertEqual(checked_setting, new_setting)
+
+
+if __name__ == "__main__":
+    unittest.main()
+

msfabricpysdkcore/tests/test_sparkjobdefinition.py

@@ -0,0 +1,55 @@
+import unittest
+from datetime import datetime
+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_spark_job_definitions(self):
+        
+        fc = self.fc
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
+
+        datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
+        spark_job_definition_name = f"sjd{datetime_str}"
+
+        spark_job_definition_w_content = fc.get_spark_job_definition(workspace_id, spark_job_definition_name="helloworld")
+
+        result = fc.run_on_demand_spark_job_definition(workspace_id=workspace_id, 
+                                                        spark_job_definition_id=spark_job_definition_w_content.id,
+                                                        job_type="sparkjob")
+        self.assertIsNotNone(result)
+        self.assertEqual(result.job_type, "sparkjob")
+        definition = fc.get_spark_job_definition_definition(workspace_id, spark_job_definition_w_content.id)
+        
+        self.assertIsNotNone(definition)
+        self.assertIn("definition", definition)
+        definition = definition["definition"]
+
+        spark_job_definition = fc.create_spark_job_definition(workspace_id, display_name=spark_job_definition_name)
+        fc.update_spark_job_definition_definition(workspace_id, spark_job_definition.id, definition=definition)
+        spark_job_definition = fc.get_spark_job_definition(workspace_id, spark_job_definition_id=spark_job_definition.id)
+        self.assertEqual(spark_job_definition.display_name, spark_job_definition_name)
+        self.assertIsNotNone(spark_job_definition.definition)
+
+        spark_job_definitions = fc.list_spark_job_definitions(workspace_id)
+        spark_job_definition_names = [sjd.display_name for sjd in spark_job_definitions]
+        self.assertGreater(len(spark_job_definitions), 0)
+        self.assertIn(spark_job_definition_name, spark_job_definition_names)
+
+        sjd = fc.get_spark_job_definition(workspace_id, spark_job_definition_name=spark_job_definition_name)
+        self.assertIsNotNone(sjd.id)
+        self.assertEqual(sjd.display_name, spark_job_definition_name)
+
+        sjd = fc.update_spark_job_definition(workspace_id, sjd.id, display_name=f"{spark_job_definition_name}2", return_item=True)
+        sjd = fc.get_spark_job_definition(workspace_id, spark_job_definition_id=sjd.id)
+        self.assertEqual(sjd.display_name, f"{spark_job_definition_name}2")
+
+        status_code = fc.delete_spark_job_definition(workspace_id, sjd.id)
+        self.assertEqual(status_code, 200)

msfabricpysdkcore/tests/test_sqldatabases.py

@@ -0,0 +1,45 @@
+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):
+
+class TestFabricClientCore(unittest.TestCase):
+
+    def __init__(self, *args, **kwargs):
+        super(TestFabricClientCore, self).__init__(*args, **kwargs)
+        #load_dotenv()
+        self.fc = FabricClientCore()
+
+   
+    def test_sql_databases(self):
+
+        fc = self.fc
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
+
+        # datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
+
+        # sql_db = fc.create_sql_database(workspace_id, display_name="sqldb" + datetime_str)
+        # self.assertEqual(sql_db.display_name, "sqldb" + datetime_str)
+
+        # sql_dbs = fc.list_sql_databases(workspace_id)
+        # sql_db_names = [db.display_name for db in sql_dbs]
+        # self.assertGreater(len(sql_dbs), 0)
+        # self.assertIn("sqldb" + datetime_str, sql_db_names)
+
+        # db = fc.get_sql_database(workspace_id, sql_database_name="sqldb" + datetime_str)
+        # self.assertIsNotNone(db.id)
+        # self.assertEqual(db.display_name, "sqldb" + datetime_str)
+
+        # db2 = fc.update_sql_database(workspace_id, db.id, display_name=f"sqldb{datetime_str}2", return_item=True)
+
+        # db = fc.get_sql_database(workspace_id, sql_database_id=db.id)
+        # self.assertEqual(db.display_name, f"sqldb{datetime_str}2")
+
+        # status_code = fc.delete_sql_database(workspace_id, db.id)
+        # self.assertEqual(status_code, 200)
+        

msfabricpysdkcore/tests/test_warehouses.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()
+
+        datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
+        self.item_name = "testitem" + datetime_str
+        self.item_type = "Notebook"
+
+
+    def test_warehouses(self):
+
+        fc = self.fc
+        workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
+
+        datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
+        warehouse1 = f"wh{datetime_str}"
+        warehouse = fc.create_warehouse(workspace_id, display_name=warehouse1)
+        self.assertIsNotNone(warehouse.id)
+
+        warehouses = fc.list_warehouses(workspace_id)
+        warehouse_names = [wh.display_name for wh in warehouses]
+        self.assertGreater(len(warehouses), 0)
+        self.assertIn(warehouse1, warehouse_names)
+
+        warehouse = fc.get_warehouse(workspace_id, warehouse_name=warehouse1)
+        self.assertIsNotNone(warehouse.id)
+        self.assertEqual(warehouse.display_name, warehouse1)
+
+        warehouse2 = fc.update_warehouse(workspace_id, warehouse.id, display_name=f"{warehouse1}2", return_item=True)
+        warehouse = fc.get_warehouse(workspace_id, warehouse_id=warehouse.id)
+        self.assertEqual(warehouse.display_name, f"{warehouse1}2")
+        self.assertEqual(warehouse.id, warehouse2.id)
+
+        status_code = fc.delete_warehouse(workspace_id, warehouse.id)
+        self.assertEqual(status_code, 200)
+
+
+if __name__ == "__main__":
+    unittest.main()