Merge pull request #1 from larsschwegmann/mvanderlee/parametized_query

Add test for PyformatParameter
kayak · Sep 28, 2024 · 2ae30df · 2ae30df
2 parents 3af56ee + a36a1aa
commit 2ae30df
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pypika/tests/test_parameter.py b/pypika/tests/test_parameter.py
@@ -196,3 +196,11 @@ def test_join_with_parameter_value_wrapper(self):
             sql,
         )
         self.assertEqual({':buz': 'buz', 'bar': 'bar'}, parameter.get_parameters())
+
+    def test_pyformat_parameter(self):
+        q = Query.into(self.table_abc).columns("a", "b", "c").insert(1, 2.2, 'foo')
+
+        parameter = PyformatParameter()
+        sql = q.get_sql(parameter=parameter)
+        self.assertEqual('INSERT INTO "abc" ("a","b","c") VALUES (%(param1)s,%(param2)s,%(param3)s)', sql)
+        self.assertEqual({"param1": 1, "param2": 2.2, "param3": "foo"}, parameter.get_parameters())