microsoft/qdk

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

use super::{Compilation, CompletionItem, get_completions};
use crate::{
    Encoding,
    protocol::CompletionList,
    test_utils::{
        compile_notebook_with_markers, compile_project_with_markers,
        compile_with_dependency_with_markers, compile_with_markers,
    },
};
use expect_test::{Expect, expect};
use indoc::indoc;
use qsc::line_column::Position;

mod class_completions;
mod openqasm;

fn check(
    compilation: &Compilation,
    cursor_uri: &str,
    cursor_position: Position,
    completions_to_check: &[&str],
    expect: &Expect,
) {
    let actual_completions =
        get_completions(compilation, cursor_uri, cursor_position, Encoding::Utf8);

    let mut checked_completions: Vec<(String, Option<&CompletionItem>)> = completions_to_check
        .iter()
        .map(|comp| {
            (
                (*comp).to_string(),
                actual_completions
                    .items
                    .iter()
                    .find(|item| item.label == **comp),
            )
        })
        .collect();

    // Sort by actual items' sort text
    checked_completions.sort_by_key(|c| {
        c.1.map_or(String::new(), |c| c.sort_text.clone().unwrap_or_default())
    });

    expect.assert_debug_eq(&ActualCompletions {
        completions: checked_completions,
    });

    assert_no_duplicates(actual_completions);
}

fn check_single_file(source_with_cursor: &str, completions_to_check: &[&str], expect: &Expect) {
    let (compilation, cursor_position, _) = compile_with_markers(source_with_cursor, true);

    check(
        &compilation,
        "<source>",
        cursor_position,
        completions_to_check,
        expect,
    );
}

fn check_with_stdlib(source_with_cursor: &str, completions_to_check: &[&str], expect: &Expect) {
    let (compilation, cursor_position, _) = compile_with_markers(source_with_cursor, false);

    check(
        &compilation,
        "<source>",
        cursor_position,
        completions_to_check,
        expect,
    );
}

fn check_project(
    sources_with_markers: &[(&str, &str)],
    completions_to_check: &[&str],
    expect: &Expect,
) {
    let (compilation, cursor_uri, cursor_position, _) =
        compile_project_with_markers(sources_with_markers, true);

    check(
        &compilation,
        &cursor_uri,
        cursor_position,
        completions_to_check,
        expect,
    );
}

fn check_notebook(
    cells_with_markers: &[(&str, &str)],
    completions_to_check: &[&str],
    expect: &Expect,
) {
    let (compilation, cell_uri, cursor_position, _) =
        compile_notebook_with_markers(cells_with_markers);

    check(
        &compilation,
        &cell_uri,
        cursor_position,
        completions_to_check,
        expect,
    );
}

fn check_with_dependency(
    source_with_cursor: &str,
    dependency_alias: &str,
    dependency_source: &str,
    completions_to_check: &[&str],
    expect: &Expect,
) {
    let (compilation, cursor_uri, cursor_position, _) = compile_with_dependency_with_markers(
        &[("<source>", source_with_cursor)],
        dependency_alias,
        &[("<dependency_source>", dependency_source)],
    );

    check(
        &compilation,
        &cursor_uri,
        cursor_position,
        completions_to_check,
        expect,
    );
}

fn check_no_completions(source_with_cursor: &str) {
    let (compilation, cursor_position, _) = compile_with_markers(source_with_cursor, true);
    let actual_completions =
        get_completions(&compilation, "<source>", cursor_position, Encoding::Utf8);
    assert_eq!(actual_completions.items, Vec::default());
}

fn assert_no_duplicates(mut actual_completions: CompletionList) {
    actual_completions
        .items
        .sort_by_key(|item| item.label.clone());
    let mut dups: Vec<&CompletionItem> = vec![];
    let mut last: Option<&CompletionItem> = None;
    for completion in &actual_completions.items {
        if let Some(last) = last.take()
            && last.label == completion.label
        {
            dups.push(last);
            dups.push(completion);
        }
        last.replace(completion);
    }

    assert!(dups.is_empty(), "duplicate completions found: {dups:#?}");
}

struct ActualCompletions<'a> {
    completions: Vec<(String, Option<&'a CompletionItem>)>,
}

impl std::fmt::Debug for ActualCompletions<'_> {
    fn fmt(&self, output: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let (found, not_found): (Vec<_>, Vec<_>) =
            self.completions.iter().partition(|(_, c)| c.is_some());

        if !found.is_empty() {
            write!(output, "found, sorted:")?;
            for (label, c) in &found {
                let c = c.expect("expected completion item");
                write!(output, "\n  {label:?} ({:?})", c.kind)?;
                if let Some(detail) = &c.detail {
                    write!(output, "\n    detail: {detail:?}")?;
                }

                if let Some(edits) = &c.additional_text_edits {
                    write!(output, "\n    additional_text_edits:")?;
                    for edit in edits {
                        write!(
                            output,
                            "\n      [{}:{}-{}:{}] {:?}",
                            edit.range.start.line,
                            edit.range.start.column,
                            edit.range.end.line,
                            edit.range.end.column,
                            edit.new_text,
                        )?;
                    }
                }
            }
        }

        if !not_found.is_empty() && !found.is_empty() {
            write!(output, "\n\n")?; // Add blank line between sections
        }

        if !not_found.is_empty() {
            write!(output, "not found:")?;
            for (label, _) in not_found {
                write!(output, "\n  {label:?}")?;
            }
        }

        Ok(())
    }
}

