Refactor test files to replace 'stress_framework' with 'gradual' in import paths

- Updated import statements in test_orchestrator.py, test_parser.py, test_phase.py, and test_runner.py to reflect the new package name.
- Ensured all relevant patches and mock setups are aligned with the updated package structure.

Author: adityakedawat

tests/test_orchestrator.py

@@ -63,15 +63,15 @@ def mock_parser(mock_scenario_config):
 
 @pytest.fixture
 def orchestrator(mock_parser):
-    with patch("stress_framework.base.orchestrator.Parser", return_value=mock_parser):
+    with patch("gradual.base.orchestrator.Parser", return_value=mock_parser):
         orch = Orchestrator("test_config.json", "request_configs.json")
         orch.parser = mock_parser
         return orch
 
 
 def test_orchestrator_initialization():
     """Test that Orchestrator initializes correctly with config paths."""
-    with patch("stress_framework.base.orchestrator.Parser") as mock_parser:
+    with patch("gradual.base.orchestrator.Parser") as mock_parser:
         orch = Orchestrator("test_config.json", "request_configs.json")
         mock_parser.assert_called_once_with("test_config.json", "request_configs.json")
         assert orch.test_config_file_path == "test_config.json"

tests/test_parser.py

@@ -5,7 +5,7 @@
 from gradual.configs.scenario import ScenarioConfig
 from gradual.configs.request import RequestConfig
 
-# filepath: src/stress_framework/configs/test_parser.py
+# filepath: src/gradual/configs/test_parser.py
 
 
 @pytest.fixture
@@ -54,7 +54,7 @@ def mock_request_yaml_data():
 
 @patch("builtins.open", new_callable=mock_open)
 @patch("yaml.safe_load")
