test(integration): add --language-extensions to invoke/start-api commands and fix deploy test

Author: bnusunny

tests/integration/deploy/test_deploy_command.py

@@ -20,6 +20,7 @@
     RUN_BY_CANARY,
     SKIP_LMI_TESTS,
     UpdatableSARTemplate,
+    get_sam_command,
 )
 
 # Deploy tests require credentials and CI/CD will only add credentials to the env if the PR is from the same repo.
@@ -1719,7 +1720,8 @@ def test_deploy_with_language_extensions(self):
 
         # Build first — sam deploy of a raw ForEach template requires a build step
         build_dir = tempfile.mkdtemp()
-        build_command_list = self.base_command_list + [
+        build_command_list = [
+            get_sam_command(),
             "build",
             "-t",
             str(template),

tests/integration/local/invoke/test_invoke_language_extensions.py

@@ -75,7 +75,7 @@ def test_invoke_foreach_expanded_function(self):
             function_to_invoke="AlphaFunction",
             template_path=str(self.built_template_path),
             no_event=True,
-        )
+        ) + ["--language-extensions"]
         stdout, stderr, invoke_exit_code = self.run_command(invoke_command, cwd=self.working_dir)
 
         self.assertEqual(invoke_exit_code, 0, f"invoke failed: stdout={stdout}, stderr={stderr}")

tests/integration/local/start_api/test_start_api_language_extensions.py

@@ -27,6 +27,12 @@ def setUpClass(cls):
         cls.build_overrides = {"Runtime": python_version}
         super().setUpClass()
 
+    @classmethod
+    def start_api(cls):
+        command = get_sam_command()
+        cls.command_list = [command, "local", "start-api", "-t", cls.template, "--language-extensions"]
+        super().start_api()
+
     @classmethod
     def build(cls):
         command = get_sam_command()