Skip to content

Commit e526264

Browse files
Fix test_rpc codegen invocation
Module-level imports cause unrecoverable errors on test failure
1 parent a4e375e commit e526264

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

tests/codegen/test_rpc.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import asyncio
22
import importlib
3+
import os
34
import shutil
45
from datetime import timedelta
56
from pathlib import Path
@@ -19,9 +20,8 @@
1920

2021
@pytest.fixture(scope="session", autouse=True)
2122
def generate_rpc_client() -> None:
22-
import tests.codegen.rpc.generated
23-
24-
shutil.rmtree("tests/codegen/rpc/generated")
23+
shutil.rmtree("tests/codegen/rpc/generated", ignore_errors=True)
24+
os.makedirs("tests/codegen/rpc/generated")
2525

2626
def file_opener(path: Path) -> TextIO:
2727
return open(path, "w")
@@ -34,6 +34,12 @@ def file_opener(path: Path) -> TextIO:
3434
file_opener=file_opener,
3535
method_filter=None,
3636
)
37+
38+
39+
@pytest.fixture(scope="session", autouse=True)
40+
def reload_rpc_import(generate_rpc_client: None) -> None:
41+
import tests.codegen.rpc.generated
42+
3743
importlib.reload(tests.codegen.rpc.generated)
3844

3945

0 commit comments

Comments
 (0)