feat(fal-file-no-scale): add tests

projects/fal/tests/cli/test_deploy.py

@@ -34,6 +34,7 @@ def mock_args(
     app_name: Optional[str] = None,
     auth: Optional[str] = None,
     strategy: Optional[str] = None,
+    no_scale: bool = False,
 ):
     args = MagicMock()
 
@@ -42,6 +43,8 @@ def mock_args(
     args.auth = auth
     args.strategy = strategy
 
+    args.no_scale = no_scale
+
     return args
 
 
@@ -67,6 +70,7 @@ def test_deploy_with_toml_success(
         args,
         "shared",
         "rolling",
+        False,
     )
 
 
@@ -92,6 +96,7 @@ def test_deploy_with_toml_no_auth(
         args,
         "private",
         "recreate",
+        False,
     )
 
 
@@ -185,6 +190,7 @@ def test_deploy_with_toml_deployment_strategy(
         args,
         "shared",
         "rolling",
+        False,
     )
 
 
@@ -208,6 +214,7 @@ def test_deploy_with_toml_default_deployment_strategy(
         args,
         "private",
         "recreate",
+        False,
     )
 
 
@@ -229,6 +236,7 @@ def test_deploy_with_cli_auth(
         args,
         "shared",
         None,
+        False,
     )
 
 
@@ -250,4 +258,49 @@ def test_deploy_with_cli_deployment_strategy(
         args,
         None,
         "rolling",
+        False,
+    )
+
+
+@patch("fal.cli._utils.find_pyproject_toml", return_value="pyproject.toml")
+@patch("fal.cli._utils.parse_pyproject_toml")
+@patch("fal.cli.deploy._deploy_from_reference")
+def test_deploy_with_cli_no_scale(
+    mock_deploy_ref, mock_parse_toml, mock_find_toml, mock_parse_pyproject_toml
+):
+    mock_parse_toml.return_value = mock_parse_pyproject_toml
+
+    args = mock_args(app_ref=("src/my_app/inference.py", "MyApp"), no_scale=True)
+
+    _deploy(args)
+
+    mock_deploy_ref.assert_called_once_with(
+        ("src/my_app/inference.py", "MyApp"),
+        None,
+        args,
+        None,
+        None,
+        True,
+    )
+
+
+@patch("fal.cli._utils.find_pyproject_toml", return_value="pyproject.toml")
+@patch("fal.cli._utils.parse_pyproject_toml")
+@patch("fal.cli.deploy._deploy_from_reference")
+def test_deploy_with_cli_scale(
+    mock_deploy_ref, mock_parse_toml, mock_find_toml, mock_parse_pyproject_toml
+):
+    mock_parse_toml.return_value = mock_parse_pyproject_toml
+
+    args = mock_args(app_ref=("src/my_app/inference.py", "MyApp"))
+
+    _deploy(args)
+
+    mock_deploy_ref.assert_called_once_with(
+        ("src/my_app/inference.py", "MyApp"),
+        None,
+        args,
+        None,
+        None,
+        False,
     )