diff --git a/.travis.yml b/.travis.yml index 655fab9d8e8e..77fac80b335a 100644 --- a/.travis.yml +++ b/.travis.yml @@ -57,6 +57,14 @@ matrix: services: - postgresql + - python: "pypy3.5" + env: TOX_ENV=pypy + + - python: "pypy3.5" + env: TOX_ENV=pypy-postgres TRIAL_FLAGS="-j 4" + services: + - postgresql + - # we only need to check for the newsfragment if it's a PR build if: type = pull_request python: 3.6 diff --git a/changelog.d/4174.misc b/changelog.d/4174.misc new file mode 100644 index 000000000000..d2cd79fc8bf7 --- /dev/null +++ b/changelog.d/4174.misc @@ -0,0 +1 @@ +Use psycopg2cffi consistently if running on PyPy diff --git a/synapse/server.py b/synapse/server.py index 9985687b95d9..0b2b64b7d51e 100644 --- a/synapse/server.py +++ b/synapse/server.py @@ -21,6 +21,7 @@ # Imports required for the default HomeServer() implementation import abc import logging +import platform from twisted.enterprise import adbapi from twisted.mail.smtp import sendmail @@ -367,6 +368,9 @@ def build_http_client(self): def build_db_pool(self): name = self.db_config["name"] + if (name == "psycopg2" and platform.python_implementation() == "PyPy"): + name = "psycopg2cffi" + return adbapi.ConnectionPool( name, cp_reactor=self.get_reactor(), diff --git a/synapse/storage/search.py b/synapse/storage/search.py index d5b5df93e6af..3c48fc110e75 100644 --- a/synapse/storage/search.py +++ b/synapse/storage/search.py @@ -186,7 +186,11 @@ def create_index(conn): # have an event_search_fts_idx; unfortunately postgres 9.4 # doesn't support CREATE INDEX IF EXISTS so we just catch the # exception and ignore it. - import psycopg2 + import platform + if platform.python_implementation() == "PyPy": + import psycopg2cffi as psycopg2 + else: + import psycopg2 try: c.execute( "CREATE INDEX CONCURRENTLY event_search_fts_idx" diff --git a/tests/utils.py b/tests/utils.py index 67ab916f30a3..3fc0614bcbe1 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -252,7 +252,11 @@ def setup_test_homeserver( else: # We need to do cleanup on PostgreSQL def cleanup(): - import psycopg2 + import platform + if platform.python_implementation() == "PyPy": + import psycopg2cffi as psycopg2 + else: + import psycopg2 # Close all the db pools hs.get_db_pool().close() diff --git a/tox.ini b/tox.ini index 03ddaeb0b71b..b3fc53ccb5cc 100644 --- a/tox.ini +++ b/tox.ini @@ -1,5 +1,5 @@ [tox] -envlist = packaging, py27, py36, pep8, check_isort +envlist = packaging, py27, py36, pypy, pep8, check_isort [base] deps = @@ -110,6 +110,17 @@ setenv = {[base]setenv} SYNAPSE_POSTGRES = 1 +[testenv:pypy] +usedevelop=true + +[testenv:pypy-postgres] +usedevelop=true +deps = + {[base]deps} + psycopg2cffi +setenv = + {[base]setenv} + SYNAPSE_POSTGRES = 1 [testenv:packaging] deps =