Cherry pick performance/stress tests from 1.7

These tests were lost somewhere in 1.7 branching/remerging from 1.7 to
4.0.

Author: Peter Wilhelmsson

tests/env.py

@@ -0,0 +1,37 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+
+# Copyright (c) 2002-2019 "Neo4j,"
+# Neo4j Sweden AB [http://neo4j.com]
+#
+# This file is part of Neo4j.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from os import getenv
+
+
+# Full path of a server package to be used for integration testing
+NEO4J_SERVER_PACKAGE = getenv("NEO4J_SERVER_PACKAGE")
+
+# An existing remote server at this URI
+NEO4J_SERVER_URI = getenv("NEO4J_URI")
+
+# Name of a user for the currently running server
+NEO4J_USER = getenv("NEO4J_USER")
+
+# Password for the currently running server
+NEO4J_PASSWORD = getenv("NEO4J_PASSWORD")
+
+NEOCTRL_ARGS = getenv("NEOCTRL_ARGS", "3.4.1")

tests/performance/test_results.py

@@ -0,0 +1,74 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+
+# Copyright (c) 2002-2019 "Neo4j,"
+# Neo4j Sweden AB [http://neo4j.com]
+#
+# This file is part of Neo4j.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from itertools import product
+
+from pytest import mark
+
+from neo4j import GraphDatabase
+from .tools import RemoteGraphDatabaseServer
+
+
+class ReadWorkload(object):
+
+    server = None
+    driver = None
+
+    @classmethod
+    def setup_class(cls):
+        cls.server = server = RemoteGraphDatabaseServer()
+        server.start()
+        cls.driver = GraphDatabase.driver(server.server_uri, auth=server.auth_token, encrypted=server.encrypted)
+
+    @classmethod
+    def teardown_class(cls):
+        cls.driver.close()
+        cls.server.stop()
+
+    def work(self, *units_of_work):
+        def runner():
+            with self.driver.session() as session:
+                for unit_of_work in units_of_work:
+                    session.read_transaction(unit_of_work)
+        return runner
+
+
+class TestReadWorkload(ReadWorkload):
+
+    @staticmethod
+    def uow(record_count, record_width, value):
+
+        def _(tx):
+            s = "UNWIND range(1, $record_count) AS _ RETURN {}".format(
+                ", ".join("$x AS x{}".format(i) for i in range(record_width)))
+            p = {"record_count": record_count, "x": value}
+            for record in tx.run(s, p):
+                assert all(x == value for x in record.values())
+
+        return _
+
+    @mark.parametrize("record_count,record_width,value", product(
+        [1, 1000],  # record count
+        [1, 10],    # record width
+        [1, u'hello, world'],        # value
+    ))
+    def test_1x1(self, benchmark, record_count, record_width, value):
+        benchmark(self.work(self.uow(record_count, record_width, value)))

tests/performance/tools.py

@@ -0,0 +1,80 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+
+# Copyright (c) 2002-2019 "Neo4j,"
+# Neo4j Sweden AB [http://neo4j.com]
+#
+# This file is part of Neo4j.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from test.integration.tools import IntegrationTestCase
+
+from os import makedirs, remove
+from os.path import basename, dirname, join as path_join, realpath, isfile, expanduser
+import platform
+from unittest import TestCase, SkipTest
+from shutil import copyfile
+from sys import exit, stderr
+
+try:
+    from urllib.request import urlretrieve
+except ImportError:
+    from urllib import urlretrieve
+
+from boltkit.controller import WindowsController, UnixController
+
+from neo4j import GraphDatabase
+from neo4j.exceptions import AuthError
+
+from test.env import NEO4J_USER, NEO4J_PASSWORD, NEO4J_SERVER_URI
+
+
+def is_listening(address):
+    from socket import create_connection
+    try:
+        s = create_connection(address)
+    except IOError:
+        return False
+    else:
+        s.close()
+        return True
+
+
+class RemoteGraphDatabaseServer(object):
+    server_uri = NEO4J_SERVER_URI or "bolt://localhost:7687"
+    auth_token = (NEO4J_USER or "neo4j", NEO4J_PASSWORD)
+    encrypted = NEO4J_SERVER_URI is not None
+
+    def __enter__(self):
+        self.start()
+        return self
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        self.stop()
+
+    @classmethod
+    def start(cls):
+        with GraphDatabase.driver(cls.server_uri, auth=cls.auth_token, encrypted=cls.encrypted) as driver:
+            try:
+                with driver.session():
+                    print("Using existing remote server {}\n".format(cls.server_uri))
+                    return
+            except AuthError as error:
+                raise RuntimeError("Failed to authenticate (%s)" % error)
+        raise SkipTest("No remote Neo4j server available for %s" % cls.__name__)
+
+    @classmethod
+    def stop(cls):
+        pass

tox-performance.ini

@@ -0,0 +1,17 @@
+[tox]
+envlist =
+    py34
+    py35
+    py36
+
+[testenv]
+passenv =
+    NEO4J_USER
+    NEO4J_PASSWORD
+    NEO4J_URI
+commands =
+    python setup.py develop
+    pip install --upgrade -r {toxinidir}/test/requirements.txt
+    coverage erase
+    coverage run -m pytest -v {posargs} test/performance
+    coverage report

tox.ini

@@ -16,6 +16,7 @@ passenv =
     TEAMCITY_USER
     TEAMCITY_PASSWORD
     JAVA_HOME
+    NEO4J_URI
     AWS_ACCESS_KEY_ID
     AWS_SECRET_ACCESS_KEY
 deps =