add tests

Author: codeskyblue

.travis.yml

@@ -15,6 +15,7 @@ allow_failure:
 
 install:
     - pip install -r requirements.txt
+    - pip install -r requirements-dev.txt
 
 before_script:
     - make prepare

requirements-dev.txt

@@ -0,0 +1,2 @@
+tornado>=5.0
+pytest-tornasync

tests/integration/test_tornado.py

@@ -0,0 +1,28 @@
+import os
+import sys
+from collections import namedtuple
+import pytest
+from rethinkdb import RethinkDB
+from rethinkdb.errors import ReqlRuntimeError
+
+Helper = namedtuple("Helper", "r connection")
+
+INTEGRATION_TEST_DB = 'integration_test'
+
+
+@pytest.mark.integration
+@pytest.mark.skipif(sys.version_info < (3, 6),
+                    reason="requires python3.6 or higher")
+async def test_tornado_connect():
+    """
+    Test the flow for 3.6 and up, async generators are
+    not supported in 3.5.
+    """
+
+    r = RethinkDB()
+    r.set_loop_type("tornado")
+
+    connection = await r.connect(os.getenv("REBIRTHDB_HOST"))
+    dbs = await r.db_list().run(connection)
+    assert isinstance(dbs, list)
+    await connection.close()