File: test_servicemetadata.py

package info (click to toggle)
python-azure 20250603%2Bgit-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, trixie
  • size: 851,724 kB
  • sloc: python: 7,362,925; ansic: 804; javascript: 287; makefile: 195; sh: 145; xml: 109
file content (141 lines) | stat: -rw-r--r-- 5,011 bytes parent folder | download | duplicates (2)
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
import os
import json
import tempfile
import unittest
from pathlib import Path

import pytest
from packaging_tools.generate_utils import update_servicemetadata


"""
Create metadata file

Update metadata file

Update MANIFEST.IN

No need to update MANIFEST.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, "_metadata.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

            # 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