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

msfabricpysdkcore/tests/test_spark.py

@@ -1,6 +1,7 @@
 import unittest
 from msfabricpysdkcore.coreapi import FabricClientCore
 from dotenv import load_dotenv
+from datetime import datetime
 
 load_dotenv()
 
@@ -18,7 +19,7 @@ class TestFabricClientCore(unittest.TestCase):
     def test_spark_workspace_custom_pools(self):
         fc = self.fc
         workspace_id = self.workspace_id
-
+        pool_name = "pool" + datetime.now().strftime("%Y%m%d%H%M%S")
         # List 
 
         pools = fc.list_workspace_custom_pools(workspace_id=workspace_id)
@@ -36,19 +37,20 @@ class TestFabricClientCore(unittest.TestCase):
         # Create
 
         pool2 = fc.create_workspace_custom_pool(workspace_id=workspace_id, 
-                                        name="pool2", 
+                                        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, "pool2")
+        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})
+                                        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)
@@ -59,7 +61,7 @@ class TestFabricClientCore(unittest.TestCase):
         self.assertEqual(status_code, 200)
 
         pools = fc.list_workspace_custom_pools(workspace_id=workspace_id)
-        self.assertNotIn("pool2", [p.name for p in pools])
+        self.assertNotIn(pool_name, [p.name for p in pools])
 
     def test_workspace_settings(self):
         fc = self.fc

msfabricpysdkcore/tests/test_sparkjobdefinition.py

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

msfabricpysdkcore/tests/test_warehouses.py

@@ -0,0 +1,51 @@
+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_warehouses(self):
+
+        fc = self.fc
+        workspace_id = 'd8a5abe0-9eed-406d-ab46-343bc57ddbe5'
+
+        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()

msfabricpysdkcore/tests/test_workspaces_capacities.py

@@ -16,6 +16,9 @@ class TestFabricClientCore(unittest.TestCase):
         fc = self.fc
         datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
         display_name = "testws" + datetime_str
+
+        datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
+        workspace_name_new = "newws" + datetime_str
         ws_created = fc.create_workspace(display_name=display_name,
                                               description="test workspace", 
                                               exists_ok=False)
@@ -109,12 +112,12 @@ class TestFabricClientCore(unittest.TestCase):
 
 #    def test_update_workspace(self):
         ws_updated = fc.update_workspace(workspace_id=ws.id, 
-                                              display_name="newn912389u8293", 
+                                              display_name=workspace_name_new, 
                                               description="new description")
-        self.assertEqual(ws_updated.display_name, "newn912389u8293")
+        self.assertEqual(ws_updated.display_name, workspace_name_new)
         self.assertEqual(ws_updated.description, "new description")
         ws = fc.get_workspace_by_id(id = ws.id)
-        self.assertEqual(ws.display_name, "newn912389u8293")
+        self.assertEqual(ws.display_name, workspace_name_new)
         self.assertEqual(ws.description, "new description")
 
 #    def test_unassign_from_capacity(self):
@@ -131,7 +134,7 @@ class TestFabricClientCore(unittest.TestCase):
 
 
 #    def test_delete_workspace(self):
-        result_status = fc.delete_workspace(display_name="newn912389u8293")
+        result_status = fc.delete_workspace(display_name=workspace_name_new)
         self.assertEqual(result_status, 200)
 
     def test_list_capacities(self):