File: src\Analyzers\CSharp\Tests\UseIsNullCheck\UseIsNullCheckForCastAndEqualityOperatorTests.cs
Web Access
Project: src\src\Features\CSharpTest\Microsoft.CodeAnalysis.CSharp.Features.UnitTests.csproj (Microsoft.CodeAnalysis.CSharp.Features.UnitTests)
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
 
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.CodeFixes;
using Microsoft.CodeAnalysis.CSharp;
using Microsoft.CodeAnalysis.CSharp.UseIsNullCheck;
using Microsoft.CodeAnalysis.Diagnostics;
using Microsoft.CodeAnalysis.Editor.CSharp.UnitTests.Diagnostics;
using Microsoft.CodeAnalysis.Test.Utilities;
using Roslyn.Test.Utilities;
using Xunit;
using Xunit.Abstractions;
 
namespace Microsoft.CodeAnalysis.Editor.CSharp.UnitTests.UseIsNullCheck;
 
[Trait(Traits.Feature, Traits.Features.CodeActionsUseIsNullCheck)]
public partial class UseIsNullCheckForCastAndEqualityOperatorTests : AbstractCSharpDiagnosticProviderBasedUserDiagnosticTest_NoEditor
{
    public UseIsNullCheckForCastAndEqualityOperatorTests(ITestOutputHelper logger)
      : base(logger)
    {
    }
 
    internal override (DiagnosticAnalyzer, CodeFixProvider) CreateDiagnosticProviderAndFixer(Workspace workspace)
        => (new CSharpUseIsNullCheckForCastAndEqualityOperatorDiagnosticAnalyzer(), new CSharpUseIsNullCheckForCastAndEqualityOperatorCodeFixProvider());
 
    [Fact]
    public async Task TestEquality()
    {
        await TestInRegularAndScriptAsync(
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ([||](object)s == null)
                        return;
                }
            }
            """,
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if (s is null)
                        return;
                }
            }
            """);
    }
 
    [Fact, WorkItem("https://github.com/dotnet/roslyn/issues/58483")]
    public async Task TestIsNullTitle()
    {
        await TestExactActionSetOfferedAsync(
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ([||](object)s == null)
                        return;
                }
            }
            """,
[CSharpAnalyzersResources.Use_is_null_check]);
    }
 
    [Fact, WorkItem("https://github.com/dotnet/roslyn/issues/58483")]
    public async Task TestIsObjectTitle()
    {
        await TestExactActionSetOfferedAsync(
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ([||](object)s != null)
                        return;
                }
            }
            """,
