Skip to content

Commit e53ceed

Browse files
author
Saquib Saifee
committed
refactor: update the unit tests based on feedback
Signed-off-by: Saquib Saifee <[email protected]>
1 parent b5a6fe8 commit e53ceed

File tree

1 file changed

+7
-10
lines changed

1 file changed

+7
-10
lines changed

tests/test_model.py

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
import datetime
2020
from enum import Enum
2121
from unittest import TestCase
22-
from uuid import uuid4
22+
from uuid import UUID
2323

2424
from ddt import ddt, named_data
2525

@@ -43,6 +43,7 @@
4343
Property,
4444
XsUri,
4545
)
46+
from cyclonedx.model.bom_ref import BomRef
4647
from cyclonedx.model.contact import OrganizationalContact
4748
from cyclonedx.model.issue import IssueClassification, IssueType, IssueTypeSource
4849
from tests import reorder
@@ -547,17 +548,13 @@ def test_sort(self) -> None:
547548
self.assertListEqual(sorted_uris, expected_uris)
548549

549550
def test_make_bom_link_without_bom_ref(self) -> None:
550-
serial_number = uuid4()
551-
version = 2
552-
bom_link = XsUri.make_bom_link(serial_number, version)
553-
self.assertEqual(bom_link, XsUri(f'urn:cdx:{serial_number}/{version}'))
551+
bom_link = XsUri.make_bom_link(UUID('e5a93409-fd7c-4ffa-bf7f-6dc1630b1b9d'), 2)
552+
self.assertEqual(bom_link.uri, 'urn:cdx:e5a93409-fd7c-4ffa-bf7f-6dc1630b1b9d/2')
554553

555554
def test_make_bom_link_with_bom_ref(self) -> None:
556-
serial_number = uuid4()
557-
version = 2
558-
bom_ref = 'componentA'
559-
bom_link = XsUri.make_bom_link(serial_number, version, bom_ref)
560-
self.assertEqual(bom_link, XsUri(f'urn:cdx:{serial_number}/{version}#{bom_ref}'))
555+
bom_link = XsUri.make_bom_link(UUID('e5a93409-fd7c-4ffa-bf7f-6dc1630b1b9d'),
556+
2, BomRef('componentA#sub-componentB%2'))
557+
self.assertEqual(bom_link.uri, 'urn:cdx:e5a93409-fd7c-4ffa-bf7f-6dc1630b1b9d/2#componentA#sub-componentB%2')
561558

562559

563560
class TestModelProperty(TestCase):

0 commit comments

Comments
 (0)