Skip to content

Commit a96f72a

Browse files
committed
test: update tests for value error fixing
1 parent c64ad1a commit a96f72a

File tree

3 files changed

+9
-9
lines changed

3 files changed

+9
-9
lines changed

app/domain/vex_generation/processors/vex_tix_initializer.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,12 +6,11 @@
66
from pytz import UTC
77

88
from app.domain.vex_generation.generators import StatementHelpers
9+
from app.logger import logger
910
from app.templates import create_tix_template, create_vex_template
1011

1112
from .statement_generator import StatementsGenerator
1213

13-
from app.logger import logger
14-
1514

1615
class VEXTIXInitializer:
1716
def __init__(self, directory: str):

tests/unit/processors/test_sbom_processor.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,7 @@ async def test_process_new_sboms(
268268
mock_initializer = AsyncMock()
269269
mock_vex = {"statements": [{"id": "vex1"}]}
270270
mock_tix = {"statements": [{"id": "tix1"}]}
271-
mock_initializer.init_vex_tix.return_value = [(mock_vex, mock_tix)]
271+
mock_initializer.init_vex_tix.return_value = [("/tmp/repo/sbom.json", mock_vex, mock_tix)]
272272
mock_initializer_class.return_value = mock_initializer
273273

274274
with patch.object(processor, 'save_vex_tix') as mock_save:
@@ -340,8 +340,8 @@ async def test_process_new_sboms_multiple_files(self, processor):
340340
with patch('app.domain.vex_generation.processors.sbom_processor.VEXTIXInitializer') as mock_init_class:
341341
mock_init = AsyncMock()
342342
mock_init.init_vex_tix.return_value = [
343-
(mock_vex1, mock_tix1),
344-
(mock_vex2, mock_tix2)
343+
("/tmp/repo/sbom1.json", mock_vex1, mock_tix1),
344+
("/tmp/repo/sbom2.json", mock_vex2, mock_tix2)
345345
]
346346
mock_init_class.return_value = mock_init
347347

tests/unit/processors/test_vex_tix_initializer.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,8 @@ async def test_init_vex_tix_single_sbom(self, initializer, valid_sbom):
194194

195195
assert len(results) == 1
196196
assert isinstance(results[0], tuple)
197-
assert len(results[0]) == 2
197+
assert len(results[0]) == 3
198+
assert results[0][0] == f.name
198199
mock_process.assert_called_once()
199200

200201
@pytest.mark.asyncio
@@ -343,6 +344,6 @@ async def test_init_vex_tix_preserves_order(self, initializer, valid_sbom):
343344

344345
results = await initializer.init_vex_tix("test-owner", files)
345346

346-
assert results[0][0]["id"] == 1
347-
assert results[1][0]["id"] == 2
348-
assert results[2][0]["id"] == 3
347+
assert results[0][1]["id"] == 1
348+
assert results[1][1]["id"] == 2
349+
assert results[2][1]["id"] == 3

0 commit comments

Comments
 (0)