First try: Add Metadata Builder for tests

core/src/test/java/org/apache/iceberg/MetadataTestUtils.java

@@ -0,0 +1,239 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+package org.apache.iceberg;
+
+import static org.apache.iceberg.TableMetadata.INITIAL_SEQUENCE_NUMBER;
+
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableMap;
+import java.util.List;
+import java.util.Map;
+import java.util.UUID;
+import org.apache.iceberg.relocated.com.google.common.collect.Lists;
+import org.apache.iceberg.relocated.com.google.common.collect.Maps;
+import org.apache.iceberg.util.SerializableSupplier;
+
+public class MetadataTestUtils {
+
+  public static TableMetadataBuilder buildTestTableMetadataFromEmpty(int formatVersion) {
+    return new TableMetadataBuilder(formatVersion);
+  }
+
+  public static class TableMetadataBuilder {
+    private String metadataLocation;
+    private int formatVersion;
+    private String uuid;
+    private String location;
+    private long lastSequenceNumber;
+    private Long lastUpdatedMillis;
+    private int lastColumnId;
+    private int currentSchemaId;
+    private List<Schema> schemas;
+    private int defaultSpecId;
+    private List<PartitionSpec> specs;
+    private int lastAssignedPartitionId;
+    private int defaultSortOrderId;
+    private List<SortOrder> sortOrders;
+    private Map<String, String> properties;
+    private long currentSnapshotId;
+    private List<HistoryEntry> snapshotLog;
+    private List<TableMetadata.MetadataLogEntry> previousFiles;
+    private List<StatisticsFile> statisticsFiles;
+    private List<PartitionStatisticsFile> partitionStatisticsFiles;
+    private List<MetadataUpdate> changes;
+    private SerializableSupplier<List<Snapshot>> snapshotsSupplier;
+    private List<Snapshot> snapshots;
+    private Map<String, SnapshotRef> refs;
+
+    private TableMetadataBuilder(int formatVersion) {
+      this.formatVersion = formatVersion;
+      this.uuid = UUID.randomUUID().toString();
+      this.lastSequenceNumber = INITIAL_SEQUENCE_NUMBER;
+      this.lastUpdatedMillis = System.currentTimeMillis();
+      this.lastColumnId = -1;
+      this.currentSchemaId = -1;
+      this.schemas = Lists.newArrayList();
+      this.defaultSpecId = -1;
+      this.specs = Lists.newArrayList();
+      this.lastAssignedPartitionId = 999;
+      this.defaultSortOrderId = -1;
+      this.sortOrders = Lists.newArrayList();
+      this.properties = Maps.newHashMap();
+      this.snapshots = Lists.newArrayList();
+      this.currentSnapshotId = -1;
+      this.changes = Lists.newArrayList();
+      this.snapshotLog = Lists.newArrayList();
+      this.previousFiles = Lists.newArrayList();
+      this.refs = Maps.newHashMap();
+      this.statisticsFiles = Lists.newArrayList();
+      this.partitionStatisticsFiles = Lists.newArrayList();
+    }
+
+    public TableMetadataBuilder withMetadataFileLocation(String metadataFileLocation) {
+      this.metadataLocation = metadataFileLocation;
+      return this;
+    }
+
+    public TableMetadataBuilder withFormatVersion(int formatVersion) {
+      this.formatVersion = formatVersion;
+      return this;
+    }
+
+    public TableMetadataBuilder withUUID(String uuid) {
+      this.uuid = uuid;
+      return this;
+    }
+
+    public TableMetadataBuilder withLocation(String location) {
+      this.location = location;
+      return this;
+    }
+
+    public TableMetadataBuilder withLastSequenceNumber(long lastSequenceNumber) {
+      this.lastSequenceNumber = lastSequenceNumber;
+      return this;
+    }
+
+    public TableMetadataBuilder withLastUpdatedMillis(long lastUpdatedMillis) {
+      this.lastUpdatedMillis = lastUpdatedMillis;
+      return this;
+    }
+
+    public TableMetadataBuilder withLastColumnId(int lastColumnId) {
+      this.lastColumnId = lastColumnId;
+      return this;
+    }
+
+    public TableMetadataBuilder withCurrentSchemaId(int currentSchemaId) {
+      this.currentSchemaId = currentSchemaId;
+      return this;
+    }
+
+    public TableMetadataBuilder withSchemas(List<Schema> schemas) {
+      this.schemas = schemas;
+      return this;
+    }
+
+    public TableMetadataBuilder withDefaultSpecId(int defaultSpecId) {
+      this.defaultSpecId = defaultSpecId;
+      return this;
+    }
+
+    public TableMetadataBuilder withSpecs(List<PartitionSpec> specs) {
+      this.specs = specs;
+      return this;
+    }
+
+    public TableMetadataBuilder withLastAssignedPartitionId(int lastAssignedPartitionId) {
+      this.lastAssignedPartitionId = lastAssignedPartitionId;
+      return this;
+    }
+
+    public TableMetadataBuilder withDefaultSortOrderId(int defaultSortOrderId) {
+      this.defaultSortOrderId = defaultSortOrderId;
+      return this;
+    }
+
+    public TableMetadataBuilder withSortOrders(List<SortOrder> sortOrders) {
+      this.sortOrders = sortOrders;
+      return this;
+    }
+
+    public TableMetadataBuilder withProperties(Map<String, String> properties) {
+      this.properties = properties;
+      return this;
+    }
+
+    public TableMetadataBuilder withCurrentSnapshotId(long snapshotId) {
+      this.currentSnapshotId = snapshotId;
+      return this;
+    }
+
+    public TableMetadataBuilder withSnapshotsSupplier(
+        SerializableSupplier<List<Snapshot>> snapshotsSupplier) {
+      this.snapshotsSupplier = snapshotsSupplier;
+      return this;
+    }
+
+    public TableMetadataBuilder withSnapshots(List<Snapshot> snapshots) {
+      this.snapshots = snapshots;
+      return this;
+    }
+
+    public TableMetadataBuilder withSnapshotLog(List<HistoryEntry> snapshotLog) {
+      this.snapshotLog = snapshotLog;
+      return this;
+    }
+
+    public TableMetadataBuilder withMetadataHistory(
+        List<TableMetadata.MetadataLogEntry> previousFiles) {
+      this.previousFiles = previousFiles;
+      return this;
+    }
+
+    public TableMetadataBuilder withRefs(Map<String, SnapshotRef> refs) {
+      this.refs = refs;
+      return this;
+    }
+
+    public TableMetadataBuilder withChanges(List<MetadataUpdate> changes) {
+      this.changes = changes;
+      return this;
+    }
+
+    public TableMetadataBuilder withStatisticsFiles(List<StatisticsFile> statisticsFiles) {
+      this.statisticsFiles = statisticsFiles;
+      return this;
+    }
+
+    public TableMetadataBuilder withPartitionStatisticsFiles(
+        List<PartitionStatisticsFile> partitionStatisticsFiles) {
+      this.partitionStatisticsFiles = partitionStatisticsFiles;
+      return this;
+    }
+
+    public TableMetadata build() {
+      return new TableMetadata(
+          metadataLocation,
+          formatVersion,
+          uuid,
+          location,
+          lastSequenceNumber,
+          lastUpdatedMillis,
+          lastColumnId,
+          currentSchemaId,
+          ImmutableList.copyOf(schemas),
+          defaultSpecId,
+          ImmutableList.copyOf(specs),
+          lastAssignedPartitionId,
+          defaultSortOrderId,
+          ImmutableList.copyOf(sortOrders),
+          ImmutableMap.copyOf(properties),
+          currentSnapshotId,
+          ImmutableList.copyOf(snapshots),
+          snapshotsSupplier,
+          ImmutableList.copyOf(snapshotLog),
+          ImmutableList.copyOf(previousFiles),
+          ImmutableMap.copyOf(refs),
+          ImmutableList.copyOf(statisticsFiles),
+          ImmutableList.copyOf(partitionStatisticsFiles),
+          ImmutableList.copyOf(changes));
+    }
+  }
+}