[core] add testserver

scripts/devops_tasks/end_coretestserver.py

@@ -0,0 +1,25 @@
+import os
+import signal
+import argparse
+
+def end_testserver(pid):
+
+    if os.name == 'nt':
+        os.kill(pid, signal.CTRL_C_EVENT)
+    else:
+        os.killpg(os.getpgid(pid), signal.SIGTERM)  # Send the signal to all the process groups
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(
+        description="Stop the testserver"
+    )
+    parser.add_argument(
+        "-p",
+        "--pid",
+        dest="pid",
+        help="The pid of the subprocess the testserver is running on",
+        required=True,
+    )
+
+    args = parser.parse_args()
+    end_testserver(int(args.pid))

scripts/devops_tasks/start_coretestserver.py

@@ -0,0 +1,14 @@
+import os
+import subprocess
+
+def start_testserver():
+    if os.name == 'nt': #On windows, subprocess creation works without being in the shell
+        os.environ["FLASK_APP"] = "coretestserver"
+        result = subprocess.Popen("flask run", env=dict(os.environ))
+    else:
+        result = subprocess.Popen("FLASK_APP=coretestserver flask run", shell=True, preexec_fn=os.setsid) #On linux, have to set shell=True
+    print('##vso[task.setvariable variable=FLASK_PID]{}'.format(result.pid))
+    print("This is used in the pipelines to set the FLASK_PID env var. If you want to stop this testserver, kill this PID.")
+
+if __name__ == "__main__":
+    start_testserver()

sdk/core/azure-core/dev_requirements.txt

@@ -7,3 +7,4 @@ opencensus-ext-threading
 mock; python_version < '3.3'
 -e ../../../tools/azure-sdk-tools
 -e ../../../tools/azure-devtools
+git+https://github.com/iscai-msft/core.testserver#subdirectory=coretestserver

sdk/core/azure-core/pytest.ini

@@ -0,0 +1,3 @@
+[pytest]
+usefixtures=testserver
+xfail_strict=true

sdk/core/azure-core/samples/conftest.py

@@ -23,8 +23,17 @@
 # IN THE SOFTWARE.
 #
 # --------------------------------------------------------------------------
+import pytest
 import sys
 
+@pytest.fixture(scope="session")
+def testserver():
+    # dummy testserver for now
+    # pytest.ini needs to be in root bc we run pytest on the pipelines with just "pytest"
+    # because of this, samples conftest needs its own def of testserver.
+    # plan to change the samples tests to use testserver as well, so it's not always going to be a gross dummy fixture
+    yield
+
 # Ignore collection of async tests for Python 2
 collect_ignore = []
 if sys.version_info < (3, 5):

sdk/core/azure-core/tests/async_tests/test_testserver_async.py

@@ -0,0 +1,37 @@
+# --------------------------------------------------------------------------
+#
+# Copyright (c) Microsoft Corporation. All rights reserved.
+#
+# The MIT License (MIT)
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the ""Software""), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+#
+# --------------------------------------------------------------------------
+import pytest
+from azure.core.pipeline.transport import HttpRequest, AioHttpTransport
+"""This file does a simple call to the testserver to make sure we can use the testserver"""
+
+@pytest.mark.asyncio
+async def test_smoke():
+    request = HttpRequest(method="GET", url="http://localhost:5000/basic/string")
+    async with AioHttpTransport() as sender:
+        response = await sender.send(request)
+        response.raise_for_status()
+        await response.load_body()
+        assert response.text() == "Hello, world!"

sdk/core/azure-core/tests/conftest.py

@@ -23,14 +23,40 @@
 # IN THE SOFTWARE.
 #
 # --------------------------------------------------------------------------
+import pytest
+import signal
+import os
+import subprocess
 import sys
 
+def start_testserver():
+    if os.name == 'nt': #On windows, subprocess creation works without being in the shell
+        os.environ["FLASK_APP"] = "coretestserver"
+        return subprocess.Popen("flask run", env=dict(os.environ))
+
+    return subprocess.Popen("FLASK_APP=coretestserver flask run", shell=True, preexec_fn=os.setsid) #On linux, have to set shell=True
+
+def terminate_testserver(process):
+    if os.name == 'nt':
+        process.kill()
+    else:
+        os.killpg(os.getpgid(process.pid), signal.SIGTERM)  # Send the signal to all the process groups
+
+@pytest.fixture(scope="session")
+def testserver():
+    """Start the Autorest testserver."""
+    if not os.environ.get("FLASK_PID"):
+        server = start_testserver()
+        yield
+        terminate_testserver(server)
+    else:
+        yield
+
 # Ignore collection of async tests for Python 2
 collect_ignore = []
 if sys.version_info < (3, 5):
     collect_ignore.append("async_tests")
 
-
 # If opencensus is loadable while doing these tests, register an empty tracer to avoid this:
 # https://github.com/census-instrumentation/opencensus-python/issues/442
 try:

sdk/core/azure-core/tests/test_testserver.py

@@ -0,0 +1,34 @@
+# --------------------------------------------------------------------------
+#
+# Copyright (c) Microsoft Corporation. All rights reserved.
+#
+# The MIT License (MIT)
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the ""Software""), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+#
+# --------------------------------------------------------------------------
+from azure.core.pipeline.transport import HttpRequest, RequestsTransport
+"""This file does a simple call to the testserver to make sure we can use the testserver"""
+
+def test_smoke():
+    request = HttpRequest(method="GET", url="http://localhost:5000/basic/string")
+    with RequestsTransport() as sender:
+        response = sender.send(request)
+        response.raise_for_status()
+        assert response.text() == "Hello, world!"

sdk/core/ci.yml

@@ -44,6 +44,18 @@ extends:
       safeName: azurecorecoretracingtelemetry
     - name: azure-common
       safeName: azurecommon
+    BeforeTestSteps:
+      - task: PythonScript@0
+        inputs:
+          scriptSource: 'scripts/devops_tasks/start_coretestserver.py'
+        displayName: "Start CoreTestServer"
+    AfterTestSteps:
+      - task: PythonScript@0
+        inputs:
+          scriptSource: 'scripts/devops_tasks/end_coretestserver.py'
+          arguments: >-
+            -p $(FLASK_PID)
+        displayName: "Shut down CoreTestServer"
     CondaArtifacts:
     - name: azure-core
       meta_source: conda-recipe/meta.yaml