Cleanup remove-pg_catalog-from-functions test code

No functional change
lelit · Sep 14, 2021 · 881854a · 881854a
1 parent e9a020b
commit 881854a
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -126,7 +126,14 @@ def test_cli_workhorse():
                                          "     , count(*)\n"
                                          "FROM t1\n")
 
-    with StringIO("select substring('123',2,3), regexp_split_to_array('x,x,x', ','), btrim('xxx'), trim('xxx'), POSITION('hour' in trim(substring('xyz hour ',1,6)))") as input:
+    in_stmt = """\
+select substring('123',2,3),
+       regexp_split_to_array('x,x,x', ','),
+       btrim('xxx'), trim('xxx'),
+       POSITION('hour' in trim(substring('xyz hour ',1,6)))
+"""
+
+    with StringIO(in_stmt) as input:
         with UnclosableStream() as output:
             with redirect_stdin(input), redirect_stdout(output):
                 main(['--compact-lists-margin', '100'])
@@ -138,7 +145,7 @@ def test_cli_workhorse():
      , pg_catalog.position(pg_catalog.btrim(pg_catalog.substring('xyz hour ', 1, 6)), 'hour')
 """
 
-    with StringIO("select substring('123',2,3), regexp_split_to_array('x,x,x', ','), btrim('xxx'), trim('xxx'), POSITION('hour' in trim(substring('xyz hour ',1,6)))") as input:
+    with StringIO(in_stmt) as input:
         with UnclosableStream() as output:
             with redirect_stdin(input), redirect_stdout(output):
                 main(['--remove-pg_catalog-from-functions', '--compact-lists-margin', '100'])
@@ -149,5 +156,3 @@ def test_cli_workhorse():
      , btrim('xxx')
      , pg_catalog.position(btrim(substring('xyz hour ', 1, 6)), 'hour')
 """
-
-