Fix tests: perform integration tests against latest Presto release

prestodb · Jul 18, 2019 · 54e7c03 · 54e7c03
1 parent 10e8ffd
commit 54e7c03
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/integration_tests/fixtures.py b/integration_tests/fixtures.py
@@ -45,7 +45,7 @@ def get_latest_release():
 
 # Temporary fix because can't start 0.203
 # Use get_latest_release() instead
-PRESTO_VERSION = os.environ.get("PRESTO_VERSION") or "0.202"
+PRESTO_VERSION = os.environ.get("PRESTO_VERSION") or get_latest_release()
 PRESTO_HOST = "127.0.0.1"
 PRESTO_PORT = 8080
 

diff --git a/integration_tests/test_dbapi.py b/integration_tests/test_dbapi.py
@@ -48,7 +48,7 @@ def test_select_query(presto_connection):
     assert len(rows) > 0
     row = rows[0]
     assert row[0] == "test"
-    assert row[2] == fixtures.PRESTO_VERSION
+    assert row[2].split("-")[0] == fixtures.PRESTO_VERSION
     columns = dict([desc[:2] for desc in cur.description])
     assert columns["node_id"] == "varchar"
     assert columns["http_uri"] == "varchar"