@@ -18,8 +18,8 @@ package com.rubensousa.projectguard.plugin
1818
1919import com.google.common.truth.Truth.assertThat
2020import com.rubensousa.projectguard.plugin.internal.BaselineConfiguration
21+ import com.rubensousa.projectguard.plugin.internal.BaselineProcessor
2122import com.rubensousa.projectguard.plugin.internal.DependencySuppression
22- import com.rubensousa.projectguard.plugin.internal.YamlProcessor
2323import org.junit.Rule
2424import org.junit.Test
2525import org.junit.rules.TemporaryFolder
@@ -58,8 +58,8 @@ class TaskBaselineTest {
5858 val outputFile = plugin.generateBaseline()
5959
6060 // then
61- val yamlProcessor = YamlProcessor ()
62- val baseline = yamlProcessor .parse(outputFile, BaselineConfiguration :: class .java )
61+ val baselineProcessor = BaselineProcessor ()
62+ val baseline = baselineProcessor .parse(outputFile)
6363 assertThat(baseline.suppressions).isEqualTo(
6464 mapOf (
6565 moduleId to listOf (
@@ -81,7 +81,7 @@ class TaskBaselineTest {
8181 @Test
8282 fun `baseline generation keeps the previous suppresion reasons` () {
8383 // given
84- val yamlProcessor = YamlProcessor ()
84+ val baselineProcessor = BaselineProcessor ()
8585 val moduleId = " :domain"
8686 val secondModuleId = " :data"
8787 val fatalModuleId = " :legacy"
@@ -104,7 +104,7 @@ class TaskBaselineTest {
104104 )
105105 )
106106 )
107- yamlProcessor .write(baselineFile, currentBaseline)
107+ baselineProcessor .write(baselineFile, currentBaseline)
108108 plugin.dumpDependencies(moduleId) {
109109 addInternalDependency(moduleId, fatalModuleId)
110110 }
@@ -121,7 +121,7 @@ class TaskBaselineTest {
121121 val outputFile = plugin.generateBaseline()
122122
123123 // then
124- val baseline = yamlProcessor .parse(outputFile, BaselineConfiguration :: class .java )
124+ val baseline = baselineProcessor .parse(outputFile)
125125 assertThat(baseline.suppressions).isEqualTo(
126126 mapOf (
127127 moduleId to listOf (
0 commit comments