Upgrade sqlparse to 0.2.4

Add test case for /api/queries/format
This commit is contained in:
ariarijp
2018-02-23 01:24:34 +09:00
parent c65b637bc4
commit f5d4ca85d8
2 changed files with 17 additions and 1 deletions

View File

@@ -183,3 +183,19 @@ class TestQueryForkResourcePost(BaseTestCase):
rv = self.make_request('post', '/api/queries/{}/fork'.format(query.id))
self.assertEqual(rv.status_code, 403)
class TestFormatSQLQueryAPI(BaseTestCase):
def test_format_sql_query(self):
admin = self.factory.create_admin()
query = 'select a,b,c FROM foobar Where x=1 and y=2;'
expected = """SELECT a,
b,
c
FROM foobar
WHERE x=1
AND y=2;"""
rv = self.make_request('post', '/api/queries/format', user=admin, data={'query': query})
self.assertEqual(rv.json['query'], expected)