def do(self):
self.finish = statement_finished(self.reader.get_unicode())
if not self.finish:
self.reader.insert(u("\n"))
def test_finished_multiline_string(self):
sql = "SELECT '\n;\n';"
self.assertTrue(statement_finished(sql))
def test_unfinished(self):
sql = "SELECT '\n;\n'"
self.assertFalse(statement_finished(sql))
def test_unfinished_string_with_newline(self):
sql = "SELECT '\n;"
self.assertFalse(statement_finished(sql))
def test_unfinished_string(self):
sql = "SELECT ';"
self.assertFalse(statement_finished(sql))
def test_finished(self):
sql = "SELECT * FROM spam WHERE eggs LIKE '%';"
self.assertTrue(statement_finished(sql))
