def test_valid_queries(self):
# Set protector to unsafe mode
protector = Protector(["prevent_delete"], [], False)
self.assertTrue(
protector.check(quote("select * from bla where x=y")).is_ok())
def get_protector():
global p
db_conf = {"redis": {"host":"", "port":"", "password":""}}
p = Protector({"query_no_aggregator": None}, [], [], db_conf, False)
def test_unknown_queries_non_safe_mode(self):
protector = Protector(["prevent_delete"], [], False)
self.assertFalse(protector.check("").is_ok())
self.assertFalse(protector.check("asdf").is_ok())