From ea27432d3133aa4ea49be9c84ada547338b42c05 Mon Sep 17 00:00:00 2001 From: qdegraaf Date: Thu, 9 Nov 2023 14:41:10 +0100 Subject: [PATCH] Add test for multi target --- .../test/fixtures/flake8_trio/TRIO115.py | 4 ++ ...lake8_trio__tests__TRIO115_TRIO115.py.snap | 57 +++++++++++++------ 2 files changed, 43 insertions(+), 18 deletions(-) diff --git a/crates/ruff_linter/resources/test/fixtures/flake8_trio/TRIO115.py b/crates/ruff_linter/resources/test/fixtures/flake8_trio/TRIO115.py index 117c2270f29406..fb95030176f541 100644 --- a/crates/ruff_linter/resources/test/fixtures/flake8_trio/TRIO115.py +++ b/crates/ruff_linter/resources/test/fixtures/flake8_trio/TRIO115.py @@ -17,6 +17,10 @@ async def func(): sleep(0) # TRIO115 + x, y = 0, 2000 + trio.sleep(x) # TRIO115 + trio.sleep(y) # OK + trio.sleep(0) # TRIO115 diff --git a/crates/ruff_linter/src/rules/flake8_trio/snapshots/ruff_linter__rules__flake8_trio__tests__TRIO115_TRIO115.py.snap b/crates/ruff_linter/src/rules/flake8_trio/snapshots/ruff_linter__rules__flake8_trio__tests__TRIO115_TRIO115.py.snap index b273e0333b8e4a..be1a56c274d6ed 100644 --- a/crates/ruff_linter/src/rules/flake8_trio/snapshots/ruff_linter__rules__flake8_trio__tests__TRIO115_TRIO115.py.snap +++ b/crates/ruff_linter/src/rules/flake8_trio/snapshots/ruff_linter__rules__flake8_trio__tests__TRIO115_TRIO115.py.snap @@ -69,6 +69,8 @@ TRIO115.py:18:5: TRIO115 [*] Use `trio.lowlevel.checkpoint()` instead of `trio.s 17 | 18 | sleep(0) # TRIO115 | ^^^^^^^^ TRIO115 +19 | +20 | x, y = 0, 2000 | = help: Replace with `trio.lowlevel.checkpoint()` @@ -79,39 +81,58 @@ TRIO115.py:18:5: TRIO115 [*] Use `trio.lowlevel.checkpoint()` instead of `trio.s 18 |- sleep(0) # TRIO115 18 |+ trio.lowlevel.checkpoint # TRIO115 19 19 | -20 20 | -21 21 | trio.sleep(0) # TRIO115 +20 20 | x, y = 0, 2000 +21 21 | trio.sleep(x) # TRIO115 -TRIO115.py:21:1: TRIO115 [*] Use `trio.lowlevel.checkpoint()` instead of `trio.sleep(0)` +TRIO115.py:21:5: TRIO115 [*] Use `trio.lowlevel.checkpoint()` instead of `trio.sleep(0)` | -21 | trio.sleep(0) # TRIO115 - | ^^^^^^^^^^^^^ TRIO115 +20 | x, y = 0, 2000 +21 | trio.sleep(x) # TRIO115 + | ^^^^^^^^^^^^^ TRIO115 +22 | trio.sleep(y) # OK | = help: Replace with `trio.lowlevel.checkpoint()` ℹ Fix 18 18 | sleep(0) # TRIO115 19 19 | -20 20 | -21 |-trio.sleep(0) # TRIO115 - 21 |+trio.lowlevel.checkpoint # TRIO115 -22 22 | +20 20 | x, y = 0, 2000 +21 |- trio.sleep(x) # TRIO115 + 21 |+ trio.lowlevel.checkpoint # TRIO115 +22 22 | trio.sleep(y) # OK 23 23 | -24 24 | def func(): +24 24 | -TRIO115.py:25:14: TRIO115 [*] Use `trio.lowlevel.checkpoint()` instead of `trio.sleep(0)` +TRIO115.py:25:1: TRIO115 [*] Use `trio.lowlevel.checkpoint()` instead of `trio.sleep(0)` | -24 | def func(): -25 | trio.run(trio.sleep(0)) # TRIO115 - | ^^^^^^^^^^^^^ TRIO115 +25 | trio.sleep(0) # TRIO115 + | ^^^^^^^^^^^^^ TRIO115 | = help: Replace with `trio.lowlevel.checkpoint()` ℹ Fix -22 22 | +22 22 | trio.sleep(y) # OK 23 23 | -24 24 | def func(): -25 |- trio.run(trio.sleep(0)) # TRIO115 - 25 |+ trio.run(trio.lowlevel.checkpoint) # TRIO115 +24 24 | +25 |-trio.sleep(0) # TRIO115 + 25 |+trio.lowlevel.checkpoint # TRIO115 +26 26 | +27 27 | +28 28 | def func(): + +TRIO115.py:29:14: TRIO115 [*] Use `trio.lowlevel.checkpoint()` instead of `trio.sleep(0)` + | +28 | def func(): +29 | trio.run(trio.sleep(0)) # TRIO115 + | ^^^^^^^^^^^^^ TRIO115 + | + = help: Replace with `trio.lowlevel.checkpoint()` + +ℹ Fix +26 26 | +27 27 | +28 28 | def func(): +29 |- trio.run(trio.sleep(0)) # TRIO115 + 29 |+ trio.run(trio.lowlevel.checkpoint) # TRIO115