1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
import os
import json
import tempfile
import unittest
from pathlib import Path
import pytest
from packaging_tools.auto_codegen import update_servicemetadata
"""
Create metadata file
Update metadata file
Update MANIFETS.IN
No need to update MANIFETS.IN
"""
MANIFEST_TEMP = """recursive-include tests *.py *.yaml
include *.md
include azure/__init__.py
include azure/mgmt/__init__.py
"""
class TestServiceMetadata(unittest.TestCase):
def setUp(self):
self.sdk_folder = "sdk"
self.data = {
"specFolder": "../azure-rest-api-specs",
"headSha": "e295fe97eee3709668d3e5f7f8b434026b814ef9",
"headRef": "master",
"repoHttpsUrl": "https://github.com/Azure/azure-rest-api-specs",
}
self.config = {
"meta": {
"autorest_options": {
"version": "3.0.6369",
"use": "@autorest/python@5.4.3",
"python": "",
"python-mode": "update",
"sdkrel:python-sdks-folder": "./sdk/.",
"multiapi": "",
"track2": "",
},
"advanced_options": {
"create_sdk_pull_requests": True,
"sdk_generation_pull_request_base": "integration_branch",
},
"repotag": "azure-sdk-for-python-track2",
"version": "0.2.0",
},
"projects": {"./azure-rest-api-specs/specification/operationalinsights/resource-manager/readme.md": {}},
}
self.folder_name = "monitor"
self.package_name = "azure-mgmt-monitor"
self.spec_folder = "./"
self.input_readme = "azure-rest-api-specs/specification/operationalinsights/resource-manager/readme.md"
def test_metadata(self):
with tempfile.TemporaryDirectory() as temp_dir:
# Init directories
self.sdk_folder = str(Path(temp_dir, "sdk"))
self.spec_folder = str(Path(temp_dir, "spec"))
os.makedirs(self.sdk_folder)
os.makedirs(self.spec_folder)
readme_file = str(Path(self.spec_folder, self.input_readme))
self.config["projects"][readme_file] = {}
package_folder = Path(self.sdk_folder, self.folder_name, self.package_name).expanduser()
metadata_file_path = os.path.join(package_folder, "_meta.json")
manifest_file = os.path.join(package_folder, "MANIFEST.in")
os.makedirs(package_folder)
# Test MANIFEST.in does not exist
update_servicemetadata(
sdk_folder=self.sdk_folder,
data=self.data,
config=self.config,
folder_name=self.folder_name,
package_name=self.package_name,
spec_folder=self.spec_folder,
input_readme=self.input_readme,
)
assert os.path.isfile(metadata_file_path) == True
# Do not create MANIFEST.in, if it does not exist
assert os.path.isfile(manifest_file) == False
# Test update metadata
with open(metadata_file_path, "w") as f:
json.dump({"autorest": "3.0.0"}, f, indent=2)
update_servicemetadata(
sdk_folder=self.sdk_folder,
data=self.data,
config=self.config,
folder_name=self.folder_name,
package_name=self.package_name,
spec_folder=self.spec_folder,
input_readme=self.input_readme,
)
assert os.path.isfile(metadata_file_path) == True
assert os.path.isfile(manifest_file) == False
with open(metadata_file_path, "r") as f:
md = json.load(f)
assert md["autorest"] == "3.0.6369"
# Test update MANIFEST.in
with open(manifest_file, "w") as f:
f.write(MANIFEST_TEMP)
update_servicemetadata(
sdk_folder=self.sdk_folder,
data=self.data,
config=self.config,
folder_name=self.folder_name,
package_name=self.package_name,
spec_folder=self.spec_folder,
input_readme=self.input_readme,
)
assert os.path.isfile(metadata_file_path) == True
assert os.path.isfile(manifest_file) == True
with open(manifest_file, "r") as f:
meta_line = "include _meta.json\n"
line = f.readline()
assert meta_line == line
# Test update MANIFEST.in again
update_servicemetadata(
sdk_folder=self.sdk_folder,
data=self.data,
config=self.config,
folder_name=self.folder_name,
package_name=self.package_name,
spec_folder=self.spec_folder,
input_readme=self.input_readme,
)
assert os.path.isfile(metadata_file_path) == True
assert os.path.isfile(manifest_file) == True
with open(manifest_file, "r") as f:
meta_line = "include _meta.json\n"
line = f.readline()
second_line = f.readline()
assert meta_line == line
assert meta_line != second_line
|
