P

lib/yql/tests/test_query_placeholders.py

@@ -0,0 +1,77 @@
+"""Set of tests for the placeholder checking"""
+
+from unittest import TestCase
+
+from nose.tools import raises
+
+import yql
+
+
+class PublicTest(TestCase):
+    @raises(ValueError)
+    def test_empty_args_raises_valueerror(self):
+        y = yql.Public()
+        query = "SELECT * from foo where dog=@dog"
+        params = {}
+        y.execute(query, params)
+
+    @raises(ValueError)
+    def test_incorrect_args_raises_valueerror(self):
+        y = yql.Public()
+        query = "SELECT * from foo where dog=@dog"
+        params = {'test': 'fail'}
+        y.execute(query, params)
+
+    @raises(ValueError)
+    def test_params_raises_when_not_dict(self):
+        y = yql.Public()
+        query = "SELECT * from foo where dog=@dog"
+        params = ['test']
+        y.execute(query, params)
+
+    @raises(ValueError)
+    def test_unecessary_args_raises_valueerror(self):
+        y = yql.Public()
+        query = "SELECT * from foo where dog='test'"
+        params = {'test': 'fail'}
+        y.execute(query, params)
+
+    @raises(ValueError)
+    def test_incorrect_type_raises_valueerror(self):
+        y = yql.Public()
+        query = "SELECT * from foo where dog=@test"
+        params = ('fail')
+        y.execute(query, params)
+
+    def test_placeholder_regex_one(self):
+        y = yql.Public()
+        query = "SELECT * from foo where email='foo@foo.com'"
+        placeholders = y.get_placeholder_keys(query)
+        self.assertEqual(placeholders, [])
+
+    def test_placeholder_regex_two(self):
+        y = yql.Public()
+        query = "SELECT * from foo where email=@foo'"
+        placeholders = y.get_placeholder_keys(query)
+        self.assertEqual(placeholders, ['foo'])
+
+    def test_placeholder_regex_three(self):
+        y = yql.Public()
+        query = "SELECT * from foo where email=@foo and test=@bar'"
+        placeholders = y.get_placeholder_keys(query)
+        self.assertEqual(placeholders, ['foo', 'bar'])
+
+    def test_placeholder_regex_four(self):
+        y = yql.Public()
+        query = "SELECT * from foo where foo='bar' LIMIT @foo"
+        placeholders = y.get_placeholder_keys(query)
+        self.assertEqual(placeholders, ['foo'])
+
+    def test_placeholder_regex_five(self):
+        y = yql.Public()
+        query = """SELECT * from foo
+                    where foo='bar' LIMIT
+                    @foo"""
+        placeholders = y.get_placeholder_keys(query)
+        self.assertEqual(placeholders, ['foo'])
+