Add test case for interface inheritance

afxres · May 3, 2024 · dc5cbb5 · dc5cbb5
1 parent b0292d7
commit dc5cbb5
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/code/Binary.SourceGeneration.ObjectCrossTests/InheritanceTests/InterfaceShadowingTests.cs b/code/Binary.SourceGeneration.ObjectCrossTests/InheritanceTests/InterfaceShadowingTests.cs
@@ -236,4 +236,19 @@ public void GetAllFieldsAndPropertiesForNonInterfaceTypeWithInvalidTypeTest(Type
         Assert.Equal(message, alpha.Message);
         Assert.Equal(message, bravo.Message);
     }
+
+    [Fact(DisplayName = "Compare Inheritance With Cancellation Token Test")]
+    public void CompareInheritanceCancellationTest()
+    {
+        var source = new CancellationTokenSource();
+        var compilation = CompilationModule.CreateCompilationFromThisAssembly();
+        var symbolString = compilation.GetSpecialType(SpecialType.System_String);
+        var symbolObject = compilation.GetSpecialType(SpecialType.System_Object);
+        var result = Symbols.CompareInheritance(compilation, symbolObject, symbolString, source.Token);
+        Assert.Equal(1, result);
+
+        source.Cancel();
+        var exception = Assert.Throws<OperationCanceledException>(() => Symbols.CompareInheritance(compilation, symbolObject, symbolString, source.Token));
+        Assert.Equal(source.Token, exception.CancellationToken);
+    }
 }
diff --git a/code/Binary/Internal/CommonModule.cs b/code/Binary/Internal/CommonModule.cs
@@ -186,7 +186,7 @@ void Insert(PropertyInfo member)
                 ThrowHelper.ThrowAmbiguousMembers(i.Key, type);
         }
 
-        return result.ToImmutable();
+        return result.DrainToImmutable();
     }
 
     [RequiresUnreferencedCode(RequiresUnreferencedCodeMessage)]