Fixing unit tests

apache · Sep 11, 2016 · 63c2db8 · 63c2db8
1 parent b8dd74f
commit 63c2db8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/caravel/views.py b/caravel/views.py
@@ -1607,7 +1607,7 @@ def csv(self, client_id):
         query = (
             db.session.query(models.Query)
             .filter_by(client_id=client_id)
-            .first()
+            .one()
         )
 
         if not self.database_access(query.database):

diff --git a/tests/core_tests.py b/tests/core_tests.py
@@ -372,7 +372,7 @@ def test_csv_endpoint(self):
             FROM ab_user
             WHERE first_name='admin'
         """
-        client_id = random.getrandbits(64)
+        client_id = "{}".format(random.getrandbits(64))[:10]
         self.run_sql(sql, 'admin', client_id)
 
         self.login('admin')