fix argument parser test

ktrueda · Jan 17, 2023 · 1fcd393 · 1fcd393
1 parent ffb7916
commit 1fcd393
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/tests/test_csv.py b/tests/test_csv.py
@@ -36,17 +36,19 @@ def parquet_file():
             'columns': [],
             'file': ['file1.parquet'],
             'head': -1,
-            'awsprofile': None
+            'awsprofile': None,
+            'endpoint_url': None
         }
     ),
     # most complex one
     (
-        '--columns col1,col2 --head 100 --awsprofile user1 file1.parquet file2.parquet', True,
+        '--columns col1,col2 --head 100 --endpoint-url http://localhost:9000 --awsprofile user1 file1.parquet file2.parquet', True,
         {
             'columns': ['col1', 'col2'],
             'file': ['file1.parquet', 'file2.parquet'],
             'head': 100,
-            'awsprofile': 'user1'
+            'awsprofile': 'user1',
+            'endpoint_url': 'http://localhost:9000'
         }
     ),
     # empty columns

diff --git a/tests/test_show.py b/tests/test_show.py
@@ -22,7 +22,8 @@ def parser():
             'file': ['file1.parquet'],
             'format': 'psql',
             'head': -1,
-            'awsprofile': None
+            'awsprofile': None,
+            'endpoint_url': None
         }
     ),
     # most complex one
@@ -33,7 +34,8 @@ def parser():
             'file': ['file1.parquet', 'file2.parquet'],
             'format': 'github',
             'head': 100,
-            'awsprofile': None
+            'awsprofile': None,
+            'endpoint_url': None
         }
     ),
     # file is on S3
@@ -44,7 +46,8 @@ def parser():
             'file': ['s3://bucket-name/file1.parquet'],
             'format': 'psql',
             'head': -1,
-            'awsprofile': 'user1'
+            'awsprofile': 'user1',
+            'endpoint_url': None
         }
     ),
     # empty columns