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.
- msfabricpysdkcore/adminapi.py +142 -2
- msfabricpysdkcore/coreapi.py +476 -9
- msfabricpysdkcore/domain.py +4 -1
- msfabricpysdkcore/item.py +6 -4
- msfabricpysdkcore/otheritems.py +93 -0
- msfabricpysdkcore/tests/__init__.py +0 -0
- msfabricpysdkcore/tests/test_admin_apis.py +174 -0
- msfabricpysdkcore/tests/test_connection.py +111 -0
- msfabricpysdkcore/tests/test_datapipelines.py +45 -0
- msfabricpysdkcore/tests/test_deployment_pipeline.py +63 -0
- msfabricpysdkcore/tests/test_domains.py +126 -0
- msfabricpysdkcore/tests/test_environments.py +114 -0
- msfabricpysdkcore/tests/test_evenhouses.py +56 -0
- msfabricpysdkcore/tests/test_evenstreams.py +52 -0
- msfabricpysdkcore/tests/test_external_data_shares.py +51 -0
- msfabricpysdkcore/tests/test_fabric_azure_client.py +80 -0
- msfabricpysdkcore/tests/test_gateways.py +99 -0
- msfabricpysdkcore/tests/test_git.py +66 -0
- msfabricpysdkcore/tests/test_graphqlapi.py +44 -0
- msfabricpysdkcore/tests/test_items.py +97 -0
- msfabricpysdkcore/tests/test_jobs.py +96 -0
- msfabricpysdkcore/tests/test_kql_dashboards.py +63 -0
- msfabricpysdkcore/tests/test_kql_queryset.py +64 -0
- msfabricpysdkcore/tests/test_kqldatabases.py +56 -0
- msfabricpysdkcore/tests/test_lakehouse.py +93 -0
- msfabricpysdkcore/tests/test_managed_private_endpoints.py +61 -0
- msfabricpysdkcore/tests/test_mirroreddatabases.py +80 -0
- msfabricpysdkcore/tests/test_ml_experiments.py +47 -0
- msfabricpysdkcore/tests/test_ml_models.py +47 -0
- msfabricpysdkcore/tests/test_mounted_adf.py +64 -0
- msfabricpysdkcore/tests/test_notebooks.py +57 -0
- msfabricpysdkcore/tests/test_one_lake_data_access_security.py +63 -0
- msfabricpysdkcore/tests/test_other_items.py +45 -0
- msfabricpysdkcore/tests/test_reflex.py +57 -0
- msfabricpysdkcore/tests/test_reports.py +56 -0
- msfabricpysdkcore/tests/test_semantic_model.py +56 -0
- msfabricpysdkcore/tests/test_shortcuts.py +60 -0
- msfabricpysdkcore/tests/test_spark.py +91 -0
- msfabricpysdkcore/tests/test_sparkjobdefinition.py +55 -0
- msfabricpysdkcore/tests/test_sqldatabases.py +45 -0
- msfabricpysdkcore/tests/test_warehouses.py +50 -0
- msfabricpysdkcore/tests/test_workspaces_capacities.py +159 -0
- msfabricpysdkcore/workspace.py +149 -5
- {msfabricpysdkcore-0.2.1.dist-info → msfabricpysdkcore-0.2.3.dist-info}/METADATA +33 -6
- msfabricpysdkcore-0.2.3.dist-info/RECORD +65 -0
- {msfabricpysdkcore-0.2.1.dist-info → msfabricpysdkcore-0.2.3.dist-info}/WHEEL +1 -1
- msfabricpysdkcore-0.2.1.dist-info/RECORD +0 -28
- {msfabricpysdkcore-0.2.1.dist-info → msfabricpysdkcore-0.2.3.dist-info/licenses}/LICENSE +0 -0
- {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()
|