Add test for object storage data file locations

Author: Sreesh Maheshwar

tests/integration/test_writes/test_writes.py

@@ -273,6 +273,7 @@ def test_data_files(spark: SparkSession, session_catalog: Catalog, arrow_table_w
     # Since we don't rewrite, this should produce a new manifest with an ADDED entry
     tbl.append(arrow_table_with_null)
 
+
     rows = spark.sql(
         f"""
         SELECT added_data_files_count, existing_data_files_count, deleted_data_files_count
@@ -285,6 +286,28 @@ def test_data_files(spark: SparkSession, session_catalog: Catalog, arrow_table_w
     assert [row.deleted_data_files_count for row in rows] == [0, 1, 0, 0, 0]
 
 
+@pytest.mark.integration
+@pytest.mark.parametrize("format_version", [1, 2])
+def test_object_storage_data_files(spark: SparkSession, session_catalog: Catalog, arrow_table_with_null: pa.Table, format_version: int) -> None:
+    # TODO: What to do about "tbl.add_files()"?
+    identifier = "default.object_stored_table"
+
+    tbl = _create_table(session_catalog, identifier, {"format-version": format_version, "write.object-storage.enabled": True}, [])
+    tbl.append(arrow_table_with_null)
+
+    paths = tbl.inspect.entries().to_pydict()["data_file"]
+    assert len(paths) == 1
+    location = paths[0]["file_path"]
+
+    parts = location.split("/")
+    assert len(parts) == 11
+
+    assert location.startswith("s3://warehouse/default/object_stored_table/data/")
+    for i in range(6, 10):
+        assert len(parts[i]) == (8 if i == 9 else 4)
+        assert all(c in "01" for c in parts[i])
+
+
 @pytest.mark.integration
 def test_python_writes_with_spark_snapshot_reads(
     spark: SparkSession, session_catalog: Catalog, arrow_table_with_null: pa.Table