improve checking of compiler generated types

coverlet-coverage · Jun 30, 2018 · a1f18b4 · a1f18b4
1 parent 017b3d9
commit a1f18b4
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/coverlet.core/Instrumentation/Instrumenter.cs b/src/coverlet.core/Instrumentation/Instrumenter.cs
@@ -66,12 +66,9 @@ private void InstrumentModule()
                 using (var module = ModuleDefinition.ReadModule(stream, parameters))
                 {
                     var types = module.GetTypes();
-                    foreach (var type in types)
+                    foreach (TypeDefinition type in types)
                     {
-                        TypeDefinition actualType = type;
-                        if (type.FullName.Contains("/"))
-                            actualType = types.FirstOrDefault(t => t.FullName == type.FullName.Split('/')[0]);
-
+                        var actualType = type.DeclaringType ?? type;
                         if (!actualType.CustomAttributes.Any(IsExcludeAttribute)
                             && !InstrumentationHelper.IsTypeExcluded(_module, actualType.FullName, _filters))
                             InstrumentType(type);