Fix tests.

mondoohq · Feb 23, 2024 · cdbaa31 · cdbaa31
1 parent 12edfec
commit cdbaa31
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/policy/scan/local_scanner_test.go b/policy/scan/local_scanner_test.go
@@ -158,6 +158,7 @@ func (s *LocalScannerSuite) TestRunIncognito_SharedQuery() {
 
 		expectedQueries := []string{
 			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/sshd-01"].CodeId,
+			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/sshd-01"].Mrn,
 		}
 		s.ElementsMatch(expectedQueries, executedQueries)
 	}
@@ -204,6 +205,8 @@ func (s *LocalScannerSuite) TestRunIncognito_ExceptionGroups() {
 		expectedQueries := []string{
 			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/ignored-query"].CodeId,
 			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/sshd-score-01"].CodeId,
+			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/ignored-query"].Mrn,
+			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/sshd-score-01"].Mrn,
 		}
 		s.ElementsMatch(expectedQueries, executedQueries)
 
@@ -270,6 +273,9 @@ func (s *LocalScannerSuite) TestRunIncognito_ExceptionGroups_RejectedReview() {
 			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/ignored-query"].CodeId,
 			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/deactivate-query"].CodeId,
 			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/sshd-score-01"].CodeId,
+			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/ignored-query"].Mrn,
+			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/deactivate-query"].Mrn,
+			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/sshd-score-01"].Mrn,
 		}
 		s.ElementsMatch(expectedQueries, executedQueries)
 
@@ -332,6 +338,8 @@ func (s *LocalScannerSuite) TestRunIncognito_QueryExceptions() {
 		expectedQueries := []string{
 			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/ignored-query"].CodeId,
 			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/sshd-score-01"].CodeId,
+			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/ignored-query"].Mrn,
+			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/sshd-score-01"].Mrn,
 		}
 		s.ElementsMatch(expectedQueries, executedQueries)
 
@@ -394,6 +402,8 @@ func (s *LocalScannerSuite) TestRunIncognito_QueryExceptions_MultipleGroups() {
 		expectedQueries := []string{
 			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/ignored-query"].CodeId,
 			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/sshd-score-01"].CodeId,
+			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/ignored-query"].Mrn,
+			bundleMap.Queries["//local.cnspec.io/run/local-execution/queries/sshd-score-01"].Mrn,
 		}
 		s.ElementsMatch(expectedQueries, executedQueries)