PY-56120 Add tests

GitOrigin-RevId: 12b363fa88b6722f6056cb82f7f8a91cacd3336a
2026-09-27 10:03:11 +07:00 · 2022-11-14 06:32:11 +00:00
parent 5d9d5042e0
commit 66297780d3
2 changed files with 160 additions and 0 deletions
@@ -0,0 +1,13 @@
+import asyncio
+
+
+async def foo(y):
+    return y + 1
+
+
+async def print_foo():
+    x = await foo(1)
+    print(x)
+
+
+asyncio.run(print_foo())
@@ -0,0 +1,147 @@
+// Copyright 2000-2022 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
+package com.jetbrains.env.debug;
+
+import com.intellij.xdebugger.XDebuggerTestUtil;
+import com.jetbrains.env.EnvTestTagsRequired;
+import com.jetbrains.env.PyEnvTestCase;
+import org.junit.Test;
+
+public class PythonDebuggerAsyncioTest extends PyEnvTestCase {
+
+  private enum TestCase {
+    CONSOLE,
+    EVALUATE,
+    BREAKPOINT
+  }
+
+  private static class AsyncioPyDebuggerTask extends PyDebuggerTask {
+    private static final String RELATIVE_PATH = "/debug";
+
+    private static final String SCRIPT_NAME = "test_asyncio_debugger.py";
+
+    private static final String AWAIT_FOO = "await foo(1)";
+
+    private static final String RUN_FOO = "asyncio.run(foo(1))";
+
+    private static final String RUN_FOO_WITH_LOOP = "loop.run_until_complete(foo(1))";
+
+    private static final String GET_EVENT_LOOP = "loop = asyncio.get_event_loop()";
+
+    private static final String CLOSE_EVENT_LOOP = "loop.close()";
+
+    private static final String RUN_UNTIL_COMPLETE = "asyncio.get_event_loop().run_until_complete(foo(1))";
+
+    private final TestCase myTestCase;
+
+    private AsyncioPyDebuggerTask(TestCase testCase) {
+      super(RELATIVE_PATH, SCRIPT_NAME);
+      myTestCase = testCase;
+    }
+
+    protected void testConsole() throws Exception {
+      consoleExec(AWAIT_FOO);
+      waitForOutput("2");
+      consoleExec(RUN_FOO);
+      waitForOutput("2");
+      consoleExec(GET_EVENT_LOOP);
+      consoleExec(RUN_FOO_WITH_LOOP);
+      waitForOutput("2");
+      consoleExec(CLOSE_EVENT_LOOP);
+    }
+
+    protected void testEvaluate() {
+      Variable await_foo = eval(AWAIT_FOO);
+      await_foo.hasValue("2");
+      Variable run_foo = eval(RUN_FOO);
+      run_foo.hasValue("2");
+      Variable loop_foo = eval(RUN_UNTIL_COMPLETE);
+      loop_foo.hasValue("2");
+    }
+
+    protected void testBreakpoints() throws Exception {
+      resume();
+      waitForOutput("2");
+      waitForTerminate();
+    }
+
+    @Override
+    public void testing() throws Exception {
+      waitForPause();
+      switch (myTestCase) {
+        case CONSOLE -> testConsole();
+        case EVALUATE -> testEvaluate();
+        case BREAKPOINT -> testBreakpoints();
+      }
+    }
+
+    @Override
+    public void before() {
+      switch (myTestCase) {
+        case CONSOLE, EVALUATE -> toggleBreakpoint(9);
+        case BREAKPOINT -> {
+          toggleBreakpoint(8);
+          toggleBreakpoint(9);
+          XDebuggerTestUtil.setBreakpointCondition(getProject(), 8, "await foo(1) == 2");
+          XDebuggerTestUtil.setBreakpointCondition(getProject(), 9, "await foo(1) != 2");
+          XDebuggerTestUtil.setBreakpointLogExpression(getProject(), 8, "await foo(1)");
+        }
+      }
+      setWaitForTermination(false);
+    }
+  }
+
+  @EnvTestTagsRequired(tags = "python3.8")
+  @Test
+  public void testAsyncioConsole38() {
+    runPythonTest(new AsyncioPyDebuggerTask(TestCase.CONSOLE));
+  }
+
+  @EnvTestTagsRequired(tags = "python3.9")
+  @Test
+  public void testAsyncioConsole39() {
+    runPythonTest(new AsyncioPyDebuggerTask(TestCase.CONSOLE));
+  }
+
+  @EnvTestTagsRequired(tags = "python3.10")
+  @Test
+  public void testAsyncioConsole310() {
+    runPythonTest(new AsyncioPyDebuggerTask(TestCase.CONSOLE));
+  }
+
+
+  @EnvTestTagsRequired(tags = "python3.8")
+  @Test
+  public void testAsyncioEvaluate38() {
+    runPythonTest(new AsyncioPyDebuggerTask(TestCase.EVALUATE));
+  }
+
+  @EnvTestTagsRequired(tags = "python3.9")
+  @Test
+  public void testAsyncioEvaluate39() {
+    runPythonTest(new AsyncioPyDebuggerTask(TestCase.EVALUATE));
+  }
+
+  @EnvTestTagsRequired(tags = "python3.10")
+  @Test
+  public void testAsyncioEvaluate310() {
+    runPythonTest(new AsyncioPyDebuggerTask(TestCase.EVALUATE));
+  }
+
+  @EnvTestTagsRequired(tags = "python3.8")
+  @Test
+  public void testAsyncioBreakpoint38() {
+    runPythonTest(new AsyncioPyDebuggerTask(TestCase.BREAKPOINT));
+  }
+
+  @EnvTestTagsRequired(tags = "python3.9")
+  @Test
+  public void testAsyncioBreakpoint39() {
+    runPythonTest(new AsyncioPyDebuggerTask(TestCase.BREAKPOINT));
+  }
+
+  @EnvTestTagsRequired(tags = "python3.10")
+  @Test
+  public void testAsyncioBreakpoint310() {
+    runPythonTest(new AsyncioPyDebuggerTask(TestCase.BREAKPOINT));
+  }
+}