class BaseInsertGrammarTest:
def setUp(self):
self.builder = QueryBuilder(PostgresGrammar, 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)
def test_can_compile_bulk_create(self):
to_sql = self.builder.bulk_create([{
"name": "Joe"
}, {
"name": "Bill"
}, {
"name": "John"
}],
query=True).to_sql()
sql = getattr(
self,
inspect.currentframe().f_code.co_name.replace("test_", ""))()
self.assertEqual(to_sql, sql)
def test_can_compile_bulk_create_qmark(self):
to_sql = self.builder.bulk_create([{
"name": "Joe"
}, {
"name": "Bill"
}, {
"name": "John"
}],
query=True).to_qmark()
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(MSSQLGrammar, 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)
def test_can_compile_bulk_create(self):
to_sql = self.builder.bulk_create([{
"name": "Joe"
}, {
"name": "Bill"
}, {
"name": "John"
}],
query=True).to_sql()
sql = "INSERT INTO [users] ([name]) VALUES ('Joe'), ('Bill'), ('John')"
self.assertEqual(to_sql, sql)
def test_can_compile_bulk_create_qmark(self):
to_sql = self.builder.bulk_create([{
"name": "Joe"
}, {
"name": "Bill"
}, {
"name": "John"
}],
query=True).to_qmark()
sql = "INSERT INTO [users] ([name]) VALUES ('?'), ('?'), ('?')"
self.assertEqual(to_sql, sql)