issue-4952 fix (GoogleCloudPlatform#11454)

irudykss · web-flow · commit 08ef000a07e7 · 2024-04-23T18:47:11.000+03:00
diff --git a/functions/helloworld/sample_http_test_integration.py b/functions/helloworld/sample_http_test_integration.py
@@ -14,42 +14,19 @@
 
 # [START functions_http_integration_test]
 import os
-import subprocess
 import uuid
 
-import requests
-from requests.packages.urllib3.util.retry import Retry
+from functions_framework import create_app
 
 
 def test_args():
     name = str(uuid.uuid4())
-    port = os.getenv(
-        "PORT", 8080
-    )  # Each functions framework instance needs a unique port
-
-    process = subprocess.Popen(
-        ["functions-framework", "--target", "hello_http", "--port", str(port)],
-        cwd=os.path.dirname(__file__),
-        stdout=subprocess.PIPE,
-    )
-
-    # Send HTTP request simulating Pub/Sub message
-    # (GCF translates Pub/Sub messages to HTTP requests internally)
-    BASE_URL = f"http://localhost:{port}"
-
-    retry_policy = Retry(total=6, backoff_factor=1)
-    retry_adapter = requests.adapters.HTTPAdapter(max_retries=retry_policy)
-
-    session = requests.Session()
-    session.mount(BASE_URL, retry_adapter)
-
-    name = str(uuid.uuid4())
-    res = session.post(BASE_URL, json={"name": name})
-    assert res.text == f"Hello {name}!"
-
-    # Stop the functions framework process
-    process.kill()
-    process.wait()
-
+    target = "hello_http"
+    source = os.path.join(os.path.dirname(__file__), "main.py")
+    client = create_app(target, source).test_client()
+    resp = client.post("/my_path", json={"name": name})
+    assert resp.status_code == 200
+    expected_response = f"Hello {name}!"
+    assert resp.data.decode("utf-8") == expected_response
 
 # [END functions_http_integration_test]
