add support for local tests in docker

Author: collerek

.gitignore

@@ -1,4 +1,5 @@
 .vscode/
+.idea/
 *.pyc
 test.db
 .coverage

scripts/check

@@ -1,9 +1,23 @@
 #!/bin/sh -e
 
 export PREFIX=""
-if [ -d 'venv' ] ; then
+UNAME=$(uname)
+case "$UNAME" in
+"Linux") export SYSTEM="Linux" ;;
+"Darwin") export SYSTEM="Linux" ;;
+CYGWIN*) export SYSTEM="Windows" ;;
+MINGW*) export SYSTEM="Windows" ;;
+*) export SYSTEM="Linux" ;;
+esac
+
+if [ -d 'venv' ]; then
+  if [ $SYSTEM = "Linux" ] ; then
     export PREFIX="venv/bin/"
+  else
+    export PREFIX="venv/Scripts/"
+  fi
 fi
+
 export SOURCE_FILES="databases tests"
 
 set -x

scripts/coverage

@@ -1,8 +1,21 @@
 #!/bin/sh -e
 
 export PREFIX=""
-if [ -d 'venv' ] ; then
+UNAME=$(uname)
+case "$UNAME" in
+"Linux") export SYSTEM="Linux" ;;
+"Darwin") export SYSTEM="Linux" ;;
+CYGWIN*) export SYSTEM="Windows" ;;
+MINGW*) export SYSTEM="Windows" ;;
+*) export SYSTEM="Linux" ;;
+esac
+
+if [ -d 'venv' ]; then
+  if [ $SYSTEM = "Linux" ] ; then
     export PREFIX="venv/bin/"
+  else
+    export PREFIX="venv/Scripts/"
+  fi
 fi
 
 set -x

scripts/docker-compose.yml

@@ -0,0 +1,20 @@
+version: '2.1'
+services:
+  postgres:
+    image: postgres:10.8
+    environment:
+      POSTGRES_USER: username
+      POSTGRES_PASSWORD: password
+      POSTGRES_DB: testsuite
+    ports:
+      - 5432:5432
+
+  mysql:
+    image: mysql:5.7
+    environment:
+      MYSQL_USER: username
+      MYSQL_PASSWORD: password
+      MYSQL_ROOT_PASSWORD: password
+      MYSQL_DATABASE: testsuite
+    ports:
+      - 3306:3306

scripts/lint

@@ -1,8 +1,21 @@
 #!/bin/sh -e
 
 export PREFIX=""
-if [ -d 'venv' ] ; then
+UNAME=$(uname)
+case "$UNAME" in
+"Linux") export SYSTEM="Linux" ;;
+"Darwin") export SYSTEM="Linux" ;;
+CYGWIN*) export SYSTEM="Windows" ;;
+MINGW*) export SYSTEM="Windows" ;;
+*) export SYSTEM="Linux" ;;
+esac
+
+if [ -d 'venv' ]; then
+  if [ $SYSTEM = "Linux" ]; then
     export PREFIX="venv/bin/"
+  else
+    export PREFIX="venv/Scripts/"
+  fi
 fi
 
 set -x

scripts/test

@@ -1,23 +1,51 @@
 #!/bin/sh
 
 export PREFIX=""
-if [ -d 'venv' ] ; then
+UNAME=$(uname)
+case "$UNAME" in
+"Linux") export SYSTEM="Linux" ;;
+"Darwin") export SYSTEM="Linux" ;;
+CYGWIN*) export SYSTEM="Windows" ;;
+MINGW*) export SYSTEM="Windows" ;;
+*) export SYSTEM="Linux" ;;
+esac
+
+if [ -d 'venv' ]; then
+  if [ $SYSTEM = "Linux" ]; then
     export PREFIX="venv/bin/"
+  else
+    export PREFIX="venv/Scripts/"
+  fi
+fi
+
+if [ -z "$TEST_DATABASE_URLS" ]; then
+  echo "Variable TEST_DATABASE_URLS must be set."
+  exit 1
 fi
 
-if [ -z "$TEST_DATABASE_URLS" ] ; then
-    echo "Variable TEST_DATABASE_URLS must be set."
-    exit 1
+if [ -n "${TEST_IN_DOCKER+x}" ]; then
+  docker-compose -f ./scripts/docker-compose.yml up -d
+  export TEST_DATABASE_URLS="sqlite:///test.db,
+  sqlite+aiosqlite:///test.db,
+  mysql+aiomysql://username:password@localhost:3306/testsuite,
+  mysql+asyncmy://username:password@localhost:3306/testsuite,
+  postgresql+aiopg://username:[email protected]:5432/testsuite,
+  postgresql+asyncpg://username:password@localhost:5432/testsuite
+  "
 fi
 
 set -ex
 
 if [ -z $GITHUB_ACTIONS ]; then
-    scripts/check
+  scripts/check
 fi
 
 ${PREFIX}coverage run -m pytest $@
 
 if [ -z $GITHUB_ACTIONS ]; then
-    scripts/coverage
+  scripts/coverage
+fi
+
+if [ -n "${TEST_IN_DOCKER+x}" ]; then
+  docker-compose -f ./scripts/docker-compose.yml down
 fi

tests/test_databases.py

@@ -12,6 +12,9 @@
 
 from databases import Database, DatabaseURL
 
+if sys.version_info >= (3, 8) and sys.platform.lower().startswith("win"):
+    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
+
 assert "TEST_DATABASE_URLS" in os.environ, "TEST_DATABASE_URLS is not set."
 
 DATABASE_URLS = [url.strip() for url in os.environ["TEST_DATABASE_URLS"].split(",")]
@@ -23,7 +26,7 @@ def mysql_versions(wrapped_func):
     """
 
     @functools.wraps(wrapped_func)
-    def check(*args, **kwargs):
+    def check(*args, **kwargs):  # pragma: no cover
         url = DatabaseURL(kwargs["database_url"])
         if url.scheme in ["mysql", "mysql+aiomysql"] and sys.version_info >= (3, 10):
             pytest.skip("aiomysql supports python 3.9 and lower")