mirror of
https://github.com/astral-sh/ruff.git
synced 2025-09-30 22:01:18 +00:00
[ty] add more lsp tests for overloads (#20148)
I decided to split out the addition of these tests from other PRs so that it's easier to follow changes to the LSP's function call handling. I'm not particularly concerned with whether the results produced by these tests are "good" or "bad" in this PR, I'm just establishing a baseline.
This commit is contained in:
Aria Desires
GitHub
parent
bbfcf6e111
commit
f40a0b3800
4 changed files with 1656 additions and 2 deletions
|
|
@ -1355,6 +1355,462 @@ class MyClass:
|
|||
"#);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn goto_declaration_overload_type_disambiguated1() {
|
||||
let test = CursorTest::builder()
|
||||
.source(
|
||||
"main.py",
|
||||
"
|
||||
from mymodule import ab
|
||||
|
||||
a<CURSOR>b(1)
|
||||
",
|
||||
)
|
||||
.source(
|
||||
"mymodule.py",
|
||||
r#"
|
||||
def ab(a):
|
||||
"""the real implementation!"""
|
||||
"#,
|
||||
)
|
||||
.source(
|
||||
"mymodule.pyi",
|
||||
r#"
|
||||
from typing import overload
|
||||
|
||||
@overload
|
||||
def ab(a: int): ...
|
||||
|
||||
@overload
|
||||
def ab(a: str): ...
|
||||
"#,
|
||||
)
|
||||
.build();
|
||||
|
||||
assert_snapshot!(test.goto_declaration(), @r"
|
||||
info[goto-declaration]: Declaration
|
||||
--> mymodule.pyi:5:5
|
||||
|
|
||||
4 | @overload
|
||||
5 | def ab(a: int): ...
|
||||
| ^^
|
||||
6 |
|
||||
7 | @overload
|
||||
|
|
||||
info: Source
|
||||
--> main.py:4:1
|
||||
|
|
||||
2 | from mymodule import ab
|
||||
3 |
|
||||
4 | ab(1)
|
||||
| ^^
|
||||
|
|
||||
|
||||
info[goto-declaration]: Declaration
|
||||
--> mymodule.pyi:8:5
|
||||
|
|
||||
7 | @overload
|
||||
8 | def ab(a: str): ...
|
||||
| ^^
|
||||
|
|
||||
info: Source
|
||||
--> main.py:4:1
|
||||
|
|
||||
2 | from mymodule import ab
|
||||
3 |
|
||||
4 | ab(1)
|
||||
| ^^
|
||||
|
|
||||
");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn goto_declaration_overload_type_disambiguated2() {
|
||||
let test = CursorTest::builder()
|
||||
.source(
|
||||
"main.py",
|
||||
r#"
|
||||
from mymodule import ab
|
||||
|
||||
a<CURSOR>b("hello")
|
||||
"#,
|
||||
)
|
||||
.source(
|
||||
"mymodule.py",
|
||||
r#"
|
||||
def ab(a):
|
||||
"""the real implementation!"""
|
||||
"#,
|
||||
)
|
||||
.source(
|
||||
"mymodule.pyi",
|
||||
r#"
|
||||
from typing import overload
|
||||
|
||||
@overload
|
||||
def ab(a: int): ...
|
||||
|
||||
@overload
|
||||
def ab(a: str): ...
|
||||
"#,
|
||||
)
|
||||
.build();
|
||||
|
||||
assert_snapshot!(test.goto_declaration(), @r#"
|
||||
info[goto-declaration]: Declaration
|
||||
--> mymodule.pyi:5:5
|
||||
|
|
||||
4 | @overload
|
||||
5 | def ab(a: int): ...
|
||||
| ^^
|
||||
6 |
|
||||
7 | @overload
|
||||
|
|
||||
info: Source
|
||||
--> main.py:4:1
|
||||
|
|
||||
2 | from mymodule import ab
|
||||
3 |
|
||||
4 | ab("hello")
|
||||
| ^^
|
||||
|
|
||||
|
||||
info[goto-declaration]: Declaration
|
||||
--> mymodule.pyi:8:5
|
||||
|
|
||||
7 | @overload
|
||||
8 | def ab(a: str): ...
|
||||
| ^^
|
||||
|
|
||||
info: Source
|
||||
--> main.py:4:1
|
||||
|
|
||||
2 | from mymodule import ab
|
||||
3 |
|
||||
4 | ab("hello")
|
||||
| ^^
|
||||
|
|
||||
"#);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn goto_declaration_overload_arity_disambiguated1() {
|
||||
let test = CursorTest::builder()
|
||||
.source(
|
||||
"main.py",
|
||||
"
|
||||
from mymodule import ab
|
||||
|
||||
a<CURSOR>b(1, 2)
|
||||
",
|
||||
)
|
||||
.source(
|
||||
"mymodule.py",
|
||||
r#"
|
||||
def ab(a, b = None):
|
||||
"""the real implementation!"""
|
||||
"#,
|
||||
)
|
||||
.source(
|
||||
"mymodule.pyi",
|
||||
r#"
|
||||
from typing import overload
|
||||
|
||||
@overload
|
||||
def ab(a: int, b: int): ...
|
||||
|
||||
@overload
|
||||
def ab(a: int): ...
|
||||
"#,
|
||||
)
|
||||
.build();
|
||||
|
||||
assert_snapshot!(test.goto_declaration(), @r"
|
||||
info[goto-declaration]: Declaration
|
||||
--> mymodule.pyi:5:5
|
||||
|
|
||||
4 | @overload
|
||||
5 | def ab(a: int, b: int): ...
|
||||
| ^^
|
||||
6 |
|
||||
7 | @overload
|
||||
|
|
||||
info: Source
|
||||
--> main.py:4:1
|
||||
|
|
||||
2 | from mymodule import ab
|
||||
3 |
|
||||
4 | ab(1, 2)
|
||||
| ^^
|
||||
|
|
||||
|
||||
info[goto-declaration]: Declaration
|
||||
--> mymodule.pyi:8:5
|
||||
|
|
||||
7 | @overload
|
||||
8 | def ab(a: int): ...
|
||||
| ^^
|
||||
|
|
||||
info: Source
|
||||
--> main.py:4:1
|
||||
|
|
||||
2 | from mymodule import ab
|
||||
3 |
|
||||
4 | ab(1, 2)
|
||||
| ^^
|
||||
|
|
||||
");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn goto_declaration_overload_arity_disambiguated2() {
|
||||
let test = CursorTest::builder()
|
||||
.source(
|
||||
"main.py",
|
||||
"
|
||||
from mymodule import ab
|
||||
|
||||
a<CURSOR>b(1)
|
||||
",
|
||||
)
|
||||
.source(
|
||||
"mymodule.py",
|
||||
r#"
|
||||
def ab(a, b = None):
|
||||
"""the real implementation!"""
|
||||
"#,
|
||||
)
|
||||
.source(
|
||||
"mymodule.pyi",
|
||||
r#"
|
||||
from typing import overload
|
||||
|
||||
@overload
|
||||
def ab(a: int, b: int): ...
|
||||
|
||||
@overload
|
||||
def ab(a: int): ...
|
||||
"#,
|
||||
)
|
||||
.build();
|
||||
|
||||
assert_snapshot!(test.goto_declaration(), @r"
|
||||
info[goto-declaration]: Declaration
|
||||
--> mymodule.pyi:5:5
|
||||
|
|
||||
4 | @overload
|
||||
5 | def ab(a: int, b: int): ...
|
||||
| ^^
|
||||
6 |
|
||||
7 | @overload
|
||||
|
|
||||
info: Source
|
||||
--> main.py:4:1
|
||||
|
|
||||
2 | from mymodule import ab
|
||||
3 |
|
||||
4 | ab(1)
|
||||
| ^^
|
||||
|
|
||||
|
||||
info[goto-declaration]: Declaration
|
||||
--> mymodule.pyi:8:5
|
||||
|
|
||||
7 | @overload
|
||||
8 | def ab(a: int): ...
|
||||
| ^^
|
||||
|
|
||||
info: Source
|
||||
--> main.py:4:1
|
||||
|
|
||||
2 | from mymodule import ab
|
||||
3 |
|
||||
4 | ab(1)
|
||||
| ^^
|
||||
|
|
||||
");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn goto_declaration_overload_keyword_disambiguated1() {
|
||||
let test = CursorTest::builder()
|
||||
.source(
|
||||
"main.py",
|
||||
"
|
||||
from mymodule import ab
|
||||
|
||||
a<CURSOR>b(1, b=2)
|
||||
",
|
||||
)
|
||||
.source(
|
||||
"mymodule.py",
|
||||
r#"
|
||||
def ab(a, *, b = None, c = None):
|
||||
"""the real implementation!"""
|
||||
"#,
|
||||
)
|
||||
.source(
|
||||
"mymodule.pyi",
|
||||
r#"
|
||||
from typing import overload
|
||||
|
||||
@overload
|
||||
def ab(a: int): ...
|
||||
|
||||
@overload
|
||||
def ab(a: int, *, b: int): ...
|
||||
|
||||
@overload
|
||||
def ab(a: int, *, c: int): ...
|
||||
"#,
|
||||
)
|
||||
.build();
|
||||
|
||||
assert_snapshot!(test.goto_declaration(), @r"
|
||||
info[goto-declaration]: Declaration
|
||||
--> mymodule.pyi:5:5
|
||||
|
|
||||
4 | @overload
|
||||
5 | def ab(a: int): ...
|
||||
| ^^
|
||||
6 |
|
||||
7 | @overload
|
||||
|
|
||||
info: Source
|
||||
--> main.py:4:1
|
||||
|
|
||||
2 | from mymodule import ab
|
||||
3 |
|
||||
4 | ab(1, b=2)
|
||||
| ^^
|
||||
|
|
||||
|
||||
info[goto-declaration]: Declaration
|
||||
--> mymodule.pyi:8:5
|
||||
|
|
||||
7 | @overload
|
||||
8 | def ab(a: int, *, b: int): ...
|
||||
| ^^
|
||||
9 |
|
||||
10 | @overload
|
||||
|
|
||||
info: Source
|
||||
--> main.py:4:1
|
||||
|
|
||||
2 | from mymodule import ab
|
||||
3 |
|
||||
4 | ab(1, b=2)
|
||||
| ^^
|
||||
|
|
||||
|
||||
info[goto-declaration]: Declaration
|
||||
--> mymodule.pyi:11:5
|
||||
|
|
||||
10 | @overload
|
||||
11 | def ab(a: int, *, c: int): ...
|
||||
| ^^
|
||||
|
|
||||
info: Source
|
||||
--> main.py:4:1
|
||||
|
|
||||
2 | from mymodule import ab
|
||||
3 |
|
||||
4 | ab(1, b=2)
|
||||
| ^^
|
||||
|
|
||||
");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn goto_declaration_overload_keyword_disambiguated2() {
|
||||
let test = CursorTest::builder()
|
||||
.source(
|
||||
"main.py",
|
||||
"
|
||||
from mymodule import ab
|
||||
|
||||
a<CURSOR>b(1, c=2)
|
||||
",
|
||||
)
|
||||
.source(
|
||||
"mymodule.py",
|
||||
r#"
|
||||
def ab(a, *, b = None, c = None):
|
||||
"""the real implementation!"""
|
||||
"#,
|
||||
)
|
||||
.source(
|
||||
"mymodule.pyi",
|
||||
r#"
|
||||
from typing import overload
|
||||
|
||||
@overload
|
||||
def ab(a: int): ...
|
||||
|
||||
@overload
|
||||
def ab(a: int, *, b: int): ...
|
||||
|
||||
@overload
|
||||
def ab(a: int, *, c: int): ...
|
||||
"#,
|
||||
)
|
||||
.build();
|
||||
|
||||
assert_snapshot!(test.goto_declaration(), @r"
|
||||
info[goto-declaration]: Declaration
|
||||
--> mymodule.pyi:5:5
|
||||
|
|
||||
4 | @overload
|
||||
5 | def ab(a: int): ...
|
||||
| ^^
|
||||
6 |
|
||||
7 | @overload
|
||||
|
|
||||
info: Source
|
||||
--> main.py:4:1
|
||||
|
|
||||
2 | from mymodule import ab
|
||||
3 |
|
||||
4 | ab(1, c=2)
|
||||
| ^^
|
||||
|
|
||||
|
||||
info[goto-declaration]: Declaration
|
||||
--> mymodule.pyi:8:5
|
||||
|
|
||||
7 | @overload
|
||||
8 | def ab(a: int, *, b: int): ...
|
||||
| ^^
|
||||
9 |
|
||||
10 | @overload
|
||||
|
|
||||
info: Source
|
||||
--> main.py:4:1
|
||||
|
|
||||
2 | from mymodule import ab
|
||||
3 |
|
||||
4 | ab(1, c=2)
|
||||
| ^^
|
||||
|
|
||||
|
||||
info[goto-declaration]: Declaration
|
||||
--> mymodule.pyi:11:5
|
||||
|
|
||||
10 | @overload
|
||||
11 | def ab(a: int, *, c: int): ...
|
||||
| ^^
|
||||
|
|
||||
info: Source
|
||||
--> main.py:4:1
|
||||
|
|
||||
2 | from mymodule import ab
|
||||
3 |
|
||||
4 | ab(1, c=2)
|
||||
| ^^
|
||||
|
|
||||
");
|
||||
}
|
||||
|
||||
impl CursorTest {
|
||||
fn goto_declaration(&self) -> String {
|
||||
let Some(targets) = goto_declaration(&self.db, self.cursor.file, self.cursor.offset)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue