Ensure reported units are exactly what is expected

trufflesecurity · Oct 16, 2023 · c3138ab · c3138ab
1 parent 52d434d
commit c3138ab
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pkg/sources/filesystem/filesystem_test.go b/pkg/sources/filesystem/filesystem_test.go
@@ -143,6 +143,9 @@ func TestEnumerate(t *testing.T) {
 	for _, unit := range reporter.Units {
 		assert.Contains(t, units, unit.SourceUnitID())
 	}
+	for _, unit := range units {
+		assert.Contains(t, reporter.Units, sources.CommonSourceUnit{ID: unit})
+	}
 }
 
 func TestChunkUnit(t *testing.T) {