Basic e2e test scaffolded in. Currently fails.

Jesse Whitehouse · Jesse Whitehouse · commit ce80df0300f9 · 2022-11-23T14:23:21.000-06:00
Signed-off-by: Jesse Whitehouse &lt;jesse.whitehouse@databricks.com&gt;
diff --git a/tests/e2e/driver_tests.py b/tests/e2e/driver_tests.py
@@ -661,7 +661,7 @@ def test_staging_ingestion_life_cycle(self):
         with open(fh, "wb") as fp:
             fp.write(original_text)
 
-        with self.connection() as conn:
+        with self.connection(extra_params={"uploads_base_path": temp_path}) as conn:
             cursor = conn.cursor()
             query = f"PUT '{temp_path}' INTO 'stage://tmp/{self.staging_ingestion_user}/tmp/11/15/file1.csv' OVERWRITE"
             cursor.execute(query)
@@ -702,6 +702,25 @@ def test_staging_ingestion_life_cycle(self):
         os.remove(new_temp_path)
 
 
+    def test_staging_ingestion_put_fails_without_uploadsbasepath(self):
+        """PUT operations are not supported unless the connection was built with
+        a parameter called uploads_base_path
+        """
+
+        fh, temp_path = tempfile.mkstemp()
+
+        original_text = "hello world!".encode("utf-8")
+
+        with open(fh, "wb") as fp:
+            fp.write(original_text)
+
+        with pytest.raises(Error):
+            with self.connection() as conn:
+                cursor = conn.cursor()
+                query = f"PUT '{temp_path}' INTO 'stage://tmp/{self.staging_ingestion_user}/tmp/11/15/file1.csv' OVERWRITE"
+                cursor.execute(query)
+
+
 def main(cli_args):
     global get_args_from_env
     get_args_from_env = True
