// 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.
 
#nullable disable
 
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.Test.Utilities;
using Xunit;
 
namespace Microsoft.CodeAnalysis.Editor.CSharp.UnitTests.AddUsing;
 
[Trait(Traits.Feature, Traits.Features.CodeActionsAddImport)]
public partial class AddUsingTests
{
    [Fact]
    public async Task TestSimpleQuery()
    {
        await TestInRegularAndScriptAsync(
@"using System;
using System.Collections.Generic;
 
class Program
{
    static void Main(string[] args)
    {
        var q = [|from x in args
                select x|]}
}",
@"using System;
using System.Collections.Generic;
using System.Linq;
 
class Program
{
    static void Main(string[] args)
    {
        var q = from x in args
                select x}
}");
    }
 
    [Fact]
    public async Task TestSimpleWhere()
    {
        await TestInRegularAndScriptAsync(
@"class Test
{
    public void SimpleWhere()
    {
        int[] numbers = {
            1,
            2,
            3
        };
        var lowNums = [|from n in numbers
                      where n < 5
                      select n|];
    }
}",
@"using System.Linq;
 
class Test
{
    public void SimpleWhere()
    {
        int[] numbers = {
            1,
            2,
            3
        };
        var lowNums = from n in numbers
                      where n < 5
                      select n;
    }
}");
    }
}