Skip to content

Commit 2190963

Browse files
committed
fix wrapper validation test
1 parent fb66ff5 commit 2190963

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

tests/test_async_wrapper_sqlite_validation.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,11 +19,14 @@
1919
class TestAsyncWrapperSQLiteValidation:
2020

2121
@pytest.fixture
22-
def test_env_setup(self):
22+
def test_env_setup(self, request):
2323
original_env = {}
2424
test_env = {
2525
"CODEFLASH_LOOP_INDEX": "1",
2626
"CODEFLASH_TEST_ITERATION": "0",
27+
"CODEFLASH_TEST_MODULE": __name__,
28+
"CODEFLASH_TEST_CLASS": "TestAsyncWrapperSQLiteValidation",
29+
"CODEFLASH_TEST_FUNCTION": request.node.name,
2730
}
2831

2932
for key, value in test_env.items():
@@ -278,7 +281,7 @@ async def schema_test_func() -> str:
278281
assert columns == expected_columns
279282
con.close()
280283

281-
def test_sync_test_context_extraction(self):
284+
def test_sync_test_context_extraction(self, test_env_setup):
282285
from codeflash.code_utils.codeflash_wrap_decorator import extract_test_context_from_frame
283286

284287
test_module, test_class, test_func = extract_test_context_from_frame()

0 commit comments

Comments
 (0)