#[test]
fn ignore_unstable_namespace() {
    check_single_file(
        r#"
        namespace Test {
            open ↘
        }"#,
        &["FakeStdLib", "Microsoft.Quantum.Unstable"],
        &expect![[r#"
            found, sorted:
              "FakeStdLib" (Module)

            not found:
              "Microsoft.Quantum.Unstable"
        "#]],
    );
}

#[test]
fn ignore_unstable_callable() {
    check_single_file(
        r#"
        namespace Test {
            import Microsoft.Quantum.Unstable.*;
            operation Foo() : Unit {
                ↘
            }
        }"#,
        &["Fake", "UnstableFake"],
        &expect![[r#"
            found, sorted:
              "Fake" (Function)
                detail: "operation Fake() : Unit"
                additional_text_edits:
                  [2:12-2:12] "import FakeStdLib.Fake;\n            "

            not found:
              "UnstableFake"
        "#]],
    );
}

#[test]
fn ignore_internal_callable() {
    check_single_file(
        r#"
        namespace Test {
            internal operation Foo() : Unit {}
            operation Bar() : Unit {
                ↘
            }
        }

        namespace Test {
            internal operation Baz() : Unit {}
        }"#,
        &["Fake", "Foo", "Baz", "Hidden"],
        &expect![[r#"
            found, sorted:
              "Baz" (Function)
                detail: "operation Baz() : Unit"
              "Foo" (Function)
                detail: "operation Foo() : Unit"
              "Fake" (Function)
                detail: "operation Fake() : Unit"
                additional_text_edits:
                  [2:12-2:12] "import FakeStdLib.Fake;\n            "

            not found:
              "Hidden"
        "#]],
    );
}

#[test]
fn in_block_contains_std_functions_from_open_namespace() {
    check_single_file(
        r#"
    namespace Test {
        open FakeStdLib;
        operation Foo() : Unit {
            ↘
        }
    }"#,
        &["Fake", "FakeWithParam", "FakeCtlAdj"],
        &expect![[r#"
            found, sorted:
              "Fake" (Function)
                detail: "operation Fake() : Unit"
              "FakeCtlAdj" (Function)
                detail: "operation FakeCtlAdj() : Unit is Adj + Ctl"
              "FakeWithParam" (Function)
                detail: "operation FakeWithParam(x : Int) : Unit"
        "#]],
    );
}

#[allow(clippy::too_many_lines)]
#[test]
fn in_block_contains_std_functions() {
    check_single_file(
        indoc! {r#"
    namespace Test {
        operation Foo() : Unit {
            ↘
        }
    }"#},
        &["Fake", "FakeWithParam", "FakeCtlAdj"],
        &expect![[r#"
            found, sorted:
              "Fake" (Function)
                detail: "operation Fake() : Unit"
                additional_text_edits:
                  [1:4-1:4] "import FakeStdLib.Fake;\n    "
              "FakeCtlAdj" (Function)
                detail: "operation FakeCtlAdj() : Unit is Adj + Ctl"
                additional_text_edits:
                  [1:4-1:4] "import FakeStdLib.FakeCtlAdj;\n    "
              "FakeWithParam" (Function)
                detail: "operation FakeWithParam(x : Int) : Unit"
                additional_text_edits:
                  [1:4-1:4] "import FakeStdLib.FakeWithParam;\n    "
        "#]],
    );
}

#[test]
fn in_block_contains_newtypes() {
    check_single_file(
        r#"
    namespace Test {
        newtype Custom = String;
        operation Foo() : Unit {
            let x: ↘
        }
    }"#,
        &["Custom", "Udt"],
        &expect![[r#"
            found, sorted:
              "Custom" (Interface)
                detail: "newtype Custom = String"
              "Udt" (Interface)
                detail: "struct Udt { x : Int, y : Int }"
                additional_text_edits:
                  [2:8-2:8] "import FakeStdLib.Udt;\n        "
        "#]],
    );
}

#[test]
fn types_only_in_signature() {
    check_single_file(
        r#"
    namespace Test {
        operation Foo(foo: ↘) : Unit {
        }
        operation Bar() : Unit {
        }
    }"#,
        &["Int", "String", "Bar"],
        &expect![[r#"
            found, sorted:
              "Int" (Interface)
              "String" (Interface)

            not found:
              "Bar"
        "#]],
    );
}

#[test]
fn in_block_no_auto_open() {
    check_single_file(
        indoc! {r#"
    namespace Test {
        open FakeStdLib;
        operation Foo() : Unit {
            ↘
        }
    }"#},
        &["Fake"],
        &expect![[r#"
            found, sorted:
              "Fake" (Function)
                detail: "operation Fake() : Unit"
        "#]],
    );
}

#[test]
fn in_block_with_alias() {
    check_single_file(
        indoc! {r#"
    namespace Test {
        open FakeStdLib as Alias;
        operation Foo() : Unit {
            ↘
        }
    }"#},
        &["Alias.Fake"],
        &expect![[r#"
            found, sorted:
              "Alias.Fake" (Function)
                detail: "operation Fake() : Unit"
        "#]],
    );
}

#[test]
fn members_of_aliased_namespace() {
    check_single_file(
        indoc! {r#"
    namespace Test {
        open FakeStdLib as Alias;
        operation Foo() : Unit {
            Alias.↘
        }
    }"#},
        &["Fake", "Alias.Fake", "Library", "Alias.Library", "Foo"],
        &expect![[r#"
            found, sorted:
              "Fake" (Function)
                detail: "operation Fake() : Unit"
              "Library" (Module)

            not found:
              "Alias.Fake"
              "Alias.Library"
              "Foo"
        "#]],
    );
}

#[test]
fn aliased_exact_import() {
    check_single_file(
        indoc! {r#"
    namespace Test {
        import FakeStdLib.Fake as Alias;
        operation Foo() : Unit {
            ↘
        }
    }"#},
        &["Fake", "Alias.Fake", "Alias"],
        &expect![[r#"
            found, sorted:
              "Alias" (Function)
                detail: "operation Fake() : Unit"

            not found:
              "Fake"
              "Alias.Fake"
        "#]],
    );
}

#[test]
fn open_from_dependency() {
    check_with_dependency(
        r"
        namespace Test {
            open MyDep.Dependency;
            operation Foo() : Unit {
                ↘
            }
        }",
        "MyDep",
        "namespace Dependency { operation Baz() : Unit {} export Baz; }",
        &["Baz"],
        &expect![[r#"
            found, sorted:
              "Baz" (Function)
                detail: "operation Baz() : Unit"
        "#]],
    );
}

#[test]
fn open_with_alias_from_dependency() {
    check_with_dependency(
        r"
        namespace Test {
            open MyDep.Dependency as Alias;
            open SamePackage as Alias1;
            operation Foo() : Unit {
                ↘
            }
        }
        namespace SamePackage { operation Bar() : Unit {} }",
        "MyDep",
        "namespace Dependency { operation Baz() : Unit {} export Baz; }",
        &["Alias.Baz", "Baz", "Alias1.Bar", "Bar"],
        &expect![[r#"
            found, sorted:
              "Alias1.Bar" (Function)
                detail: "operation Bar() : Unit"
              "Alias.Baz" (Function)
                detail: "operation Baz() : Unit"

            not found:
              "Baz"
              "Bar"
        "#]],
    );
}

#[test]
fn import_ns_with_alias_from_dependency() {
    check_with_dependency(
        r"
        namespace Test {
            import MyDep.Dependency as Alias;
            import SamePackage as Alias1;
            operation Foo() : Unit {
                ↘
            }
        }
        namespace SamePackage { operation Bar() : Unit {} }",
        "MyDep",
        "namespace Dependency { operation Baz() : Unit {} export Baz; }",
        &["Alias.Baz", "Baz", "Alias1.Bar", "Bar"],
        &expect![[r#"
            found, sorted:
              "Alias1.Bar" (Function)
                detail: "operation Bar() : Unit"
              "Alias.Baz" (Function)
                detail: "operation Baz() : Unit"

            not found:
              "Baz"
              "Bar"
        "#]],
    );
}

#[test]
fn exact_import_from_dependency() {
    check_with_dependency(
        r"
        namespace Test {
            import MyDep.Dependency.Baz;
            operation Foo() : Unit {
                ↘
            }
        }",
        "MyDep",
        "namespace Dependency { operation Baz() : Unit {} export Baz; }",
        &["Baz"],
        &expect![[r#"
            found, sorted:
              "Baz" (Function)
                detail: "operation Baz() : Unit"
        "#]],
    );
}

#[test]
fn in_block_from_other_namespace() {
    check_single_file(
        indoc! {r#"
    namespace Test {
        operation Bar() : Unit {
            ↘
        }
        export Bar;
    }
    namespace Other {
        operation Foo() : Unit {}
        export Foo;
    }"#},
        &["Foo"],
        &expect![[r#"
            found, sorted:
              "Foo" (Function)
                detail: "operation Foo() : Unit"
                additional_text_edits:
                  [1:4-1:4] "import Other.Foo;\n    "
        "#]],
    );
}

#[test]
fn auto_open_multiple_files() {
    check_project(
        &[
            (
                "foo.qs",
                indoc! {r#"namespace Foo { operation FooOperation() : Unit {} export FooOperation; }
                "#},
            ),
            (
                "bar.qs",
                indoc! {r#"namespace Bar { operation BarOperation() : Unit { ↘ } export BarOperation; }
                "#},
            ),
        ],
        &["FooOperation"],
        &expect![[r#"
            found, sorted:
              "FooOperation" (Function)
                detail: "operation FooOperation() : Unit"
                additional_text_edits:
                  [0:16-0:16] "import Foo.FooOperation;\n "
        "#]],
    );
}

#[test]
fn in_block_nested_op() {
    check_single_file(
        indoc! {r#"
    namespace Test {
        operation Bar() : Unit {
            operation Foo() : Unit {}
            ↘
        }
    }"#},
        &["Foo"],
        &expect![[r#"
            found, sorted:
              "Foo" (Function)
                detail: "operation Foo() : Unit"
        "#]],
    );
}

#[test]
fn in_block_hidden_nested_op() {
    check_single_file(
        indoc! {r#"
    namespace Test {
        operation Baz() : Unit {
            ↘
        }
        operation Foo() : Unit {
            operation Bar() : Unit {}
        }
    }"#},
        &["Bar"],
        &expect![[r#"
            not found:
              "Bar"
        "#]],
    );
}

#[test]
fn in_namespace_contains_open() {
    check_single_file(
        indoc! {r#"
    namespace Test {
        ↘
        operation Foo() : Unit {
        }
    }"#},
        &["open"],
        &expect![[r#"
            found, sorted:
              "open" (Keyword)
        "#]],
    );
}

#[test]
fn top_level_contains_namespace() {
    check_single_file(
        indoc! {r#"
        namespace Test {}
        ↘
        "#},
        &["namespace"],
        &expect![[r#"
            found, sorted:
              "namespace" (Keyword)
        "#]],
    );
}

#[test]
fn attributes() {
    check_single_file(
        indoc! {r#"
        namespace Test {
            @↘
        }
        "#},
        &["EntryPoint"],
        &expect![[r#"
            found, sorted:
              "EntryPoint" (Interface)
        "#]],
    );
}

#[test]
fn stdlib_udt() {
    check_single_file(
        indoc! {r#"
        namespace Test {
            operation Foo() : Unit {
                ↘
            }
        "#},
        &["TakesUdt"],
        &expect![[r#"
            found, sorted:
              "TakesUdt" (Function)
                detail: "function TakesUdt(input : Udt) : Udt"
                additional_text_edits:
                  [1:4-1:4] "import FakeStdLib.TakesUdt;\n    "
        "#]],
    );
}

#[test]
fn notebook_top_level() {
    check_notebook(
        &[(
            "cell1",
            indoc! {r#"operation Foo() : Unit {}
            ↘
        "#},
        )],
        &["operation", "namespace", "let", "Fake"],
        &expect![[r#"
            found, sorted:
              "let" (Keyword)
              "namespace" (Keyword)
              "operation" (Keyword)
              "Fake" (Function)
                detail: "operation Fake() : Unit"
                additional_text_edits:
                  [0:0-0:0] "import FakeStdLib.Fake;\n"
        "#]],
    );
}

#[test]
fn notebook_top_level_global() {
    check_notebook(
        &[(
            "cell1",
            indoc! {r#"operation Foo() : Unit {}
            ↘
        "#},
        )],
        &["Fake"],
        &expect![[r#"
            found, sorted:
              "Fake" (Function)
                detail: "operation Fake() : Unit"
                additional_text_edits:
                  [0:0-0:0] "import FakeStdLib.Fake;\n"
        "#]],
    );
}

#[test]
fn notebook_top_level_namespace_already_open_for_global() {
    check_notebook(
        &[(
            "cell1",
            indoc! {r#"
            open FakeStdLib;
            operation Foo() : Unit {}
            ↘
        "#},
        )],
        &["Fake"],
        &expect![[r#"
            found, sorted:
              "Fake" (Function)
                detail: "operation Fake() : Unit"
        "#]],
    );
}

#[test]
fn notebook_block() {
    check_notebook(
        &[(
            "cell1",
            indoc! {r#"operation Foo() : Unit {
                ↘
            }
        "#},
        )],
        &["Fake", "let"],
        &expect![[r#"
            found, sorted:
              "let" (Keyword)
              "Fake" (Function)
                detail: "operation Fake() : Unit"
                additional_text_edits:
                  [0:0-0:0] "import FakeStdLib.Fake;\n"
        "#]],
    );
}

#[test]
fn notebook_auto_open_start_of_cell_empty() {
    check_notebook(
        &[
            (
                "cell1",
                indoc! {"
                    //qsharp
                    namespace Foo { operation Bar() : Unit {} }"
                },
            ),
            (
                "cell2",
                indoc! {"
                    //qsharp
                    ↘"
                },
            ),
        ],
        &["Fake"],
        &expect![[r#"
            found, sorted:
              "Fake" (Function)
                detail: "operation Fake() : Unit"
                additional_text_edits:
                  [1:0-1:0] "import FakeStdLib.Fake;\n"
        "#]],
    );
}

#[test]
fn notebook_auto_open_start_of_cell() {
    check_notebook(
        &[
            (
                "cell1",
                indoc! {"
                    //qsharp
                    namespace Foo { operation Bar() : Unit {} }"
                },
            ),
            (
                "cell2",
                indoc! {r#"
                    //qsharp
                    Message("hi")
                    ↘"#
                },
            ),
        ],
        &["Fake"],
        &expect![[r#"
            found, sorted:
              "Fake" (Function)
                detail: "operation Fake() : Unit"
                additional_text_edits:
                  [1:0-1:0] "import FakeStdLib.Fake;\n"
        "#]],
    );
}

#[test]
fn notebook_last_expr() {
    check_notebook(
        &[(
            "cell1",
            indoc! {"
                    //qsharp
                    function Foo() : Unit {}
                    3 + ↘"
            },
        )],
        &["Foo", "Fake"],
        &expect![[r#"
            found, sorted:
              "Foo" (Function)
                detail: "function Foo() : Unit"
              "Fake" (Function)
                detail: "operation Fake() : Unit"
                additional_text_edits:
                  [1:0-1:0] "import FakeStdLib.Fake;\n"
        "#]],
    );
}

#[test]
fn local_vars() {
    check_single_file(
        r#"
    namespace Test {
        operation Foo() : Unit {
            let bar = 3;
            ↘
            let foo = 3;
        }
    }"#,
        &["foo", "bar"],
        &expect![[r#"
            found, sorted:
              "bar" (Variable)
                detail: "bar : Int"

            not found:
              "foo"
        "#]],
    );
}

#[test]
fn local_items() {
    check_single_file(
        r#"
    namespace Test {
        operation Baz() : Unit {
            operation Foo() : Unit {}
            ↘
            operation Bar() : Unit {}
            newtype Custom = String;
        }
    }"#,
        &["Foo", "Bar", "Custom"],
        &expect![[r#"
            found, sorted:
              "Bar" (Function)
                detail: "operation Bar() : Unit"
              "Custom" (Interface)
                detail: "newtype Custom = String"
              "Foo" (Function)
                detail: "operation Foo() : Unit"
        "#]],
    );
}

#[test]
fn type_params() {
    check_single_file(
        r#"
    namespace Test {
        operation Foo<'T>() : Unit {
            let x: ↘
        }
    }"#,
        &["'T", "Bar"],
        &expect![[r#"
            found, sorted:
              "'T" (TypeParameter)

            not found:
              "Bar"
        "#]],
    );
}

#[test]
fn scoped_local_vars() {
    check_single_file(
        r#"
    namespace Test {
        operation Foo() : Unit {
            {
                let foo = 3;
            }
            ↘
        }
    }"#,
        &["foo"],
        &expect![[r#"
            not found:
              "foo"
        "#]],
    );
}

#[test]
fn callable_params() {
    check_single_file(
        r#"
    namespace Test {
        newtype Custom = String;
        operation Foo(foo: Int, bar: Custom) : Unit {
            {
                ↘
            }
        }
    }"#,
        &["foo", "bar"],
        &expect![[r#"
            found, sorted:
              "bar" (Variable)
                detail: "bar : Custom"
              "foo" (Variable)
                detail: "foo : Int"
        "#]],
    );
}

#[test]
fn local_var_in_callable_parent_scope() {
    check_single_file(
        r#"
    namespace Test {
        operation Foo(foo: Int) : Unit {
            let bar = 3;
            operation Bar() : Unit {
                let baz = 3;
                ↘
            }
        }
    }"#,
        &["foo", "bar", "baz"],
        &expect![[r#"
            found, sorted:
              "baz" (Variable)
                detail: "baz : Int"

            not found:
              "foo"
              "bar"
        "#]],
    );
}

#[test]
#[ignore = "completion list ignores shadowing rules for open statements"]
fn local_var_and_open_shadowing_rules() {
    check_single_file(
        r#"
        namespace Foo {
            operation Bar() : Unit {
            }
        }

        namespace Test {
            operation Main() : Unit {
                let Bar = 3;
                Bar;
                {
                    // open Foo should shadow the local Bar declaration
                    open Foo;
                    Bar;
                    ↘
                }

            }
        }"#,
        &["Bar"],
        &expect![[r#"
            [
                Some(
                    CompletionItem {
                        label: "Bar",
                        kind: Function,
                        sort_text: Some(
                            "0700Bar",
                        ),
                        detail: Some(
                            "operation Bar() : Unit",
                        ),
                        additional_text_edits: None,
                    },
                ),
            ]
        "#]],
    );
}

// no additional text edits for Foo or Bar because FooNs is already glob imported
#[test]
fn dont_import_if_already_glob_imported() {
    check_single_file(
        r#"
        namespace FooNs {
            operation Foo() : Unit {
            }
            operation Bar() : Unit { }
        }

        namespace Test {
            import FooNs.*;
            operation Main() : Unit {
                ↘
            }
        }"#,
        &["Foo", "Bar"],
        &expect![[r#"
            found, sorted:
              "Bar" (Function)
                detail: "operation Bar() : Unit"
              "Foo" (Function)
                detail: "operation Foo() : Unit"
        "#]],
    );
}

// expect an auto-import for `Foo.Bar`, separate from the preexisting glob import `Foo.Bar.*`
#[test]
fn glob_import_item_with_same_name() {
    check_single_file(
        r#"
        namespace Foo {
            operation Bar() : Unit {
            }
        }

        namespace Foo.Bar {
        }

        namespace Baz {
            import Foo.Bar.*;
            operation Main(): Unit {
                ↘
            }
        }"#,
        &["Bar"],
        &expect![[r#"
            found, sorted:
              "Bar" (Function)
                detail: "operation Bar() : Unit"
                additional_text_edits:
                  [10:12-10:12] "import Foo.Bar;\n            "
        "#]],
    );
}

// no additional text edits for Foo because Foo is directly imported,
// but additional text edits for Bar because Bar is not directly imported
#[test]
fn dont_import_if_already_directly_imported() {
    check_single_file(
        r#"
        namespace FooNs {
            operation Foo() : Unit { }
            operation Bar() : Unit { }
        }

        namespace Test {
            import FooNs.Foo;
            operation Main() : Unit {
                ↘
            }
        }"#,
        &["Foo", "Bar"],
        &expect![[r#"
            found, sorted:
              "Foo" (Function)
                detail: "operation Foo() : Unit"
              "Bar" (Function)
                detail: "operation Bar() : Unit"
                additional_text_edits:
                  [7:12-7:12] "import FooNs.Bar;\n            "
        "#]],
    );
}

#[test]
fn auto_import_from_qir_runtime() {
    check_with_stdlib(
        r#"
        namespace Test {
            operation Main() : Unit {
               AllocateQubitA↘
            }
        }"#,
        &["AllocateQubitArray"],
        &expect![[r#"
            found, sorted:
              "AllocateQubitArray" (Function)
                detail: "operation AllocateQubitArray(size : Int) : Qubit[]"
                additional_text_edits:
                  [2:12-2:12] "import QIR.Runtime.AllocateQubitArray;\n            "
        "#]],
    );
}

#[test]
fn dont_generate_import_for_core_prelude() {
    check_with_stdlib(
        r#"
        namespace Test {
            operation Main() : Unit {
               Length↘
            }
        }"#,
        &["Length"],
        // additional text edits should be None because Length is in the core prelude
        &expect![[r#"
            found, sorted:
              "Length" (Function)
                detail: "function Length<'T>(a : 'T[]) : Int"
        "#]],
    );
}

#[test]
fn dont_generate_import_for_stdlib_prelude() {
    check_with_stdlib(
        r#"
        namespace Test {
            operation Main() : Unit {
               MResetZ↘
            }
        }"#,
        &["MResetZ"],
        // additional text edits should be None because MResetZ is in Std.Measurement, which
        // is in the prelude.
        &expect![[r#"
            found, sorted:
              "MResetZ" (Function)
                detail: "operation MResetZ(target : Qubit) : Result"
        "#]],
    );
}

#[test]
fn callable_from_same_file() {
    check_single_file(
        r#"
        namespace Test {
            function MyCallable() : Unit {}
            operation Main() : Unit {
               MyCall↘
            }
        }"#,
        &["MyCallable"],
        &expect![[r#"
            found, sorted:
              "MyCallable" (Function)
                detail: "function MyCallable() : Unit"
        "#]],
    );
}

#[test]
fn member_completion() {
    check_single_file(
        r#"
        namespace Test {
            function MyCallable() : Unit {}
        }

        namespace Main {
            operation Main() : Unit {
               Test.↘
            }
        }

        "#,
        &["MyCallable"],
        &expect![[r#"
            found, sorted:
              "MyCallable" (Function)
                detail: "function MyCallable() : Unit"
        "#]],
    );
}

#[test]
fn member_completion_in_imported_namespace() {
    check_single_file(
        r#"
        namespace Test.Foo {
            function MyCallable() : Unit {}
        }

        namespace Test.Foo.Bar {
            function CallableInBar()  : Unit {}
        }

        namespace Main {
            open Test;
            operation Main() : Unit {
               Foo.↘
            }
        }

        "#,
        &["MyCallable", "Bar"],
        &expect![[r#"
            found, sorted:
              "MyCallable" (Function)
                detail: "function MyCallable() : Unit"
              "Bar" (Module)
        "#]],
    );
}

#[test]
fn namespace_completion() {
    check_single_file(
        r#"
        namespace Test.Foo {
            function MyCallable() : Unit {}
        }

        namespace Main {
            operation Main() : Unit {
               Test.↘
            }
        }

        "#,
        &["Foo"],
        &expect![[r#"
            found, sorted:
              "Foo" (Module)
        "#]],
    );
}

#[test]
fn nested_namespace() {
    check_single_file(
        r#"
        namespace Test.Foo {
            function MyCallable() : Unit {}
        }

        namespace Test {
            function MyCallable2() : Unit {
                Foo.↘
            }
        }"#,
        &["MyCallable", "MyCallable2"],
        &expect![[r#"
            found, sorted:
              "MyCallable" (Function)
                detail: "function MyCallable() : Unit"

            not found:
              "MyCallable2"
        "#]],
    );
}

#[test]
fn std_member() {
    check_single_file(
        r#"
        namespace Test {
            function MyCallable2() : Unit {
                FakeStdLib.↘
            }
        }"#,
        &["Fake", "Library"],
        &expect![[r#"
            found, sorted:
              "Fake" (Function)
                detail: "operation Fake() : Unit"
              "Library" (Module)
        "#]],
    );
}

#[test]
fn open_namespace() {
    check_single_file(
        r#"
        namespace Test {
            open FakeStdLib.↘;
        }"#,
        &["Fake", "Library"],
        &expect![[r#"
            found, sorted:
              "Library" (Module)

            not found:
              "Fake"
        "#]],
    );
}

#[test]
fn open_namespace_no_semi() {
    check_single_file(
        r#"
        namespace Test {
            open FakeStdLib.↘
        }"#,
        &["Fake", "Library"],
        &expect![[r#"
            found, sorted:
              "Library" (Module)

            not found:
              "Fake"
        "#]],
    );
}

#[test]
fn open_namespace_no_semi_followed_by_decl() {
    check_single_file(
        r#"
        namespace Test {
            open FakeStdLib.↘
            operation Foo() : Unit {}
        }"#,
        &["Fake", "Library"],
        &expect![[r#"
            found, sorted:
              "Library" (Module)

            not found:
              "Fake"
        "#]],
    );
}

#[test]
fn open_namespace_partial_path_part() {
    check_single_file(
        r#"
        namespace Test {
            open FakeStdLib.↘F
            operation Foo() : Unit {}
        }"#,
        &["Fake", "Library"],
        &expect![[r#"
            found, sorted:
              "Library" (Module)

            not found:
              "Fake"
        "#]],
    );
}

#[test]
fn let_stmt_type() {
    check_single_file(
        r#"
        namespace Test {
            function Main() : Unit {
                let x: ↘
            }
        }"#,
        &["Udt", "Qubit", "Int", "Main", "FakeWithParam"],
        &expect![[r#"
            found, sorted:
              "Int" (Interface)
              "Qubit" (Interface)
              "Udt" (Interface)
                detail: "struct Udt { x : Int, y : Int }"
                additional_text_edits:
                  [2:12-2:12] "import FakeStdLib.Udt;\n            "

            not found:
              "Main"
              "FakeWithParam"
        "#]],
    );
}

#[test]
fn let_stmt_type_before_next_stmt() {
    check_single_file(
        r#"
        namespace Test {
            function Main() : Unit {
                use q = Qubit();
                let x: ↘
                H(q);
            }
        }"#,
        &["Udt", "Qubit", "Int", "Main", "FakeWithParam"],
        &expect![[r#"
            found, sorted:
              "Int" (Interface)
              "Qubit" (Interface)
              "Udt" (Interface)
                detail: "struct Udt { x : Int, y : Int }"
                additional_text_edits:
                  [2:12-2:12] "import FakeStdLib.Udt;\n            "

            not found:
              "Main"
              "FakeWithParam"
        "#]],
    );
}

#[test]
fn type_position_namespace() {
    check_single_file(
        r#"
        namespace Test {
            function Main() : Unit {
                let x: FakeStdLib.↘ ;
            }
        }"#,
        &["Udt", "Qubit", "Int", "Main", "FakeWithParam"],
        &expect![[r#"
            found, sorted:
              "Udt" (Interface)
                detail: "struct Udt { x : Int, y : Int }"

            not found:
              "Qubit"
              "Int"
              "Main"
              "FakeWithParam"
        "#]],
    );
}

#[test]
fn udt_base_type_part() {
    check_single_file(
        r#"
        namespace Test {
            newtype Foo = FakeStdLib.↘
        }"#,
        &["Udt", "Qubit", "FakeWithParam"],
        &expect![[r#"
            found, sorted:
              "Udt" (Interface)
                detail: "struct Udt { x : Int, y : Int }"

            not found:
              "Qubit"
              "FakeWithParam"
        "#]],
    );
}

#[test]
fn struct_init() {
    check_single_file(
        r#"
        namespace Test {
            function Main() : Unit {
                let x = new ↘ ;
            }
        }"#,
        &["Udt", "Qubit", "Int", "Main", "FakeWithParam"],
        &expect![[r#"
            found, sorted:
              "Udt" (Interface)
                detail: "struct Udt { x : Int, y : Int }"
                additional_text_edits:
                  [2:12-2:12] "import FakeStdLib.Udt;\n            "

            not found:
              "Qubit"
              "Int"
              "Main"
              "FakeWithParam"
        "#]],
    );
}

#[test]
fn struct_init_path_part() {
    check_single_file(
        r#"
        namespace Test {
            function Main() : Unit {
                let x = new FakeStdLib.↘ ;
            }
        }"#,
        &["Udt", "Qubit", "Int", "Main", "FakeWithParam"],
        &expect![[r#"
            found, sorted:
              "Udt" (Interface)
                detail: "struct Udt { x : Int, y : Int }"

            not found:
              "Qubit"
              "Int"
              "Main"
              "FakeWithParam"
        "#]],
    );
}

#[test]
fn struct_init_path_part_in_field_assigment() {
    check_single_file(
        r#"
        namespace Test {
            function Main() : Unit {
                let x = new FakeStdLib.Udt { x = FakeStdLib.↘ } ;
            }
        }"#,
        &["Udt", "Qubit", "FakeWithParam"],
        &expect![[r#"
            found, sorted:
              "FakeWithParam" (Function)
                detail: "operation FakeWithParam(x : Int) : Unit"
              "Udt" (Interface)
                detail: "struct Udt { x : Int, y : Int }"

            not found:
              "Qubit"
        "#]],
    );
}

#[test]
fn export_path() {
    check_single_file(
        r#"
        namespace Test {
            export ↘ ;
            function Main() : Unit {
            }
        }"#,
        &["Udt", "Qubit", "Int", "Main", "FakeWithParam", "FakeStdLib"],
        &expect![[r#"
            found, sorted:
              "Main" (Function)
                detail: "function Main() : Unit"
              "FakeStdLib" (Module)

            not found:
              "Udt"
              "Qubit"
              "Int"
              "FakeWithParam"
        "#]],
    );
}

#[test]
fn export_path_part() {
    check_single_file(
        r#"
        namespace Test {
            export FakeStdLib.↘ ;
            function Main() : Unit {
            }
        }"#,
        &["Udt", "Qubit", "Int", "Main", "FakeWithParam", "FakeStdLib"],
        &expect![[r#"
            found, sorted:
              "FakeWithParam" (Function)
                detail: "operation FakeWithParam(x : Int) : Unit"
              "Udt" (Interface)
                detail: "struct Udt { x : Int, y : Int }"

            not found:
              "Qubit"
              "Int"
              "Main"
              "FakeStdLib"
        "#]],
    );
}

#[test]
fn partially_typed_name() {
    check_single_file(
        r#"
        namespace Test {
            export Fo↘
            function Foo() : Unit {
            }
        }"#,
        &["Foo"],
        &expect![[r#"
            found, sorted:
              "Foo" (Function)
                detail: "function Foo() : Unit"
        "#]],
    );
}

#[test]
fn from_dependency_main() {
    check_with_dependency(
        "namespace Test { function Foo() : Unit { ↘ } }",
        "MyDep",
        "namespace Main { export MainFunc; function MainFunc() : Unit {} }
        namespace Other { export OtherFunc; function OtherFunc() : Unit {} }
        ",
        &["MainFunc", "OtherFunc"],
        &expect![[r#"
            found, sorted:
              "MainFunc" (Function)
                detail: "function MainFunc() : Unit"
                additional_text_edits:
                  [0:17-0:17] "import MyDep.MainFunc;\n "
              "OtherFunc" (Function)
                detail: "function OtherFunc() : Unit"
                additional_text_edits:
                  [0:17-0:17] "import MyDep.Other.OtherFunc;\n "
        "#]],
    );
}

#[test]
fn package_aliases() {
    check_with_dependency(
        "namespace Test { function Foo() : Unit { ↘ } }",
        "MyDep",
        "namespace Main { export MainFunc; function MainFunc() : Unit {} }",
        &["MyDep", "Main"],
        &expect![[r#"
            found, sorted:
              "MyDep" (Module)

            not found:
              "Main"
        "#]],
    );
}

#[test]
fn package_alias_members() {
    check_with_dependency(
        "namespace Test { function Foo() : Unit { MyDep.↘ } }",
        "MyDep",
        "namespace Main { export MainFunc; function MainFunc() : Unit {} }
        namespace Other { export OtherFunc; function OtherFunc() : Unit {} }
        namespace Other.Sub { export OtherFunc; function OtherFunc() : Unit {} }
        ",
        &["Main", "Other", "MainFunc", "Other.Sub", "Sub", "Std"],
        &expect![[r#"
            found, sorted:
              "MainFunc" (Function)
                detail: "function MainFunc() : Unit"
              "Other" (Module)

            not found:
              "Main"
              "Other.Sub"
              "Sub"
              "Std"
        "#]],
    );
}

#[test]
fn dependency_namespace_members() {
    check_with_dependency(
        "namespace Test { function Foo() : Unit { MyDep.Other.↘ } }",
        "MyDep",
        "namespace Main { export MainFunc; function MainFunc() : Unit {} }
        namespace Other { export OtherFunc; function OtherFunc() : Unit {} }
        namespace Other.Sub { export OtherFunc; function OtherFunc() : Unit {} }
        ",
        &["Main", "Other", "MainFunc", "Other.Sub", "Sub", "OtherFunc"],
        &expect![[r#"
            found, sorted:
              "OtherFunc" (Function)
                detail: "function OtherFunc() : Unit"
              "Sub" (Module)

            not found:
              "Main"
              "Other"
              "MainFunc"
              "Other.Sub"
        "#]],
    );
}

#[test]
fn package_alias_members_in_open() {
    check_with_dependency(
        "namespace Test { open MyDep.↘  }",
        "MyDep",
        "namespace Main { export MainFunc; function MainFunc() : Unit {} }
        namespace Other { export OtherFunc; function OtherFunc() : Unit {} }
        namespace Other.Sub { export OtherFunc; function OtherFunc() : Unit {} }
        ",
        &["Main", "Other", "MainFunc", "Other.Sub", "Sub"],
        &expect![[r#"
            found, sorted:
              "Other" (Module)

            not found:
              "Main"
              "MainFunc"
              "Other.Sub"
              "Sub"
        "#]],
    );
}

#[test]
fn member_completion_in_imported_namespace_from_dependency() {
    check_with_dependency(
        "namespace Main {
            open MyDep.Test;
            operation Main() : Unit {
               Foo.↘
            }
        }",
        "MyDep",
        "
        namespace Test.Foo {
            function CallableInFoo() : Unit {}
            export CallableInFoo;
        }

        namespace Test.Foo.Bar {
            function CallableInBar()  : Unit {}
            export CallableInBar;
        }
        ",
        &["CallableInFoo", "Bar"],
        &expect![[r#"
            found, sorted:
              "CallableInFoo" (Function)
                detail: "function CallableInFoo() : Unit"
              "Bar" (Module)
        "#]],
    );
}

#[test]
fn aliased_namespace_in_dependency() {
    check_with_dependency(
        "namespace Main {
            open MyDep.Test.Foo as Alias;
            operation Main() : Unit {
               Alias.↘
            }
        }",
        "MyDep",
        "
        namespace Test.Foo {
            function CallableInFoo() : Unit {}
            export CallableInFoo;
        }

        namespace Test.Foo.Bar {
            function CallableInBar()  : Unit {}
            export CallableInBar;
        }
        ",
        &["CallableInFoo", "Bar"],
        &expect![[r#"
            found, sorted:
              "CallableInFoo" (Function)
                detail: "function CallableInFoo() : Unit"
              "Bar" (Module)
        "#]],
    );
}

#[test]
fn open_does_not_match_pkg_alias() {
    check_with_dependency(
        "namespace Main {
            open Test.Foo as Alias;
            operation Main() : Unit {
               Alias.↘
            }
        }",
        "MyDep",
        "
        namespace Test.Foo {
            function CallableInFoo() : Unit {}
            export CallableInFoo;
        }

        namespace Test.Foo.Bar {
            function CallableInBar()  : Unit {}
            export CallableInBar;
        }
        ",
        &["CallableInFoo", "Bar"],
        &expect![[r#"
            not found:
              "CallableInFoo"
              "Bar"
        "#]],
    );
}

#[test]
fn field_access_expr() {
    check_single_file(
        "namespace Test {
        struct Foo {
            bar : Int,
        }

        function Main() : Unit {
            (new Foo { bar = 3 }).↘
        }
    }",
        &["bar"],
        &expect![[r#"
            found, sorted:
              "bar" (Field)
                detail: "Int"
        "#]],
    );
}

#[test]
fn input_type_missing() {
    check_single_file(
        "namespace Test { function Foo(x : FakeStdLib.↘ ) : Unit { body intrinsic; } }",
        &["Udt", "Library"],
        &expect![[r#"
            found, sorted:
              "Udt" (Interface)
                detail: "struct Udt { x : Int, y : Int }"
              "Library" (Module)
        "#]],
    );
}

#[test]
fn notebook_top_level_path_part() {
    check_notebook(
        &[(
            "cell1",
            ("
        FakeStdLib.↘
    "),
        )],
        &["Udt", "Library", "FakeStdLib", "FakeWithParam"],
        &expect![[r#"
            found, sorted:
              "FakeWithParam" (Function)
                detail: "operation FakeWithParam(x : Int) : Unit"
              "Udt" (Interface)
                detail: "struct Udt { x : Int, y : Int }"
              "Library" (Module)

            not found:
              "FakeStdLib"
        "#]],
    );
}

#[test]
fn field_access_path() {
    check_single_file(
        "namespace Test {
        struct Foo {
            bar : Int,
        }

        function Main() : Unit {
            let foo = new Foo { bar = 3 };
            foo.↘
        }
    }",
        &["bar"],
        &expect![[r#"
            found, sorted:
              "bar" (Field)
                detail: "Int"
        "#]],
    );
}

#[test]
fn notebook_top_level_path_part_in_type() {
    check_notebook(
        &[(
            "cell1",
            ("
        let x : FakeStdLib.↘
    "),
        )],
        &["Udt", "Library", "FakeStdLib", "FakeWithParam"],
        &expect![[r#"
            found, sorted:
              "Udt" (Interface)
                detail: "struct Udt { x : Int, y : Int }"
              "Library" (Module)

            not found:
              "FakeStdLib"
              "FakeWithParam"
        "#]],
    );
}

#[test]
fn prefix_ops() {
    check_single_file(
        "namespace Test { function Main() : Unit { let x = ↘ ; } }",
        &["and", "or", "not", "Adjoint"],
        &expect![[r#"
            found, sorted:
              "Adjoint" (Keyword)
              "not" (Keyword)

            not found:
              "and"
              "or"
        "#]],
    );
}

#[test]
fn binary_ops() {
    check_single_file(
        "namespace Test { function Main() : Unit { let x = 1 ↘ ; } }",
        &["and", "or", "not"],
        &expect![[r#"
            found, sorted:
              "and" (Keyword)
              "or" (Keyword)

            not found:
              "not"
        "#]],
    );
}

#[test]
fn array_size() {
    check_single_file(
        "namespace Test { function Main() : Unit { let x = [0, ↘] ; } }",
        &["size"],
        &expect![[r#"
            found, sorted:
              "size" (Keyword)
        "#]],
    );
}

#[test]
fn path_segment_partial_ident_is_keyword() {
    check_single_file(
        "namespace Test { import FakeStdLib.struct↘ }",
        &["StructFn"],
        &expect![[r#"
            found, sorted:
              "StructFn" (Interface)
                detail: "struct StructFn { inner : (Int -> Int) }"
        "#]],
    );
}

#[test]
fn path_segment_followed_by_wslash() {
    // `w/` is a single token, so it gets tricky
    // to separate out the `w` and treat it as an identifier.
    // We're just not going to worry about doing anything clever here.
    check_single_file(
        "namespace Test { import FakeStdLib.w↘/ }",
        &["StructFn"],
        &expect![[r#"
            not found:
              "StructFn"
        "#]],
    );
}

#[test]
fn path_segment_followed_by_op_token() {
    // Invoking in the middle of a multi-character op token
    // shouldn't break anything.
    check_single_file(
        "namespace Test { import FakeStdLib.<↘<< }",
        &["StructFn"],
        &expect![[r#"
            not found:
              "StructFn"
        "#]],
    );
}

#[test]
fn path_segment_before_glob() {
    check_single_file(
        "namespace Test { import FakeStdLib.↘* }",
        &["StructFn"],
        &expect![[r#"
            found, sorted:
              "StructFn" (Interface)
                detail: "struct StructFn { inner : (Int -> Int) }"
        "#]],
    );
}

#[test]
fn field_in_initializer() {
    check_single_file(
        "namespace Test {
        struct Foo {
            bar : Int,
        }

        function Main() : Unit {
            new Foo { ↘ };
        }
    }",
        &["bar"],
        &expect![[r#"
            found, sorted:
              "bar" (Field)
                detail: "Int"
        "#]],
    );
}

#[test]
fn stdlib_struct_field_init() {
    check_single_file(
        "namespace Test {
            import FakeStdLib.FakeStruct as StructAlias;
            function Main() : Unit {
                new StructAlias { ↘ };
            }
        }",
        &["x"],
        &expect![[r#"
            found, sorted:
              "x" (Field)
                detail: "Int"
        "#]],
    );
}

#[test]
fn newtype_named_field() {
    check_single_file(
        "namespace Test {
            newtype Foo = (field : Int);
            function Main() : Unit {
                Foo(3).↘
            }
        }",
        &["field"],
        &expect![[r#"
            found, sorted:
              "field" (Field)
                detail: "Int"
        "#]],
    );
}

#[test]
fn field_access_path_chained() {
    check_single_file(
        "namespace Test {
            newtype Foo = ( fieldFoo : Int );
            struct Bar { fieldBar : Foo );
            function Main() : Unit {
                let bar = new Bar { fieldBar = Foo(3) };
                bar.fieldBar.↘
            }
        }",
        &["fieldFoo", "fieldBar"],
        &expect![[r#"
            found, sorted:
              "fieldFoo" (Field)
                detail: "Int"

            not found:
              "fieldBar"
        "#]],
    );
}

#[test]
fn field_access_expr_chained() {
    check_single_file(
        "namespace Test {
            newtype Foo = ( fieldFoo : Int );
            struct Bar { fieldBar : Foo );
            function Main() : Unit {
                (new Bar { fieldBar = Foo(3) }).fieldBar.↘
            }
        }",
        &["fieldFoo", "fieldBar"],
        &expect![[r#"
            found, sorted:
              "fieldFoo" (Field)
                detail: "Int"

            not found:
              "fieldBar"
        "#]],
    );
}

#[test]
fn field_assignment_rhs() {
    check_single_file(
        "namespace Test {
        struct Foo {
            bar : Int,
        }

        function Main() : Unit {
            let var = 3;
            new Foo { bar = ↘ };
        }
    }",
        &["bar", "var"],
        &expect![[r#"
            found, sorted:
              "var" (Variable)
                detail: "var : Int"

            not found:
              "bar"
        "#]],
    );
}

#[test]
fn field_access_local_shadows_global() {
    check_single_file(
        "namespace Test {
        struct Foo {
            bar : Int,
        }

        function Main() : Unit {
            let FakeStdLib = new Foo { bar = 3 };
            FakeStdLib.↘
        }
    }",
        &["Fake", "bar"],
        &expect![[r#"
            found, sorted:
              "bar" (Field)
                detail: "Int"

            not found:
              "Fake"
        "#]],
    );
}

#[test]
fn ty_param_in_signature() {
    check_single_file(
        r"namespace Test {
            operation Test<'T>(x: ↘) : Unit {}
        }",
        &["'T", "FakeStdLib"],
        &expect![[r#"
            found, sorted:
              "'T" (TypeParameter)
              "FakeStdLib" (Module)
        "#]],
    );
}

#[test]
fn ty_param_in_return_type() {
    check_single_file(
        r"namespace Test {
            operation Test<'T>(x: 'T) : ↘ {}
        }",
        &["'T", "FakeStdLib"],
        &expect![[r#"
            found, sorted:
              "'T" (TypeParameter)
              "FakeStdLib" (Module)
        "#]],
    );
}

#[test]
fn path_segment_in_return_type() {
    check_single_file(
        r"namespace Test {
            operation Test(x: 'T) : FakeStdLib.↘ {}
        }",
        &["Udt"],
        &expect![[r#"
            found, sorted:
              "Udt" (Interface)
                detail: "struct Udt { x : Int, y : Int }"
        "#]],
    );
}

#[test]
fn return_type_in_partial_callable_signature() {
    check_single_file(
        r"namespace Test {
            operation Test<'T>() : ↘
        }",
        &["'T", "FakeStdLib"],
        &expect![[r#"
            found, sorted:
              "'T" (TypeParameter)
              "FakeStdLib" (Module)
        "#]],
    );
}

#[test]
fn arg_type_in_partial_callable_signature() {
    check_single_file(
        r"namespace Test {
            operation Test<'T>(x: ↘)
        }",
        &["'T", "FakeStdLib"],
        &expect![[r#"
            found, sorted:
              "'T" (TypeParameter)
              "FakeStdLib" (Module)
        "#]],
    );
}

#[test]
fn incomplete_return_type_in_partial_callable_signature() {
    check_single_file(
        r"namespace Test {
            operation Test<'T>() : () => ↘
        }",
        &["'T", "FakeStdLib"],
        &expect![[r#"
            found, sorted:
              "'T" (TypeParameter)
              "FakeStdLib" (Module)
        "#]],
    );
}

#[test]
fn no_path_segment_completion_inside_attr() {
    check_no_completions(
        "namespace Test {

        @Config(FakeStdLib.↘)
        function Main() : Unit {
        }
    }",
    );
}

#[test]
fn no_completion_inside_attr() {
    check_no_completions(
        "namespace Test {

        @Config(↘)
        function Main() : Unit {
        }
    }",
    );
}

#[test]
fn in_comment() {
    check_no_completions(
        "namespace Test {
            import Foo;
            // Hello there ↘
            import Bar;
        }",
    );
}

#[test]
fn in_doc_comment() {
    check_no_completions(
        "namespace Test {
            import Foo;
            /// Hello there ↘
            import Bar;
        }",
    );
}

#[test]
fn in_trailing_comment() {
    check_no_completions(
        "namespace Test {
            import Foo; // Hello there ↘
        }",
    );
}

#[test]
fn export_from_dependency_in_scope() {
    check_with_dependency(
        r"
        namespace Test {
            open MyDep;
            operation Foo() : Unit {
                ↘
            }
        }
        ",
        "MyDep",
        "
        namespace Bar {
            operation Baz() : Unit {}
            export Baz;
        }
        namespace Main {
            operation Qux() : Unit {}
            export Qux, Bar.Baz;
        }
        ",
        &["Qux", "Baz", "Bar"],
        &expect![[r#"
            found, sorted:
              "Baz" (Function)
                detail: "operation Baz() : Unit"
              "Qux" (Function)
                detail: "operation Qux() : Unit"
              "Bar" (Module)
        "#]],
    );
}

#[test]
fn aliased_export_from_dependency_in_scope() {
    check_with_dependency(
        r"
        namespace Test {
            open MyDep;
            operation Foo() : Unit {
                ↘
            }
        }
        ",
        "MyDep",
        "
        namespace Bar {
            operation Baz() : Unit {}
            export Baz;
        }
        namespace Main {
            export Bar.Baz, Bar.Baz as BazAlias;
        }
        ",
        &["BazAlias", "Baz"],
        &expect![[r#"
            found, sorted:
              "Baz" (Function)
                detail: "operation Baz() : Unit"
              "BazAlias" (Function)
                detail: "operation Baz() : Unit"
        "#]],
    );
}

#[test]
fn namespace_export_from_dependency_qualified() {
    check_with_dependency(
        r"
        namespace Test {
            open MyDep.Baz.↘
        }",
        "MyDep",
        "namespace Foo.Bar {
            operation Qux() : Unit {}
            export Qux
         }
         namespace Baz {
            export Foo.Bar;
         }",
        &["Bar"],
        &expect![[r#"
            found, sorted:
              "Bar" (Module)
        "#]],
    );
}

#[test]
fn export_from_dependency_qualified() {
    check_with_dependency(
        r"
            namespace Test {
                operation Test() : Unit {
                    MyDep.↘
                }
            }",
        "MyDep",
        "namespace Foo {
                operation Baz() : Unit {}
                export Baz;
             }
             namespace Main {
                operation Qux() : Unit {}
                export Qux, Foo.Baz;
             }",
        &["Qux", "Baz"],
        &expect![[r#"
            found, sorted:
              "Baz" (Function)
                detail: "operation Baz() : Unit"
              "Qux" (Function)
                detail: "operation Qux() : Unit"
        "#]],
    );
}

#[test]
fn reexport_namespace_from_dependency_members() {
    check_with_dependency(
        r"
        namespace Test {
            operation Main() : Unit {
                MyDep.Baz.Bar.↘
            }
        }",
        "MyDep",
        "namespace Foo.Bar {
            operation Zud() : Unit {}
            export Zud
         }
         namespace Baz {
            operation Qux() : Unit {}
            export Qux, Foo.Bar;
         }",
        &["Zud"],
        &expect![[r#"
            found, sorted:
              "Zud" (Function)
                detail: "operation Zud() : Unit"
        "#]],
    );
}

#[test]
fn import_in_local_scope() {
    check_single_file(
        r"
        namespace Foo {
            operation Bar() : Unit {}
        }
        namespace A {
            operation Main() : Unit {
                import Foo.Bar;
                ↘
            }
        }",
        &["Bar"],
        &expect![[r#"
            found, sorted:
              "Bar" (Function)
                detail: "operation Bar() : Unit"
        "#]],
    );
}