def test_and_or(self):
s = SQLValidator("height IS NOT NULL and height > 20")
self.assertTrue(s.valid)
s = SQLValidator("height IS NOT NULL or height > 20")
self.assertTrue(s.valid)
s = SQLValidator("height IS NOT NULL or height > 20 and height < 30")
self.assertTrue(s.valid)
def test_invalid_sql(self):
s = SQLValidator("drop table planet_osm_polygon")
self.assertFalse(s.valid)
self.assertEqual(s.errors,['SQL could not be parsed.'])
s = SQLValidator("(drop table planet_osm_polygon)")
self.assertFalse(s.valid)
self.assertEqual(s.errors,['SQL could not be parsed.'])
s = SQLValidator ("")
self.assertFalse(s.valid)
self.assertEqual(s.errors,['SQL could not be parsed.'])
s = SQLValidator("name = 'a name'; blah")
self.assertFalse(s.valid)
self.assertEqual(s.errors,['SQL could not be parsed.'])
def test_identifier_list(self):
s = SQLValidator("natural in ('water','cliff')")
self.assertTrue(s.valid)
def test_column_names(self):
s = SQLValidator("(admin IS NOT NULL and level > 4) AND height is not null")
self.assertTrue(s.valid)
self.assertEqual(s.column_names,['admin','level','height'])
def test_basic(self):
s = SQLValidator("name = 'a name'")
self.assertTrue(s.valid)
def test_colons_etc(self):
s = SQLValidator("addr:housenumber IS NOT NULL")
self.assertTrue(s.valid)
s = SQLValidator("admin_level IS NOT NULL")
self.assertTrue(s.valid)
def test_parens(self):
s = SQLValidator("(admin IS NOT NULL and level > 4)")
self.assertTrue(s.valid)
s = SQLValidator("(admin IS NOT NULL and level > 4) AND height is not null")
self.assertTrue(s.valid)
def test_not_null(self):
s = SQLValidator("height IS NOT NULL")
self.assertTrue(s.valid)
def test_float_value(self):
s = SQLValidator("height > 20")
self.assertTrue(s.valid)