Match main tests to current CLI

Signed-off-by: Samuel Monson <[email protected]>

Author: sjmonson

tests/unit/test_main.py

@@ -19,10 +19,10 @@ def test_benchmark_run_with_backend_args():
             "--backend-args",
             '{"headers": {"Authorization": "Bearer my-token"}, "verify": false}',
             "--target",
-            "http://localhost:8000",
+            "http://localhost:9",
             "--data",
             "prompt_tokens=1,output_tokens=1",
-            "--rate-type",
+            "--profile",
             "constant",
             "--rate",
             "1",
@@ -36,7 +36,7 @@ def test_benchmark_run_with_backend_args():
     assert "Invalid header format" not in result.output
 
 
-@patch("guidellm.__main__.benchmark_with_scenario")
+@patch("guidellm.__main__.benchmark_generative_text")
 def test_cli_backend_args_header_removal(mock_benchmark_func, tmp_path: Path):
     """
     Tests that --backend-args from the CLI correctly overrides scenario
@@ -47,11 +47,11 @@ def test_cli_backend_args_header_removal(mock_benchmark_func, tmp_path: Path):
     # Create a scenario file with a header that should be overridden and removed
     scenario_content = {
         "backend_type": "openai_http",
-        "backend_args": {"headers": {"Authorization": "should-be-removed"}},
+        "backend_kwargs": {"headers": {"Authorization": "should-be-removed"}},
         "data": "prompt_tokens=10,output_tokens=10",
         "max_requests": 1,
         "target": "http://dummy-target",
-        "rate_type": "synchronous",
+        "profile": "synchronous",
         "processor": "gpt2",
     }
     with scenario_path.open("w") as f:
@@ -65,7 +65,7 @@ def test_cli_backend_args_header_removal(mock_benchmark_func, tmp_path: Path):
             "run",
             "--scenario",
             str(scenario_path),
-            "--backend-args",
+            "--backend-kwargs",
             '{"headers": {"Authorization": null, "Custom-Header": "Custom-Value"}}',
         ],
         catch_exceptions=False,
@@ -79,6 +79,6 @@ def test_cli_backend_args_header_removal(mock_benchmark_func, tmp_path: Path):
     scenario = call_args["scenario"]
 
     # Verify the backend_args were merged correctly
-    backend_args = scenario.backend_args
+    backend_args = scenario.backend_kwargs
     expected_headers = {"Authorization": None, "Custom-Header": "Custom-Value"}
     assert backend_args["headers"] == expected_headers