def test_handle_extended_like(self):
sql = "SELECT * FROM table WHERE param1 = ?"
params = ["param1"]
db = DB()
new_sql, new_params = db.handle_extended_like(sql, params)
self.assertEqual(1, len(new_params))
self.assertEqual("param1", new_params[0])
self.assertEqual("SELECT * FROM table WHERE param1 = ?", new_sql)
def test_handle_extended_like2(self):
sql = "SELECT * FROM table WHERE param0 = ? AND param1 <EXTENDED_LIKE=1> ? AND param2 = ? AND param3 <EXTENDED_LIKE=3> ? AND param4 = ?"
params = [
"param0", "search1 search2 search3", "param2", "param31 param32",
"param4"
]
db = DB()
new_sql, new_params = db.handle_extended_like(sql, params)
self.assertEqual(8, len(new_params))
self.assertEqual("param0", new_params[0])
self.assertEqual("%search1%", new_params[1])
self.assertEqual("%search2%", new_params[2])
self.assertEqual("%search3%", new_params[3])
self.assertEqual("param2", new_params[4])
self.assertEqual("%param31%", new_params[5])
self.assertEqual("%param32%", new_params[6])
self.assertEqual("param4", new_params[7])
self.assertEqual(
"SELECT * FROM table WHERE param0 = ? "
"AND (param1 LIKE ? AND param1 LIKE ? AND param1 LIKE ?) "
"AND param2 = ? "
"AND (param3 LIKE ? AND param3 LIKE ?) "
"AND param4 = ?", new_sql)
