class BaseInsertGrammarTest: def setUp(self): self.builder = QueryBuilder(GrammarFactory.make(self.grammar), table="users") def test_can_compile_insert(self): to_sql = self.builder.create({"name": "Joe"}, query=True).to_sql() sql = getattr( self, inspect.currentframe().f_code.co_name.replace("test_", "") )() self.assertEqual(to_sql, sql) def test_can_compile_insert_with_keywords(self): to_sql = self.builder.create(name="Joe", query=True).to_sql() sql = getattr( self, inspect.currentframe().f_code.co_name.replace("test_", "") )() self.assertEqual(to_sql, sql)
class TestMySQLInsertGrammar(unittest.TestCase): def setUp(self): self.builder = QueryBuilder(GrammarFactory.make("mssql"), table="users") def test_can_compile_insert(self): to_sql = self.builder.create({"name": "Joe"}, query=True).to_sql() sql = "INSERT INTO [users] ([users].[name]) VALUES ('Joe')" self.assertEqual(to_sql, sql)