- Notifications
You must be signed in to change notification settings - Fork 637
/
Copy pathtest_azure_utils.py
21 lines (15 loc) · 662 Bytes
/
test_azure_utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
importpytest
fromsky.clouds.utilsimportazure_utils
deftest_validate_image_id():
# Valid marketplace image ID
azure_utils.validate_image_id("publisher:offer:sku:version")
# Valid community image ID
azure_utils.validate_image_id(
"/CommunityGalleries/gallery-name/Images/image-name")
# Invalid format (neither marketplace nor community)
withpytest.raises(ValueError):
azure_utils.validate_image_id(
"CommunityGalleries/gallery-name/Images/image-name")
# Invalid marketplace image ID (too few parts)
withpytest.raises(ValueError):
azure_utils.validate_image_id("publisher:offer:sku")
