processing/src/test/java/org/apache/druid/segment/column/SegmentSchemaMappingTest.java

/*
 * 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.druid.segment.column;

import com.fasterxml.jackson.databind.ObjectMapper;
import org.apache.druid.common.config.NullHandling;
import org.apache.druid.query.aggregation.firstlast.last.StringLastAggregatorFactory;
import org.apache.druid.segment.SchemaPayload;
import org.apache.druid.segment.SchemaPayloadPlus;
import org.apache.druid.segment.SegmentMetadata;
import org.apache.druid.segment.SegmentSchemaMapping;
import org.apache.druid.segment.TestHelper;
import org.apache.druid.timeline.SegmentId;
import org.junit.Assert;
import org.junit.Test;

import java.io.IOException;
import java.util.Collections;

public class SegmentSchemaMappingTest
{
  static {
    NullHandling.initializeForTests();
  }

  private ObjectMapper mapper = TestHelper.makeJsonMapper();

  @Test
  public void testSerde() throws IOException
  {
    RowSignature rowSignature = RowSignature.builder().add("c", ColumnType.FLOAT).build();

    SegmentId segmentId = SegmentId.dummy("ds1");

    StringLastAggregatorFactory factory = new StringLastAggregatorFactory("billy", "nilly", null, 20);
    SchemaPayload payload = new SchemaPayload(rowSignature, Collections.singletonMap("twosum", factory));
    SchemaPayloadPlus schemaPayloadPlus = new SchemaPayloadPlus(payload, 20L);

    SegmentSchemaMapping segmentSchemaMapping = new SegmentSchemaMapping(
        Collections.singletonMap(segmentId.toString(), new SegmentMetadata(20L, "fp1")),
        Collections.singletonMap("fp1", payload),
        1
    );

    byte[] bytes = mapper.writeValueAsBytes(segmentSchemaMapping);
    SegmentSchemaMapping deserialized = mapper.readValue(bytes, SegmentSchemaMapping.class);

    Assert.assertEquals(segmentSchemaMapping, deserialized);

    SegmentSchemaMapping copy = new SegmentSchemaMapping(1);
    copy.merge(segmentSchemaMapping);

    Assert.assertEquals(segmentSchemaMapping, copy);

    SegmentSchemaMapping copy2 = new SegmentSchemaMapping(1);
    copy2.addSchema(segmentId, schemaPayloadPlus, "fp1");

    Assert.assertEquals(segmentSchemaMapping, copy2);
  }

  @Test
  public void testEquals()
  {
    RowSignature rowSignature = RowSignature.builder().add("c", ColumnType.FLOAT).build();

    SegmentId segmentId = SegmentId.dummy("ds1");

    StringLastAggregatorFactory factory = new StringLastAggregatorFactory("billy", "nilly", null, 20);
    SchemaPayload payload = new SchemaPayload(rowSignature, Collections.singletonMap("twosum", factory));

    SegmentSchemaMapping segmentSchemaMapping = new SegmentSchemaMapping(
        Collections.singletonMap(segmentId.toString(), new SegmentMetadata(20L, "fp1")),
        Collections.singletonMap("fp1", payload),
        1
    );

    SegmentSchemaMapping segmentSchemaMappingWithDifferentVersion = new SegmentSchemaMapping(
        Collections.singletonMap(segmentId.toString(), new SegmentMetadata(20L, "fp1")),
        Collections.singletonMap("fp1", payload),
        0
    );

    Assert.assertNotEquals(segmentSchemaMapping, segmentSchemaMappingWithDifferentVersion);

    SegmentSchemaMapping segmentSchemaMappingWithDifferentPayload = new SegmentSchemaMapping(
        Collections.emptyMap(),
        Collections.emptyMap(),
        0
    );

    Assert.assertNotEquals(segmentSchemaMapping, segmentSchemaMappingWithDifferentPayload);
  }
}