speed test harness for cached builds of openssl

inorton · inorton · commit b9e9780dae55 · 2016-05-24T23:37:02.000+01:00
diff --git a/speedtest/fixtures.py b/speedtest/fixtures.py
@@ -0,0 +1,21 @@
+"""
+Test fixtures
+"""
+
+import os
+import pytest
+from utils import find_visual_studio, get_vc_envs
+
+
+@pytest.fixture()
+def clcache_envs():
+    """
+    return a dict of envs suitable for clcache to work with
+    :return:
+    """
+    vcdir, _ = find_visual_studio()
+    envs = get_vc_envs()
+    cachedir = os.path.join("clcache_cachedir")
+    envs["CLCACHE_DIR"] = cachedir
+    envs["CLCACHE_CL"] = os.path.join(vcdir, "cl.exe")
+    return envs
diff --git a/speedtest/test_performance_curl.py b/speedtest/test_performance_curl.py
@@ -0,0 +1,107 @@
+#!/usr/bin/python
+"""
+A py.test test that attempts to build libcurl and benchmark the effect of clcache
+"""
+
+import sys
+import os
+import pytest
+import urllib
+import zipfile
+import subprocess
+
+from utils import block_message, retry_delete, common_module_setup, check_program, get_vc_envs, download_file
+from utils import DISTDIR, RETAIN_CACHE, EMPTY_CACHE
+
+from fixtures import clcache_envs
+
+CURL_URL = "https://github.com/curl/curl/archive/curl-7_49_0.zip"
+CURL_ZIP = "curl-7_49_0.zip"
+SOURCES = None
+
+
+def clean_build():
+    """
+    Unpack the curl source, possibly deleting the previous one
+    :return:
+    """
+    with zipfile.ZipFile(CURL_ZIP, "r") as unzip:
+        folder = unzip.namelist()[0]
+        if os.path.exists(folder):
+            with block_message("delete old curl folder"):
+                retry_delete(folder)
+
+        with block_message("unzip curl"):
+            unzip.extractall()
+
+        global SOURCES
+        SOURCES = folder.rstrip("/")
+
+
+def setup_function(request):
+    """
+    Ensure a clean build tree before each test
+    :return:
+    """
+    os.chdir(DISTDIR)
+    clean_build()
+
+
+def setup_module():
+    """
+    Check that our exe has been built.
+    :return:
+    """
+    common_module_setup()
+    download_file(CURL_URL, CURL_ZIP)
+
+
+def build_curl(addpath=None, envs=get_vc_envs(), pdbs=False):
+    """
+    Build curl with nmake
+    :return:
+    """
+    check_program(envs, "nmake.exe")
+    workdir = os.path.join(SOURCES, "winbuild")
+
+    if addpath is not None:
+        envs["PATH"] = addpath + os.pathsep + envs["PATH"]
+
+    gen_pdbs = "GEN_PDB=no"
+    if pdbs:
+        gen_pdbs = "GEN_PDB=yes"
+
+    with block_message("build curl"):
+        subprocess.check_output(["nmake", "-f", "Makefile.vc", "mode=static",
+                                 "ENABLE_SSPI=no",
+                                 "ENABLE_IPV6=no",
+                                 "ENABLE_IDN=no",
+                                 gen_pdbs],
+                                shell=True,
+                                cwd=workdir,
+                                env=envs)
+
+
+def test_build_nocache():
+    """
+    Build curl with no caching
+    :return:
+    """
+    build_curl()
+
+
+@pytest.mark.parametrize("cache_setting", [EMPTY_CACHE, RETAIN_CACHE])
+def test_build_withclcache(clcache_envs, cache_setting):
+    """
+    Time a curl build with a cold cache
+    :param clcache_envs: clcache environment vars fixture
+    :param cache_setting: if True, delete the cache from disk
+    :return:
+    """
+    if cache_setting == EMPTY_CACHE:
+        retry_delete(clcache_envs["CLCACHE_DIR"])
+    build_curl(DISTDIR, clcache_envs)
+
+
+if __name__ == "__main__":
+    pytest.main(sys.argv[1:])
diff --git a/speedtest/test_performance_openssl.py b/speedtest/test_performance_openssl.py
@@ -0,0 +1,149 @@
+#!/usr/bin/python
+"""
+A py.test test that attempts to build openssl and benchmark the effect of clcache
+"""
+import sys
+import os
+import pytest
+import zipfile
+import subprocess
+
+from utils import block_message, retry_delete, common_module_setup, check_program, get_vc_envs, download_file
+from utils import DISTDIR, RETAIN_CACHE, EMPTY_CACHE
+
+from fixtures import clcache_envs
+
+OPENSSL_ZIP = "OpenSSL_1_0_2-stable.zip"
+OPENSSL_URL = "https://codeload.github.com/openssl/openssl/zip/OpenSSL_1_0_2-stable"
+SOURCES = None
+
+
+def clean_build():
+    """
+    Unpack the openssl source, possibly deleting the previous one
+    :return:
+    """
+    with zipfile.ZipFile(OPENSSL_ZIP, "r") as unzip:
+        folder = unzip.namelist()[0]
+        if os.path.exists(folder):
+            with block_message("delete old openssl folder"):
+                retry_delete(folder)
+
+        with block_message("unzip openssl"):
+            unzip.extractall()
+
+        global SOURCES
+        SOURCES = folder.rstrip("/")
+
+
+def configure_openssl(envs):
+    """
+    Run the configure steps (requires perl)
+    :param envs:
+    :return:
+    """
+    check_program(envs, "nmake.exe")
+    check_program(envs, "perl.exe")
+
+    with block_message("configure openssl"):
+        subprocess.check_call(["perl",
+                               "Configure", "VC-WIN32", "no-asm", "--prefix=c:\openssl"],
+                              env=envs,
+                              cwd=SOURCES)
+
+    with block_message("generate makefiles"):
+        subprocess.check_call([os.path.join("ms", "do_ms.bat")],
+                              shell=True,
+                              env=envs,
+                              cwd=SOURCES)
+
+
+def setup_function(request):
+    """
+    Ensure a clean build tree before each test
+    :return:
+    """
+    os.chdir(DISTDIR)
+    clean_build()
+    configure_openssl(get_vc_envs())
+
+
+def setup_module():
+    """
+    Check that our exe has been built.
+    :return:
+    """
+    common_module_setup()
+    download_file(OPENSSL_URL, OPENSSL_ZIP)
+
+
+def replace_wipe_cflags(filename):
+    """
+    Open the nmake file given and turn off PDB generation for .obj files
+    :param filename:
+    :return:
+    """
+    lines = []
+    with open(filename, "rb") as makefile:
+        for line in makefile.readlines():
+            if line.startswith("APP_CFLAG="):
+                lines.append("APP_CFLAG=")
+            elif line.startswith("LIB_CFLAG="):
+                lines.append("LIB_CFLAG=/Zl")
+            else:
+                lines.append(line.rstrip())
+
+    with open(filename, "wb") as makefile:
+        for line in lines:
+            makefile.write(line + "\r\n")
+
+
+def build_openssl(addpath=None, envs=get_vc_envs(), pdbs=False):
+    """
+    Build openssl, optionally prefixing addpath to $PATH
+    :param addpath:
+    :param envs: env var dict to use
+    :param pdbs: if False, turn off pdb generation in the makefile
+    :return:
+    """
+    nmakefile = os.path.join("ms", "nt.mak")
+    if not pdbs:
+        replace_wipe_cflags(os.path.join(SOURCES, nmakefile))
+
+    if addpath is not None:
+        envs["PATH"] = addpath + os.pathsep + envs["PATH"]
+
+    try:
+        with block_message("running nmake"):
+            subprocess.check_output(["nmake", "-f", nmakefile],
+                                    shell=True,
+                                    env=envs,
+                                    cwd=SOURCES)
+    except subprocess.CalledProcessError as cpe:
+        print cpe.output
+        raise
+
+
+def test_build_nocache():
+    """
+    Time an openssl build with no caching involved at all
+    :return:
+    """
+    build_openssl()
+
+
+@pytest.mark.parametrize("cache_setting", [EMPTY_CACHE, RETAIN_CACHE])
+def test_build_withclcache(clcache_envs, cache_setting):
+    """
+    Time an openssl build with a cold cache
+    :param clcache_envs: clcache environment vars fixture
+    :param cache_setting: if True, delete the cache from disk
+    :return:
+    """
+    if cache_setting == EMPTY_CACHE:
+        retry_delete(clcache_envs["CLCACHE_DIR"])
+    build_openssl(DISTDIR, clcache_envs)
+
+
+if __name__ == "__main__":
+    pytest.main(sys.argv[1:])
diff --git a/speedtest/utils.py b/speedtest/utils.py
