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())