[CSharpAnalyzersResources.Use_is_object_check],
new TestParameters(parseOptions: CSharpParseOptions.Default.WithLanguageVersion(LanguageVersion.CSharp8)));
    }
 
    [Fact, WorkItem("https://github.com/dotnet/roslyn/issues/58483")]
    public async Task TestIsNotNullTitle()
    {
        await TestExactActionSetOfferedAsync(
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ([||](object)s != null)
                        return;
                }
            }
            """,
[CSharpAnalyzersResources.Use_is_not_null_check],
new TestParameters(parseOptions: CSharpParseOptions.Default.WithLanguageVersion(LanguageVersion.CSharp9)));
    }
 
    [Fact]
    public async Task TestEqualitySwapped()
    {
        await TestInRegularAndScriptAsync(
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ([||]null == (object)s)
                        return;
                }
            }
            """,
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if (s is null)
                        return;
                }
            }
            """);
    }
 
    [Fact]
    public async Task TestNotEquality_CSharp8()
    {
        await TestInRegularAndScriptAsync(
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ([||](object)s != null)
                        return;
                }
            }
            """,
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if (s is object)
                        return;
                }
            }
            """, parseOptions: CSharpParseOptions.Default.WithLanguageVersion(LanguageVersion.CSharp8));
    }
 
    [Fact]
    public async Task TestNotEquality_CSharp9()
    {
        await TestInRegularAndScriptAsync(
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ([||](object)s != null)
                        return;
                }
            }
            """,
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if (s is not null)
                        return;
                }
            }
            """, parseOptions: CSharpParseOptions.Default.WithLanguageVersion(LanguageVersion.CSharp9));
    }
 
    [Fact]
    public async Task TestNotEqualitySwapped_CSharp8()
    {
        await TestInRegularAndScriptAsync(
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ([||]null != (object)s)
                        return;
                }
            }
            """,
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if (s is object)
                        return;
                }
            }
            """, parseOptions: CSharpParseOptions.Default.WithLanguageVersion(LanguageVersion.CSharp8));
    }
 
    [Fact]
    public async Task TestNotEqualitySwapped_CSharp9()
    {
        await TestInRegularAndScriptAsync(
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ([||]null != (object)s)
                        return;
                }
            }
            """,
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if (s is not null)
                        return;
                }
            }
            """, parseOptions: CSharpParseOptions.Default.WithLanguageVersion(LanguageVersion.CSharp9));
    }
 
    [Fact]
    public async Task TestMissingPreCSharp7()
    {
        await TestMissingAsync(
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ([||](object)s == null)
                        return;
                }
            }
            """, parameters: new TestParameters(parseOptions: CSharpParseOptions.Default.WithLanguageVersion(LanguageVersion.CSharp6)));
    }
 
    [Fact]
    public async Task TestOnlyForObjectCast()
    {
        await TestMissingAsync(
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ([||](string)s == null)
                        return;
                }
            }
            """);
    }
 
    [Fact]
    public async Task TestFixAll1()
    {
        await TestInRegularAndScriptAsync(
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ({|FixAllInDocument:|}(object)s == null)
                        return;
 
                    if (null == (object)s)
                        return;
                }
            }
            """,
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if (s is null)
                        return;
 
                    if (s is null)
                        return;
                }
            }
            """);
    }
 
    [Fact]
    public async Task TestFixAllNested1()
    {
        await TestInRegularAndScriptAsync(
            """
            using System;
 
            class C
            {
                void M(string s2)
                {
                    if ({|FixAllInDocument:|}(object)((object)s2 == null) == null))
                        return;
                }
            }
            """,
            """
            using System;
 
            class C
            {
                void M(string s2)
                {
                    if ((s2 is null) is null))
                        return;
                }
            }
            """);
    }
 
    [Fact]
    public async Task TestTrivia1()
    {
        await TestInRegularAndScriptAsync(
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ( /*1*/ [||]( /*2*/ object /*3*/ ) /*4*/ s /*5*/ == /*6*/ null /*7*/ )
                        return;
                }
            }
            """,
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ( /*1*/ s /*5*/ is /*6*/ null /*7*/ )
                        return;
                }
            }
            """);
    }
 
    [Fact]
    public async Task TestTrivia2()
    {
        await TestInRegularAndScriptAsync(
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ( /*1*/ [||]null /*2*/ == /*3*/ ( /*4*/ object /*5*/ ) /*6*/ s /*7*/ )
                        return;
                }
            }
            """,
            """
            using System;
 
            class C
            {
                void M(string s)
                {
                    if ( /*1*/ s /*2*/ is /*3*/ null /*7*/ )
                        return;
                }
            }
            """);
    }
 
    [Fact]
    public async Task TestConstrainedTypeParameter()
    {
        await TestInRegularAndScriptAsync(
            """
            using System;
 
            class C
            {
                void M<T>(T s) where T : class
                {
                    if ([||](object)s == null)
                        return;
                }
            }
            """,
            """
            using System;
 
            class C
            {
                void M<T>(T s) where T : class
                {
                    if (s is null)
                        return;
                }
            }
            """);
    }
 
    [Fact]
    public async Task TestUnconstrainedTypeParameter()
    {
        await TestMissingAsync(
            """
            using System;
 
            class C
            {
                void M<T>(T s)
                {
                    if ([||](object)s == null)
                        return;
                }
            }
            """);
    }
 
    [Fact]
    public async Task TestComplexExpr()
    {
        await TestInRegularAndScriptAsync(
            """
            using System;
 
            class C
            {
                void M(string[] s)
                {
                    if ([||](object)s[0] == null)
                        return;
                }
            }
            """,
            """
            using System;
 
            class C
            {
                void M(string[] s)
                {
                    if (s[0] is null)
                        return;
                }
            }
            """);
    }
 
    [Fact]
    public async Task TestNotOnDefault()
    {
        await TestMissingAsync(
            """
            using System;
 
            class C
            {
                void M(string[] s)
                {
                    if ([||](object)s[0] == default)
                        return;
                }
            }
            """);
    }
}