From a056ba90426b7d166642d921aa0479e06be9ee4b Mon Sep 17 00:00:00 2001 From: Dmitry R Date: Sun, 1 Mar 2026 18:09:45 +0100 Subject: [PATCH 1/2] Test: fix test_generate_aibom_purl_encoding assertions and mock data --- tests/test_service.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/test_service.py b/tests/test_service.py index 0e00036..3494be0 100644 --- a/tests/test_service.py +++ b/tests/test_service.py @@ -37,7 +37,7 @@ def test_generate_aibom_basic(self, mock_extractor_cls, mock_score): def test_generate_aibom_purl_encoding(self, mock_extractor_cls, mock_score): # Setup mock_extractor = mock_extractor_cls.return_value - mock_extractor.extract_metadata.return_value = {"name": "test-model", "author": "tester"} + mock_extractor.extract_metadata.return_value = {"name": "test-model", "author": "tester", "commit": "123456"} mock_extractor.extraction_results = {} mock_score.return_value = {"total_score": 50} @@ -48,12 +48,12 @@ def test_generate_aibom_purl_encoding(self, mock_extractor_cls, mock_score): aibom = self.service.generate_aibom(model_id) # Verify PURL encoding (slash should be / now, case preserved) - # Expected: pkg:huggingface/owner/model@12345678 (truncated hash) - - # Check components section (ML model) + # Expected: pkg:huggingface/owner/model@123456 ml_cmp = aibom["components"][0] - # Hash "123456" is less than 8 chars, so it remains "123456" - # Let's use a longer hash to test truncation + + self.assertEqual(ml_cmp["version"], "123456") + self.assertIn("pkg:huggingface/owner/model@123456", ml_cmp["purl"]) + self.assertIn("pkg:huggingface/owner/model@123456", ml_cmp["bom-ref"]) @patch("src.models.service.calculate_completeness_score") @patch("src.models.service.EnhancedExtractor") From 360440a5ee5c93d22582f1aca41b18ec28c26c01 Mon Sep 17 00:00:00 2001 From: Dmitry R Date: Sun, 1 Mar 2026 18:10:32 +0100 Subject: [PATCH 2/2] Test: add gguf to requirements.txt for CI testing --- requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index bf42baa..2b75862 100644 --- a/requirements.txt +++ b/requirements.txt @@ -23,4 +23,5 @@ packageurl-python>=0.11.1 # Test dependencies pytest>=7.0.0 pytest-mock>=3.10.0 -pytest-cov>=4.0.0 \ No newline at end of file +pytest-cov>=4.0.0 +gguf>=0.6.0 \ No newline at end of file