From d43ef44def47446a1b287b8ed169277a4a9efa8d Mon Sep 17 00:00:00 2001
From: Elvis Pranskevichus <elvis@edgedb.com>
Date: Thu, 17 Oct 2024 12:37:30 -0700
Subject: [PATCH] Add a test

---
 tests/test_copy.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/tests/test_copy.py b/tests/test_copy.py
index be2aabaf..e119e6d8 100644
--- a/tests/test_copy.py
+++ b/tests/test_copy.py
@@ -148,12 +148,14 @@ async def test_copy_from_query_with_args(self):
 
         res = await self.con.copy_from_query('''
             SELECT
-                i, i * 10
+                i,
+                i * 10,
+                $2::text
             FROM
                 generate_series(1, 5) AS i
             WHERE
                 i = $1
-        ''', 3, output=f)
+        ''', 3, None, output=f)
 
         self.assertEqual(res, 'COPY 1')
 
@@ -161,7 +163,7 @@ async def test_copy_from_query_with_args(self):
         self.assertEqual(
             output,
             [
-                '3\t30',
+                '3\t30\t\\N',
                 ''
             ]
         )