From 4116ab9f66f08ea2dd3914a670e874bc8438d4c7 Mon Sep 17 00:00:00 2001 From: Benedikt Bartscher Date: Tue, 5 Nov 2024 22:34:59 +0100 Subject: [PATCH] do not set appharness env twice, already done in AppHarnessProd --- tests/integration/conftest.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/tests/integration/conftest.py b/tests/integration/conftest.py index 712c30a4e..337f4e174 100644 --- a/tests/integration/conftest.py +++ b/tests/integration/conftest.py @@ -3,7 +3,7 @@ import os import re from pathlib import Path -from typing import Generator, Type +from typing import Type import pytest @@ -81,18 +81,14 @@ def pytest_exception_interact(node, call, report): ) def app_harness_env( request: pytest.FixtureRequest, -) -> Generator[Type[AppHarness], None, None]: +) -> Type[AppHarness]: """Parametrize the AppHarness class to use for the test, either dev or prod. Args: request: The pytest fixture request object. - Yields: + Returns: The AppHarness class to use for the test. """ harness: Type[AppHarness] = request.param - if issubclass(harness, AppHarnessProd): - EnvironmentVariables.REFLEX_ENV_MODE.set(reflex.constants.Env.PROD) - yield harness - if issubclass(harness, AppHarnessProd): - EnvironmentVariables.REFLEX_ENV_MODE.set(None) + return harness