-@patch("stress_framework.configs.parser.assert_not_empty")
+@patch("gradual.configs.parser.assert_not_empty")
 def test_read_configs(
     mock_assert_not_empty,
     mock_yaml_safe_load,
@@ -112,7 +112,7 @@ def test_read_configs(
 
 @patch("builtins.open", new_callable=mock_open)
 @patch("yaml.safe_load")
-@patch("stress_framework.configs.parser.assert_not_empty")
+@patch("gradual.configs.parser.assert_not_empty")
 def test_parser_multiple_phases_and_scenarios(
     mock_assert_not_empty, mock_yaml_safe_load, mock_open_file
 ):
@@ -203,7 +203,7 @@ def test_parser_multiple_phases_and_scenarios(
 
 @patch("builtins.open", new_callable=mock_open)
 @patch("yaml.safe_load")
-@patch("stress_framework.configs.parser.assert_not_empty")
+@patch("gradual.configs.parser.assert_not_empty")
 def test_parser_missing_required_fields_raises(
     mock_assert_not_empty, mock_yaml_safe_load, mock_open_file
 ):
@@ -216,7 +216,7 @@ def test_parser_missing_required_fields_raises(
 
 @patch("builtins.open", new_callable=mock_open)
 @patch("yaml.safe_load")
-@patch("stress_framework.configs.parser.assert_not_empty")
+@patch("gradual.configs.parser.assert_not_empty")
 def test_parser_invalid_ramp_up_type_raises(
     mock_assert_not_empty, mock_yaml_safe_load, mock_open_file
 ):
@@ -259,7 +259,7 @@ def test_parser_invalid_ramp_up_type_raises(
 
 @patch("builtins.open", new_callable=mock_open)
 @patch("yaml.safe_load")
-@patch("stress_framework.configs.parser.assert_not_empty")
+@patch("gradual.configs.parser.assert_not_empty")
 def test_parser_optional_fields(
     mock_assert_not_empty, mock_yaml_safe_load, mock_open_file
 ):

tests/test_phase.py

@@ -27,8 +27,8 @@ def mock_phase_config():
 def test_phase_initialization(mock_phase_config):
     """Test that Phase initializes correctly with config and run name."""
     with (
-        patch("stress_framework.runners.phase.Stats") as mock_stats,
-        patch("stress_framework.runners.phase.Runner") as mock_runner,
+        patch("gradual.runners.phase.Stats") as mock_stats,
+        patch("gradual.runners.phase.Runner") as mock_runner,
     ):
         phase = Phase(mock_phase_config, "test_run")
         mock_stats.assert_called_once_with(mock_phase_config, "test_run")
@@ -41,8 +41,8 @@ def test_phase_initialization(mock_phase_config):
 def test_phase_execution(mock_phase_config):
     """Test normal phase execution without timeout."""
     with (
-        patch("stress_framework.runners.phase.Stats"),
-        patch("stress_framework.runners.phase.Runner") as mock_runner,
+        patch("gradual.runners.phase.Stats"),
+        patch("gradual.runners.phase.Runner") as mock_runner,
     ):
         phase = Phase(mock_phase_config, "test_run")
         phase.runner.start_test = Mock()
@@ -65,8 +65,8 @@ def test_phase_execution(mock_phase_config):
 def test_phase_timeout(mock_phase_config):
     """Test phase execution with timeout."""
     with (
-        patch("stress_framework.runners.phase.Stats"),
-        patch("stress_framework.runners.phase.Runner") as mock_runner,
+        patch("gradual.runners.phase.Stats"),
+        patch("gradual.runners.phase.Runner") as mock_runner,
     ):
         phase = Phase(mock_phase_config, "test_run")
         phase.runner.stop_runner = Mock()
@@ -84,8 +84,8 @@ def test_phase_timeout(mock_phase_config):
 def test_stop_phase(mock_phase_config):
     """Test stopping a phase."""
     with (
-        patch("stress_framework.runners.phase.Stats"),
-        patch("stress_framework.runners.phase.Runner") as mock_runner,
+        patch("gradual.runners.phase.Stats"),
+        patch("gradual.runners.phase.Runner") as mock_runner,
     ):
         phase = Phase(mock_phase_config, "test_run")
         phase.runner.stop_runner = Mock()
@@ -96,8 +96,8 @@ def test_stop_phase(mock_phase_config):
 def test_phase_execution_error_handling(mock_phase_config):
     """Test error handling during phase execution."""
     with (
-        patch("stress_framework.runners.phase.Stats"),
-        patch("stress_framework.runners.phase.Runner") as mock_runner,
+        patch("gradual.runners.phase.Stats"),
+        patch("gradual.runners.phase.Runner") as mock_runner,
     ):
         phase = Phase(mock_phase_config, "test_run")
         phase.runner.stop_runner = Mock()

tests/test_runner.py

@@ -36,15 +36,15 @@ def mock_scenario_configs():
 
 @pytest.fixture
 def runner(mock_scenario_configs):
-    with patch("stress_framework.runners.runner.Scenario") as mock_scenario:
+    with patch("gradual.runners.runner.Scenario") as mock_scenario:
         mock_scenario.return_value = Mock()
         runner = Runner(mock_scenario_configs)
         return runner
 
 
 def test_runner_initialization(mock_scenario_configs):
     """Test that Runner initializes correctly with scenario configs."""
-    with patch("stress_framework.runners.runner.Scenario") as mock_scenario:
+    with patch("gradual.runners.runner.Scenario") as mock_scenario:
         mock_scenario.return_value = Mock()
         runner = Runner(mock_scenario_configs)
 
@@ -139,9 +139,9 @@ def get_next_request(self):
     with (
         patch("gevent.spawn") as mock_spawn,
         patch("gevent.wait") as mock_wait,
-        patch("stress_framework.runners.runner.Scenario") as mock_scenario,
-        patch("stress_framework.runners.scenario._Request", MockRequest),
-        patch("stress_framework.runners.scenario.RequestIterator", MockIterator),
+        patch("gradual.runners.runner.Scenario") as mock_scenario,
+        patch("gradual.runners.scenario._Request", MockRequest),
+        patch("gradual.runners.scenario.RequestIterator", MockIterator),
     ):
         # Create a new mock for each scenario
         mock_instances = []