|
| 1 | +import pytest |
| 2 | + |
1 | 3 | from mobster.cmd.generate.oci_image.buildprobe import ImageData, SBOMMetadata |
2 | 4 | from mobster.error import SBOMError |
3 | | -import pytest |
4 | 5 |
|
5 | | -def test_imagedata_ok(): |
| 6 | + |
| 7 | +def test_imagedata_ok() -> None: |
6 | 8 | pullspec = "quay.io/asdf:123" |
7 | 9 | digest = "sha256:1234567890123456789012345678901212345678901234567890123456789012" |
8 | | - imagedata = ImageData.from_dict({ |
9 | | - "pullspec": pullspec, |
10 | | - "digest": digest, |
11 | | - }) |
| 10 | + imagedata = ImageData.from_dict( |
| 11 | + { |
| 12 | + "pullspec": pullspec, |
| 13 | + "digest": digest, |
| 14 | + } |
| 15 | + ) |
12 | 16 | assert imagedata.pullspec == pullspec |
13 | 17 | assert imagedata.digest == digest |
14 | 18 |
|
15 | | -def test_imagedata_bad(): |
| 19 | + |
| 20 | +def test_imagedata_bad() -> None: |
16 | 21 | with pytest.raises(SBOMError): |
17 | 22 | ImageData.from_dict({}) |
18 | 23 |
|
19 | | -def test_sbommetadata_ok(): |
| 24 | + |
| 25 | +def test_sbommetadata_ok() -> None: |
20 | 26 | pullspec = "quay.io/asdf:123" |
21 | 27 | digest = "sha256:1234567890123456789012345678901212345678901234567890123456789012" |
22 | | - sbom_metadata = SBOMMetadata.from_dict({ |
23 | | - "image": { |
24 | | - "pullspec": pullspec, |
25 | | - "digest": digest, |
26 | | - }, |
27 | | - "base_images": [ |
28 | | - { |
| 28 | + sbom_metadata = SBOMMetadata.from_dict( |
| 29 | + { |
| 30 | + "image": { |
29 | 31 | "pullspec": pullspec, |
30 | 32 | "digest": digest, |
31 | | - } |
32 | | - ], |
33 | | - "extra_images": [ |
34 | | - { |
35 | | - "pullspec": pullspec, |
36 | | - "digest": digest, |
37 | | - } |
38 | | - ] |
39 | | - }) |
| 33 | + }, |
| 34 | + "base_images": [ |
| 35 | + { |
| 36 | + "pullspec": pullspec, |
| 37 | + "digest": digest, |
| 38 | + } |
| 39 | + ], |
| 40 | + "extra_images": [ |
| 41 | + { |
| 42 | + "pullspec": pullspec, |
| 43 | + "digest": digest, |
| 44 | + } |
| 45 | + ], |
| 46 | + } |
| 47 | + ) |
40 | 48 | assert sbom_metadata.image.pullspec == pullspec |
41 | 49 | assert sbom_metadata.image.digest == digest |
42 | 50 | assert sbom_metadata.base_images[0].pullspec == pullspec |
43 | 51 | assert sbom_metadata.base_images[0].digest == digest |
44 | 52 | assert sbom_metadata.extra_images[0].pullspec == pullspec |
45 | 53 | assert sbom_metadata.extra_images[0].digest == digest |
46 | 54 |
|
47 | | -def test_sbommetadata_bad(): |
| 55 | + |
| 56 | +def test_sbommetadata_bad() -> None: |
48 | 57 | with pytest.raises(SBOMError): |
49 | 58 | SBOMMetadata.from_dict({}) |
0 commit comments