@@ -26,13 +26,10 @@ import io.kotest.matchers.collections.shouldContainExactlyInAnyOrder
2626import io.kotest.matchers.shouldBe
2727import io.kotest.matchers.shouldNot
2828
29- import java.time.Instant
30-
31- import org.ossreviewtoolkit.model.AdvisorResult
3229import org.ossreviewtoolkit.model.Identifier
3330import org.ossreviewtoolkit.model.Package
3431import org.ossreviewtoolkit.model.VcsInfo
35- import org.ossreviewtoolkit.plugins.advisors.api.normalizeVulnerabilityData
32+ import org.ossreviewtoolkit.model.vulnerabilities.Vulnerability
3633import org.ossreviewtoolkit.utils.test.identifierToPackage
3734import org.ossreviewtoolkit.utils.test.readResourceValue
3835
@@ -80,13 +77,15 @@ class OsvFunTest : WordSpec({
8077 identifierToPackage(it)
8178 }
8279
83- val packageFindings = osv.retrievePackageFindings(packages).mapKeys { it.key.id }
80+ val packageFindings = osv.retrievePackageFindings(packages).entries.associate {
81+ it.key.id to it.value.vulnerabilities
82+ }
8483
85- val expectedResult = readResourceValue<Map <Identifier , AdvisorResult >>(
84+ val expectedResult = readResourceValue<Map <Identifier , List < Vulnerability > >>(
8685 "/retrieve-package-findings-expected-result.yml"
8786 )
8887
89- packageFindings.patchTimes() shouldBe expectedResult.patchTimes()
88+ packageFindings shouldBe expectedResult
9089 }
9190
9291 " return the vulnerabilities for the commit of Hadoop 3.3.1" {
@@ -95,25 +94,17 @@ class OsvFunTest : WordSpec({
9594 vcsProcessed = VcsInfo .EMPTY .copy(revision = "a3b9c37a397ad4188041dd80621bdeefc46885f2")
9695 )
9796
98- val packageFindings = osv.retrievePackageFindings(setOf(pkg)).mapKeys { it.key.id }
97+ val packageFindings = osv.retrievePackageFindings(setOf(pkg)).entries.associate {
98+ it.key.id to it.value.vulnerabilities
99+ }
99100
100- val expectedResult = readResourceValue<Map <Identifier , AdvisorResult >>(
101+ val expectedResult = readResourceValue<Map <Identifier , List < Vulnerability > >>(
101102 "/hadoop-commit-has-expected-result.yml"
102103 )
103104
104- packageFindings.patchTimes() shouldBe expectedResult.patchTimes()
105+ packageFindings shouldBe expectedResult
105106 }
106107 }
107108})
108109
109110private fun createOsv (): Osv = OsvFactory .create()
110-
111- private fun Map <Identifier , AdvisorResult >.patchTimes (): Map <Identifier , AdvisorResult > =
112- mapValues { (_, advisorResult) ->
113- advisorResult.normalizeVulnerabilityData().copy(
114- summary = advisorResult.summary.copy(
115- startTime = Instant .EPOCH ,
116- endTime = Instant .EPOCH
117- )
118- )
119- }
0 commit comments