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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (49) hide show
  1. msfabricpysdkcore/adminapi.py +142 -2
  2. msfabricpysdkcore/coreapi.py +476 -9
  3. msfabricpysdkcore/domain.py +4 -1
  4. msfabricpysdkcore/item.py +6 -4
  5. msfabricpysdkcore/otheritems.py +93 -0
  6. msfabricpysdkcore/tests/__init__.py +0 -0
  7. msfabricpysdkcore/tests/test_admin_apis.py +174 -0
  8. msfabricpysdkcore/tests/test_connection.py +111 -0
  9. msfabricpysdkcore/tests/test_datapipelines.py +45 -0
  10. msfabricpysdkcore/tests/test_deployment_pipeline.py +63 -0
  11. msfabricpysdkcore/tests/test_domains.py +126 -0
  12. msfabricpysdkcore/tests/test_environments.py +114 -0
  13. msfabricpysdkcore/tests/test_evenhouses.py +56 -0
  14. msfabricpysdkcore/tests/test_evenstreams.py +52 -0
  15. msfabricpysdkcore/tests/test_external_data_shares.py +51 -0
  16. msfabricpysdkcore/tests/test_fabric_azure_client.py +80 -0
  17. msfabricpysdkcore/tests/test_gateways.py +99 -0
  18. msfabricpysdkcore/tests/test_git.py +66 -0
  19. msfabricpysdkcore/tests/test_graphqlapi.py +44 -0
  20. msfabricpysdkcore/tests/test_items.py +97 -0
  21. msfabricpysdkcore/tests/test_jobs.py +96 -0
  22. msfabricpysdkcore/tests/test_kql_dashboards.py +63 -0
  23. msfabricpysdkcore/tests/test_kql_queryset.py +64 -0
  24. msfabricpysdkcore/tests/test_kqldatabases.py +56 -0
  25. msfabricpysdkcore/tests/test_lakehouse.py +93 -0
  26. msfabricpysdkcore/tests/test_managed_private_endpoints.py +61 -0
  27. msfabricpysdkcore/tests/test_mirroreddatabases.py +80 -0
  28. msfabricpysdkcore/tests/test_ml_experiments.py +47 -0
  29. msfabricpysdkcore/tests/test_ml_models.py +47 -0
  30. msfabricpysdkcore/tests/test_mounted_adf.py +64 -0
  31. msfabricpysdkcore/tests/test_notebooks.py +57 -0
  32. msfabricpysdkcore/tests/test_one_lake_data_access_security.py +63 -0
  33. msfabricpysdkcore/tests/test_other_items.py +45 -0
  34. msfabricpysdkcore/tests/test_reflex.py +57 -0
  35. msfabricpysdkcore/tests/test_reports.py +56 -0
  36. msfabricpysdkcore/tests/test_semantic_model.py +56 -0
  37. msfabricpysdkcore/tests/test_shortcuts.py +60 -0
  38. msfabricpysdkcore/tests/test_spark.py +91 -0
  39. msfabricpysdkcore/tests/test_sparkjobdefinition.py +55 -0
  40. msfabricpysdkcore/tests/test_sqldatabases.py +45 -0
  41. msfabricpysdkcore/tests/test_warehouses.py +50 -0
  42. msfabricpysdkcore/tests/test_workspaces_capacities.py +159 -0
  43. msfabricpysdkcore/workspace.py +149 -5
  44. {msfabricpysdkcore-0.2.1.dist-info → msfabricpysdkcore-0.2.3.dist-info}/METADATA +33 -6
  45. msfabricpysdkcore-0.2.3.dist-info/RECORD +65 -0
  46. {msfabricpysdkcore-0.2.1.dist-info → msfabricpysdkcore-0.2.3.dist-info}/WHEEL +1 -1
  47. msfabricpysdkcore-0.2.1.dist-info/RECORD +0 -28
  48. {msfabricpysdkcore-0.2.1.dist-info → msfabricpysdkcore-0.2.3.dist-info/licenses}/LICENSE +0 -0
  49. {msfabricpysdkcore-0.2.1.dist-info → msfabricpysdkcore-0.2.3.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,114 @@
1
+ import unittest
2
+ from datetime import datetime
3
+ from dotenv import load_dotenv
4
+ from time import sleep
5
+ from msfabricpysdkcore.coreapi import FabricClientCore
6
+
7
+ load_dotenv()
8
+
9
+ class TestFabricClientCore(unittest.TestCase):
10
+
11
+ def __init__(self, *args, **kwargs):
12
+ super(TestFabricClientCore, self).__init__(*args, **kwargs)
13
+ self.fc = FabricClientCore()
14
+
15
+ def test_environments_crudl(self):
16
+ fc = self.fc
17
+ workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
18
+ datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
19
+
20
+ env_name = "env" + datetime_str
21
+ environment1 = fc.create_environment(workspace_id, display_name=env_name)
22
+ self.assertEqual(environment1.display_name, env_name)
23
+
24
+ environments = fc.list_environments(workspace_id)
25
+ environment_names = [env.display_name for env in environments]
26
+ self.assertGreater(len(environments), 0)
27
+ self.assertIn(env_name, environment_names)
28
+
29
+ env = fc.get_environment(workspace_id, environment_name=env_name)
30
+ self.assertIsNotNone(env.id)
31
+ self.assertEqual(env.display_name, env_name)
32
+ new_name = env_name + "2"
33
+ env2 = fc.update_environment(workspace_id, env.id, display_name=new_name, return_item=True)
34
+
35
+ env = fc.get_environment(workspace_id, environment_id=env.id)
36
+ self.assertEqual(env.display_name, new_name)
37
+ self.assertEqual(env.id, env2.id)
38
+
39
+ status_code = fc.delete_environment(workspace_id, env.id)
40
+ self.assertEqual(status_code, 200)
41
+
42
+ def test_environment_details(self):
43
+ fc = FabricClientCore()
44
+ workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
45
+ environment_id = '5648be67-28fa-48b6-9d1f-3c87c3704d3c'
46
+ published_settings = fc.get_published_settings(workspace_id=workspace_id, environment_id=environment_id)
47
+ self.assertIsNotNone(published_settings)
48
+ self.assertIn("instancePool", published_settings)
49
+ self.assertIn("dynamicExecutorAllocation", published_settings)
50
+ staging_settings = fc.get_staging_settings(workspace_id=workspace_id, environment_id=environment_id)
51
+ self.assertIsNotNone(staging_settings)
52
+ self.assertIn("instancePool", staging_settings)
53
+ self.assertIn("dynamicExecutorAllocation", staging_settings)
54
+ if staging_settings["driverCores"] == 8:
55
+ driver_cores = 4
56
+ else:
57
+ driver_cores = 8
58
+ updated_settings = fc.update_staging_settings(workspace_id=workspace_id, environment_id=environment_id, driver_cores=driver_cores)
59
+ self.assertIn("instancePool", updated_settings)
60
+ self.assertIn("dynamicExecutorAllocation", updated_settings)
61
+ self.assertEqual(updated_settings["driverCores"], driver_cores)
62
+ updated_settings = fc.get_staging_settings(workspace_id=workspace_id, environment_id=environment_id)
63
+ self.assertIn("instancePool", updated_settings)
64
+ self.assertIn("dynamicExecutorAllocation", updated_settings)
65
+ self.assertEqual(updated_settings["driverCores"], driver_cores)
66
+
67
+
68
+ def test_environment_spark_libraries(self):
69
+ fc = self.fc
70
+ workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
71
+ environment_id = '5648be67-28fa-48b6-9d1f-3c87c3704d3c'
72
+
73
+ resp = fc.get_published_libraries(workspace_id, environment_id)
74
+ self.assertIn('customLibraries', resp)
75
+ self.assertIn('wheelFiles', resp['customLibraries'])
76
+ self.assertIn('msfabricpysdkcore-0.1.1-py3-none-any.whl', resp['customLibraries']['wheelFiles'])
77
+
78
+ resp = fc.upload_staging_library(workspace_id, environment_id, 'dummy.whl')
79
+ self.assertEqual(resp.status_code, 200)
80
+
81
+ resp = fc.get_staging_libraries(workspace_id, environment_id)
82
+
83
+ self.assertIn('customLibraries', resp)
84
+ self.assertIn('wheelFiles', resp['customLibraries'])
85
+ self.assertIn('dummy.whl', resp['customLibraries']['wheelFiles'])
86
+ self.assertIn('staging.whl', resp['customLibraries']['wheelFiles'])
87
+
88
+
89
+ resp = fc.publish_environment(workspace_id, environment_id)
90
+ self.assertIn('publishDetails', resp)
91
+ self.assertIn('state', resp['publishDetails'])
92
+ self.assertEqual(resp['publishDetails']['state'].lower(), 'running')
93
+
94
+
95
+ resp = fc.cancel_publish(workspace_id, environment_id)
96
+ self.assertIn('publishDetails', resp)
97
+ self.assertIn('state', resp['publishDetails'])
98
+ self.assertEqual(resp['publishDetails']['state'].lower(), 'cancelled')
99
+
100
+ resp = fc.delete_staging_library(workspace_id, environment_id, 'dummy.whl')
101
+ self.assertEqual(resp.status_code, 200)
102
+
103
+ resp = fc.get_staging_libraries(workspace_id, environment_id)
104
+
105
+ self.assertIn('customLibraries', resp)
106
+ self.assertIn('wheelFiles', resp['customLibraries'])
107
+ self.assertNotIn('dummy.whl', resp['customLibraries']['wheelFiles'])
108
+ self.assertIn('staging.whl', resp['customLibraries']['wheelFiles'])
109
+
110
+
111
+
112
+
113
+
114
+
@@ -0,0 +1,56 @@
1
+ import unittest
2
+ from datetime import datetime
3
+ from dotenv import load_dotenv
4
+ from time import sleep
5
+ from msfabricpysdkcore.coreapi import FabricClientCore
6
+
7
+ load_dotenv()
8
+
9
+ class TestFabricClientCore(unittest.TestCase):
10
+
11
+ def __init__(self, *args, **kwargs):
12
+ super(TestFabricClientCore, self).__init__(*args, **kwargs)
13
+ #load_dotenv()
14
+ self.fc = FabricClientCore()
15
+
16
+ datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
17
+ self.item_name = "testitem" + datetime_str
18
+ self.item_type = "Notebook"
19
+
20
+
21
+
22
+ def test_eventhouses(self):
23
+
24
+ fc = self.fc
25
+ workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
26
+ datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
27
+ eventhouse_name = "evh" + datetime_str
28
+ eventhouse1 = fc.create_eventhouse(workspace_id, display_name=eventhouse_name)
29
+ self.assertEqual(eventhouse1.display_name, eventhouse_name)
30
+
31
+ eventhouses = fc.list_eventhouses(workspace_id)
32
+ eventhouse_names = [eh.display_name for eh in eventhouses]
33
+ self.assertGreater(len(eventhouses), 0)
34
+ self.assertIn(eventhouse_name, eventhouse_names)
35
+
36
+ eh = fc.get_eventhouse(workspace_id, eventhouse_name=eventhouse_name)
37
+ self.assertIsNotNone(eh.id)
38
+ self.assertEqual(eh.display_name, eventhouse_name)
39
+ new_display_name = eventhouse_name + "2"
40
+ eh2 = fc.update_eventhouse(workspace_id, eh.id, display_name=new_display_name, return_item=True)
41
+
42
+ definition = fc.get_eventhouse_definition(workspace_id, eventhouse_id=eh.id)
43
+
44
+ self.assertIn("definition", definition)
45
+ self.assertIn("parts", definition["definition"])
46
+ self.assertGreaterEqual(len(definition["definition"]["parts"]), 2)
47
+
48
+ response = fc.update_eventhouse_definition(workspace_id, eventhouse_id=eh.id, definition=definition["definition"])
49
+ self.assertIn(response.status_code, [200, 202])
50
+
51
+ eh = fc.get_eventhouse(workspace_id, eventhouse_id=eh.id)
52
+ self.assertEqual(eh.display_name, new_display_name)
53
+ self.assertEqual(eh.id, eh2.id)
54
+
55
+ status_code = fc.delete_eventhouse(workspace_id, eh.id)
56
+ self.assertEqual(status_code, 200)
@@ -0,0 +1,52 @@
1
+ import unittest
2
+ from datetime import datetime
3
+ from dotenv import load_dotenv
4
+ from time import sleep
5
+ from msfabricpysdkcore.coreapi import FabricClientCore
6
+
7
+ load_dotenv()
8
+
9
+ class TestFabricClientCore(unittest.TestCase):
10
+
11
+ def __init__(self, *args, **kwargs):
12
+ super(TestFabricClientCore, self).__init__(*args, **kwargs)
13
+ #load_dotenv()
14
+ self.fc = FabricClientCore()
15
+
16
+ def test_eventstreams(self):
17
+
18
+ fc = self.fc
19
+ workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
20
+
21
+ datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
22
+ es_name = "es" + datetime_str
23
+
24
+ eventstream = fc.create_eventstream(workspace_id, display_name=es_name)
25
+ self.assertEqual(eventstream.display_name, es_name)
26
+
27
+ eventstreams = fc.list_eventstreams(workspace_id)
28
+ eventstream_names = [es.display_name for es in eventstreams]
29
+ self.assertGreater(len(eventstreams), 0)
30
+ self.assertIn(es_name, eventstream_names)
31
+
32
+
33
+ es = fc.get_eventstream(workspace_id, eventstream_name=es_name)
34
+ self.assertIsNotNone(es.id)
35
+ self.assertEqual(es.display_name, es_name)
36
+
37
+ es2 = fc.update_eventstream(workspace_id, es.id, display_name=f"{es_name}2", return_item=True)
38
+
39
+ es = fc.get_eventstream(workspace_id, eventstream_id=es.id)
40
+ self.assertEqual(es.display_name, f"{es_name}2")
41
+ self.assertEqual(es.id, es2.id)
42
+
43
+ response = fc.update_eventstream_definition(workspace_id, eventstream_id=es.id, definition=es.definition)
44
+ self.assertIn(response.status_code, [200, 202])
45
+
46
+ definition = fc.get_eventstream_definition(workspace_id, eventstream_id=es.id)
47
+ self.assertIn("definition", definition)
48
+ self.assertIn("parts", definition["definition"])
49
+ self.assertGreaterEqual(len(definition["definition"]["parts"]), 3)
50
+
51
+ status_code = fc.delete_eventstream(workspace_id, es.id)
52
+ self.assertEqual(status_code, 200)
@@ -0,0 +1,51 @@
1
+ import unittest
2
+ from dotenv import load_dotenv
3
+ from msfabricpysdkcore.coreapi import FabricClientCore
4
+
5
+ load_dotenv()
6
+
7
+ class TestFabricClientCore(unittest.TestCase):
8
+
9
+ def __init__(self, *args, **kwargs):
10
+ super(TestFabricClientCore, self).__init__(*args, **kwargs)
11
+ self.fc = FabricClientCore()
12
+
13
+ def test_external_data_shares(self):
14
+
15
+ fc = self.fc
16
+
17
+ workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
18
+ item_id = "148ef579-4a5d-4048-8a48-0a703c5e3a1a"
19
+
20
+ recipient = {
21
+ "userPrincipalName": "lisa4@fabrikam.com"
22
+ }
23
+ paths=["Files/to_share"]
24
+
25
+ resp = fc.create_external_data_share(workspace_id, item_id, paths, recipient)
26
+ self.assertIsNotNone(resp)
27
+ self.assertIn('id', resp)
28
+
29
+
30
+ get = fc.get_external_data_share(workspace_id, item_id, resp['id'])
31
+ self.assertIsNotNone(get)
32
+ self.assertEqual(get['id'], resp['id'])
33
+
34
+
35
+ resp = fc.list_external_data_shares_in_item(workspace_id, item_id)
36
+ self.assertGreater(len(resp), 0)
37
+
38
+ data_share_ids = [ds['id'] for ds in resp]
39
+ self.assertIn(get['id'], data_share_ids)
40
+
41
+
42
+ resp = fc.revoke_external_data_share(workspace_id, item_id, get['id'])
43
+ self.assertEqual(resp, 200)
44
+
45
+ get2 = fc.get_external_data_share(workspace_id, item_id, get['id'])
46
+ self.assertIsNotNone(get2)
47
+
48
+ self.assertEqual(get['id'], get2['id'])
49
+ self.assertEqual(get2['status'], 'Revoked')
50
+
51
+
@@ -0,0 +1,80 @@
1
+ import unittest
2
+ from dotenv import load_dotenv
3
+ from datetime import datetime
4
+ from time import sleep
5
+ from msfabricpysdkcore import FabricAzureClient
6
+
7
+ load_dotenv()
8
+
9
+
10
+ class TestFabricClientCore(unittest.TestCase):
11
+
12
+ def __init__(self, *args, **kwargs):
13
+ super(TestFabricClientCore, self).__init__(*args, **kwargs)
14
+ #load_dotenv()
15
+ self.fac = FabricAzureClient()
16
+
17
+ def test_azure_capacity(self):
18
+
19
+ fac = self.fac
20
+
21
+ subscription_id = "c77cc8fc-43bb-4d44-bdc5-6e20511ed2a8"
22
+ resource_group_name = "fabricdemo"
23
+ capacity_name = "westeuropeajrederer"
24
+ capacity_name_new = "westeuropeajrederer" + datetime.now().strftime("%Y%m%d%H%M%S")
25
+
26
+ resp = fac.check_name_availability(subscription_id, "westeurope", capacity_name_new)
27
+ self.assertIn('nameAvailable', resp)
28
+ self.assertEqual(resp['nameAvailable'], True)
29
+
30
+ resp = fac.create_or_update_capacity(subscription_id, resource_group_name, capacity_name_new,
31
+ location="westeurope",
32
+ properties_administration={"members": ['admin@MngEnvMCAP065039.onmicrosoft.com']},
33
+ sku = "F2")
34
+ self.assertIsNotNone(resp.name)
35
+ self.assertEqual(resp.name, capacity_name_new)
36
+
37
+ resp = fac.get_capacity(subscription_id, resource_group_name, capacity_name_new)
38
+ self.assertIsNotNone(resp.name)
39
+ self.assertEqual(resp.name, capacity_name_new)
40
+
41
+ sku = resp.sku['name']
42
+
43
+ sleep(60)
44
+
45
+ resp = fac.delete_capacity(subscription_id, resource_group_name, capacity_name_new)
46
+ self.assertEqual(resp.status_code, 202)
47
+
48
+
49
+ resp = fac.list_by_resource_group(subscription_id, resource_group_name)
50
+ cap_names = [cap["name"] for cap in resp]
51
+ self.assertIn(capacity_name, cap_names)
52
+
53
+
54
+
55
+ resp = fac.list_by_subscription(subscription_id)
56
+ cap_names = [cap["name"] for cap in resp]
57
+ self.assertIn(capacity_name, cap_names)
58
+
59
+
60
+ resp = fac.list_skus(subscription_id)
61
+ self.assertGreater(len(resp), 0, msg=f"No SKUs found: {resp}")
62
+
63
+
64
+ resp = fac.list_skus_for_capacity(subscription_id, resource_group_name, capacity_name)
65
+ self.assertGreater(len(resp), 0, msg=f"No SKUs found: {resp}")
66
+
67
+ resp = fac.resume_capacity(subscription_id, resource_group_name, capacity_name)
68
+ self.assertEqual(resp.status_code, 202)
69
+
70
+ sleep(60)
71
+ resp = fac.suspend_capacity(subscription_id, resource_group_name, capacity_name)
72
+ self.assertEqual(resp.status_code, 202)
73
+ sleep(60)
74
+
75
+ if sku != "F4":
76
+ resp = fac.update_capacity(subscription_id, resource_group_name, capacity_name, sku="F4")
77
+ self.assertEqual(resp.sku["name"], "F4")
78
+ else:
79
+ resp = fac.update_capacity(subscription_id, resource_group_name, capacity_name, sku="F2")
80
+ self.assertEqual(resp.sku["name"], "F2")
@@ -0,0 +1,99 @@
1
+ import unittest
2
+ from datetime import datetime
3
+ from dotenv import load_dotenv
4
+ from time import sleep
5
+ from msfabricpysdkcore.coreapi import FabricClientCore
6
+
7
+ load_dotenv()
8
+
9
+ class TestFabricClientCore(unittest.TestCase):
10
+
11
+ def __init__(self, *args, **kwargs):
12
+ super(TestFabricClientCore, self).__init__(*args, **kwargs)
13
+ #load_dotenv()
14
+ self.fc = FabricClientCore()
15
+
16
+ def test_gateways(self):
17
+
18
+ fc = self.fc
19
+
20
+ gateways = fc.list_gateways()
21
+
22
+ for gw in gateways:
23
+ if "publicKey" not in gw:
24
+ fc.delete_gateway(gw['id'])
25
+ if "publicKey" in gw:
26
+ gw_id = gw['id']
27
+
28
+ datetime_str = datetime.now().strftime("%Y-%m-%d-%H-%M-%S")
29
+ display_name = 'fabricvnet-' + datetime_str
30
+ gwr = {'displayName': display_name,
31
+ 'capacityId': '339c785f-4489-46ae-a649-e049e7610479',
32
+ 'virtualNetworkAzureResource': {'virtualNetworkName': 'fabricvnet',
33
+ 'subnetName': 'default3',
34
+ 'resourceGroupName': 'fabricdemo',
35
+ 'subscriptionId': 'c77cc8fc-43bb-4d44-bdc5-6e20511ed2a8'},
36
+ 'inactivityMinutesBeforeSleep': 30,
37
+ 'numberOfMemberGateways': 2,
38
+ 'type': 'VirtualNetwork'}
39
+
40
+ gw = fc.create_gateway(gwr)
41
+ self.assertEqual(gw['displayName'], gwr['displayName'])
42
+
43
+ gateways = fc.list_gateways()
44
+ self.assertEqual(len(gateways), 2)
45
+
46
+ gateways = [g for g in gateways if g['displayName'] == display_name]
47
+ gw = gateways[0]
48
+
49
+ ras = fc.list_gateway_role_assignments(gw['id'])
50
+ self.assertEqual(len(ras), 1)
51
+
52
+ principal = {"id" : "755f273c-98f8-408c-a886-691794938bd8",
53
+ "type" : "ServicePrincipal"}
54
+
55
+ new_ras = fc.add_gateway_role_assignment(gw['id'], principal, 'ConnectionCreator')
56
+ self.assertIn("id", new_ras)
57
+ self.assertEqual(2, len(fc.list_gateway_role_assignments(gw['id'])))
58
+
59
+ new_ras = fc.update_gateway_role_assignment(gw['id'], new_ras['id'], 'Admin')
60
+ self.assertEqual('Admin', new_ras['role'])
61
+
62
+ new_ras_ = fc.get_gateway_role_assignment(gw['id'], new_ras['id'])
63
+ self.assertEqual('Admin', new_ras_['role'])
64
+ self.assertEqual(new_ras['id'], new_ras_['id'])
65
+
66
+ resp_code = fc.delete_gateway_role_assignment(gw['id'], new_ras['id'])
67
+ self.assertEqual(200, resp_code)
68
+ self.assertEqual(1, len(fc.list_gateway_role_assignments(gw['id'])))
69
+
70
+
71
+
72
+ gw_members = fc.list_gateway_members(gw_id)
73
+ self.assertGreater(len(gw_members), 0)
74
+ self.assertIn('id', gw_members[0])
75
+ self.assertIn('displayName', gw_members[0])
76
+
77
+ display_name_member = "surface_desktop" + datetime_str
78
+
79
+ gw_member = fc.update_gateway_member(gateway_id = gw_id, gateway_member_id = gw_id, display_name=display_name_member, enabled=True)
80
+ self.assertEqual(display_name_member, gw_member['displayName'])
81
+
82
+ gw_ = fc.get_gateway(gw["id"])
83
+ self.assertEqual(display_name, gw_['displayName'])
84
+
85
+ gwr = {
86
+ "type": "OnPremises",
87
+ "displayName": display_name,
88
+ "loadBalancingSetting": "Failover",
89
+ "allowCloudConnectionRefresh": False,
90
+ "allowCustomConnectors": False
91
+ }
92
+
93
+ gw_ = fc.update_gateway(gw_id, gwr)
94
+ self.assertEqual(display_name, gw_['displayName'])
95
+
96
+ resp_code = fc.delete_gateway(gw["id"])
97
+ self.assertEqual(200, resp_code)
98
+
99
+ self.assertEqual(len(fc.list_gateways()), 1)
@@ -0,0 +1,66 @@
1
+ import unittest
2
+ from msfabricpysdkcore.coreapi import FabricClientCore
3
+ from datetime import datetime
4
+ from dotenv import load_dotenv
5
+
6
+ load_dotenv()
7
+
8
+ class TestFabricClientCore(unittest.TestCase):
9
+
10
+ def __init__(self, *args, **kwargs):
11
+ super(TestFabricClientCore, self).__init__(*args, **kwargs)
12
+ #load_dotenv()
13
+ self.fc = FabricClientCore()
14
+
15
+ def test_git(self):
16
+
17
+ datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
18
+ ws2_name = "git" + datetime_str
19
+ self.fc.create_workspace(display_name=ws2_name)
20
+ ws2 = self.fc.get_workspace_by_name(name=ws2_name)
21
+ self.fc.assign_to_capacity(workspace_id=ws2.id, capacity_id="840a6c1e-5289-4094-bbc8-716daabaeeba")
22
+
23
+ git_provider_details = {'organizationName': 'MngEnvMCAP065039',
24
+ 'projectName': 'fabricdevops',
25
+ 'gitProviderType': 'AzureDevOps',
26
+ 'repositoryName': 'fabricdevops',
27
+ 'branchName': 'main',
28
+ 'directoryName': '/sdkdemoRTI'}
29
+
30
+ status_code = self.fc.git_connect(workspace_id=ws2.id, git_provider_details=git_provider_details)
31
+
32
+ self.assertEqual(status_code, 200)
33
+
34
+ initialization_strategy = "PreferWorkspace"
35
+
36
+ status_code = self.fc.git_initialize_connection(workspace_id=ws2.id, initialization_strategy=initialization_strategy)
37
+ self.assertEqual(status_code, 200)
38
+
39
+ connection_details = self.fc.git_get_connection(workspace_id=ws2.id)
40
+ self.assertEqual(connection_details['gitConnectionState'], 'ConnectedAndInitialized')
41
+
42
+ status = self.fc.git_get_status(workspace_id=ws2.id)
43
+ self.assertTrue(len(status["changes"]) > 0)
44
+
45
+ git_credentials = self.fc.get_my_git_credentials('e624ffea-990e-482c-b27c-4ed5adae73c6')
46
+ self.assertTrue(git_credentials["source"] == "Automatic")
47
+
48
+ status_code = self.fc.update_from_git(workspace_id=ws2.id, remote_commit_hash=status["remoteCommitHash"])
49
+
50
+ self.assertEqual(status_code, 202)
51
+
52
+ blubb_lakehouse = False
53
+ for item in ws2.list_items():
54
+ if item.type == "Lakehouse" and item.display_name == "blubb":
55
+ blubb_lakehouse = True
56
+
57
+ self.assertTrue(blubb_lakehouse)
58
+
59
+ status_code = self.fc.git_disconnect(workspace_id=ws2.id)
60
+
61
+ self.assertEqual(status_code, 200)
62
+
63
+ ws2.delete()
64
+
65
+ if __name__ == "__main__":
66
+ unittest.main()
@@ -0,0 +1,44 @@
1
+ import unittest
2
+ from datetime import datetime
3
+ from dotenv import load_dotenv
4
+ from time import sleep
5
+ from msfabricpysdkcore.coreapi import FabricClientCore
6
+
7
+ load_dotenv()
8
+
9
+ # class TestFabricClientCore(unittest.TestCase):
10
+
11
+ class TestFabricClientCore(unittest.TestCase):
12
+
13
+ def __init__(self, *args, **kwargs):
14
+ super(TestFabricClientCore, self).__init__(*args, **kwargs)
15
+ #load_dotenv()
16
+ self.fc = FabricClientCore()
17
+
18
+
19
+ def test_graphql_api(self):
20
+
21
+ fc = self.fc
22
+ workspace_id = '46425c13-5736-4285-972c-6d034020f3ff'
23
+
24
+ datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
25
+
26
+ graph_ql = fc.create_graphql_api(workspace_id, display_name="graphql" + datetime_str)
27
+ self.assertEqual(graph_ql.display_name, "graphql" + datetime_str)
28
+
29
+ graph_qls = fc.list_graphql_apis(workspace_id)
30
+ graph_ql_names = [gql.display_name for gql in graph_qls]
31
+ self.assertGreater(len(graph_qls), 0)
32
+ self.assertIn("graphql" + datetime_str, graph_ql_names)
33
+
34
+ gql = fc.get_graphql_api(workspace_id, graphql_api_name="graphql" + datetime_str)
35
+ self.assertIsNotNone(gql.id)
36
+ self.assertEqual(gql.display_name, "graphql" + datetime_str)
37
+
38
+ gql2 = fc.update_graphql_api(workspace_id, gql.id, display_name=f"graphql{datetime_str}2", return_item=True)
39
+
40
+ gql = fc.get_graphql_api(workspace_id, graphql_api_id=gql.id)
41
+ self.assertEqual(gql.display_name, f"graphql{datetime_str}2")
42
+
43
+ status_code = fc.delete_graphql_api(workspace_id, gql.id)
44
+ self.assertEqual(status_code, 200)
@@ -0,0 +1,97 @@
1
+ import unittest
2
+ from datetime import datetime
3
+ from dotenv import load_dotenv
4
+ from time import sleep
5
+ from msfabricpysdkcore.coreapi import FabricClientCore
6
+
7
+ load_dotenv()
8
+
9
+ class TestFabricClientCore(unittest.TestCase):
10
+
11
+ def __init__(self, *args, **kwargs):
12
+ super(TestFabricClientCore, self).__init__(*args, **kwargs)
13
+ #load_dotenv()
14
+ self.fc = FabricClientCore()
15
+ self.workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3'
16
+
17
+
18
+ datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
19
+ self.item_name = "testitem" + datetime_str
20
+ self.item_type = "Notebook"
21
+
22
+ def test_item_end_to_end(self):
23
+
24
+ item = self.fc.create_item(display_name=self.item_name, type=self.item_type, workspace_id=self.workspace_id)
25
+ self.assertEqual(item.display_name, self.item_name)
26
+ self.assertEqual(item.type, self.item_type)
27
+ self.assertEqual(item.workspace_id, self.workspace_id)
28
+ self.assertEqual(item.description, "")
29
+
30
+ item = self.fc.get_item(workspace_id=self.workspace_id, item_id=item.id)
31
+ item_ = self.fc.get_item(workspace_id=self.workspace_id,
32
+ item_name=self.item_name, item_type=self.item_type)
33
+ self.assertEqual(item.id, item_.id)
34
+ self.assertEqual(item.display_name, self.item_name)
35
+ self.assertEqual(item.type, self.item_type)
36
+ self.assertEqual(item.workspace_id, self.workspace_id)
37
+ self.assertEqual(item.description, "")
38
+
39
+ item_list = self.fc.list_items(workspace_id=self.workspace_id)
40
+ self.assertTrue(len(item_list) > 0)
41
+
42
+ item_ids = [item_.id for item_ in item_list]
43
+ self.assertIn(item.id, item_ids)
44
+
45
+ self.fc.update_item(workspace_id=self.workspace_id, item_id=item.id, display_name=f"u{self.item_name}", return_item=True)
46
+ item = self.fc.get_item(workspace_id=self.workspace_id, item_id=item.id)
47
+ self.assertEqual(item.display_name, f"u{self.item_name}")
48
+
49
+ status_code = self.fc.delete_item(workspace_id=self.workspace_id, item_id=item.id)
50
+
51
+ self.assertAlmostEqual(status_code, 200)
52
+
53
+ def test_item_definition(self):
54
+
55
+ sjd = self.fc.get_item(workspace_id=self.workspace_id, item_name="helloworld", item_type="SparkJobDefinition")
56
+ self.assertIsNotNone(sjd.definition)
57
+ datetime_str = datetime.now().strftime("%Y%m%d%H%M%S")
58
+ blubb2 = "blubb2" + datetime_str
59
+ blubb3 = "blubb3" + datetime_str
60
+ blubb2 = self.fc.create_item(display_name=blubb2, type="SparkJobDefinition", workspace_id=self.workspace_id,
61
+ definition=sjd.definition)
62
+
63
+ blubb3 = self.fc.create_item(display_name=blubb3, type="SparkJobDefinition", workspace_id=self.workspace_id)
64
+
65
+ response = self.fc.update_item_definition(workspace_id=self.workspace_id,
66
+ item_id=blubb3.id, definition=sjd.definition)
67
+
68
+ self.assertEqual(response.status_code, 200)
69
+ blubb3 = self.fc.get_item(workspace_id=self.workspace_id, item_id=blubb3.id)
70
+
71
+ self.assertIn("parts", blubb3.definition)
72
+
73
+ self.assertEqual(len(blubb3.definition["parts"]), len(sjd.definition["parts"]))
74
+ sjd_defintion = [part["path"] for part in sjd.definition["parts"] if part["path"] == "SparkJobDefinitionV1.json"]
75
+ blubb3_defintion = [part["path"] for part in blubb3.definition["parts"] if part["path"] == "SparkJobDefinitionV1.json"]
76
+ self.assertEqual(sjd_defintion, blubb3_defintion)
77
+
78
+ self.assertNotEqual(blubb2.id, sjd.id)
79
+ self.assertIn("parts", blubb2.definition)
80
+
81
+ self.assertEqual(len(blubb2.definition["parts"]), len(sjd.definition["parts"]))
82
+ sjd_defintion = [part["path"] for part in sjd.definition["parts"] if part["path"] == "SparkJobDefinitionV1.json"]
83
+ blubb2_defintion = [part["path"] for part in blubb2.definition["parts"] if part["path"] == "SparkJobDefinitionV1.json"]
84
+ self.assertEqual(sjd_defintion, blubb2_defintion)
85
+ self.assertNotEqual(blubb2.id, blubb3.id)
86
+
87
+ blubb2.delete()
88
+ blubb3.delete()
89
+
90
+ def test_item_connections(self):
91
+
92
+ fc = self.fc
93
+ connections = fc.list_item_connections(workspace_id = '63aa9e13-4912-4abe-9156-8a56e565b7a3', item_id = '1bcc8b76-8e1f-428d-a594-f91ce1b9b076')
94
+ self.assertEqual(len(connections), 0)
95
+
96
+ if __name__ == "__main__":
97
+ unittest.main()