content: /*
   * 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.cassandra.db.rows;
  
  import java.nio.ByteBuffer;
  import java.util.Set;
  
  import org.junit.After;
  import org.junit.Before;
  import org.junit.Test;
  
  import org.apache.cassandra.db.marshal.*;
  import org.apache.cassandra.schema.ColumnMetadata;
  
  import static java.util.Arrays.asList;
  import static org.apache.cassandra.cql3.FieldIdentifier.forUnquoted;
  import static org.apache.cassandra.utils.ByteBufferUtil.bytes;
  import static org.junit.Assert.assertEquals;
  import static org.junit.Assert.fail;
  
  public class ColumnMetadataVersionComparatorTest
  {
      private UserType udtWith2Fields;
      private UserType udtWith3Fields;
  
      @Before
      public void setUp()
      {
          udtWith2Fields = new UserType("ks",
                                        bytes("myType"),
                                        asList(forUnquoted("a"), forUnquoted("b")),
                                        asList(Int32Type.instance, Int32Type.instance),
                                        false);
          udtWith3Fields = new UserType("ks",
                                        bytes("myType"),
                                        asList(forUnquoted("a"), forUnquoted("b"), forUnquoted("c")),
                                        asList(Int32Type.instance, Int32Type.instance, Int32Type.instance),
                                        false);
      }
  
      @After
      public void tearDown()
      {
          udtWith2Fields = null;
          udtWith3Fields = null;
      }
  
      @Test
      public void testWithSimpleTypes()
      {
          checkComparisonResults(Int32Type.instance, BytesType.instance);
          checkComparisonResults(EmptyType.instance, BytesType.instance);
      }
  
      @Test
      public void testWithTuples()
      {
          checkComparisonResults(new TupleType(asList(Int32Type.instance, Int32Type.instance)),
                                 new TupleType(asList(Int32Type.instance, Int32Type.instance, Int32Type.instance)));
      }
  
      @Test
      public void testWithUDTs()
      {
          checkComparisonResults(udtWith2Fields, udtWith3Fields);
      }
  
      @Test
      public void testWithUDTsNestedWithinSet()
      {
          for (boolean isMultiCell : new boolean[]{false, true})
          {
              SetType<ByteBuffer> set1 = SetType.getInstance(udtWith2Fields, isMultiCell);
              SetType<ByteBuffer> set2 = SetType.getInstance(udtWith3Fields, isMultiCell);
              checkComparisonResults(set1, set2);
          }
      }
  
      @Test
      public void testWithUDTsNestedWithinList()
      {
          for (boolean isMultiCell : new boolean[]{false, true})
          {
              ListType<ByteBuffer> list1 = ListType.getInstance(udtWith2Fields, isMultiCell);
              ListType<ByteBuffer> list2 = ListType.getInstance(udtWith3Fields, isMultiCell);
              checkComparisonResults(list1, list2);
          }
      }
  
      @Test
      public void testWithUDTsNestedWithinMap()
      {
          for (boolean isMultiCell : new boolean[]{false, true})
          {
              MapType<ByteBuffer, Integer> map1 = MapType.getInstance(udtWith2Fields, Int32Type.instance, isMultiCell);
              MapType<ByteBuffer, Integer> map2 = MapType.getInstance(udtWith3Fields, Int32Type.instance, isMultiCell);
              checkComparisonResults(map1, map2);
          }
  
          for (boolean isMultiCell : new boolean[]{false, true})
          {
              MapType<Integer, ByteBuffer> map1 = MapType.getInstance(Int32Type.instance, udtWith2Fields, isMultiCell);
              MapType<Integer, ByteBuffer> map2 = MapType.getInstance(Int32Type.instance, udtWith3Fields, isMultiCell);
              checkComparisonResults(map1, map2);
          }
      }
  
      @Test
      public void testWithUDTsNestedWithinTuple()
      {
          TupleType tuple1 = new TupleType(asList(udtWith2Fields, Int32Type.instance));
          TupleType tuple2 = new TupleType(asList(udtWith3Fields, Int32Type.instance));
          checkComparisonResults(tuple1, tuple2);
      }
  
      @Test
      public void testWithUDTsNestedWithinComposite()
      {
          CompositeType composite1 = CompositeType.getInstance(asList(udtWith2Fields, Int32Type.instance));
          CompositeType composite2 = CompositeType.getInstance(asList(udtWith3Fields, Int32Type.instance));
          checkComparisonResults(composite1, composite2);
      }
  
      @Test
      public void testWithDeeplyNestedUDT()
      {
          for (boolean isMultiCell : new boolean[]{false, true})
          {
              ListType<Set<ByteBuffer>> list1 = ListType.getInstance(SetType.getInstance(new TupleType(asList(udtWith2Fields, Int32Type.instance)), isMultiCell), isMultiCell);
              ListType<Set<ByteBuffer>> list2 = ListType.getInstance(SetType.getInstance(new TupleType(asList(udtWith3Fields, Int32Type.instance)), isMultiCell), isMultiCell);
              checkComparisonResults(list1, list2);
          }
      }
  
      @Test
      public void testInvalidComparison()
      {
          assertInvalidComparison("Found 2 incompatible versions of column c in ks.t: one of type org.apache.cassandra.db.marshal.Int32Type and one of type org.apache.cassandra.db.marshal.UTF8Type (but both types are incompatible)",
                                  Int32Type.instance,
                                  UTF8Type.instance);
          assertInvalidComparison("Found 2 incompatible versions of column c in ks.t: one of type org.apache.cassandra.db.marshal.FrozenType(org.apache.cassandra.db.marshal.UserType(ks,6d7954797065,61:org.apache.cassandra.db.marshal.Int32Type,62:org.apache.cassandra.db.marshal.Int32Type)) and one of type org.apache.cassandra.db.marshal.Int32Type (but both types are incompatible)",
                                  udtWith2Fields,
                                  Int32Type.instance);
          assertInvalidComparison("Found 2 incompatible versions of column c in ks.t: one of type org.apache.cassandra.db.marshal.SetType(org.apache.cassandra.db.marshal.UTF8Type) and one of type org.apache.cassandra.db.marshal.SetType(org.apache.cassandra.db.marshal.InetAddressType) (but both types are incompatible)",
                                  SetType.getInstance(UTF8Type.instance, true),
                                  SetType.getInstance(InetAddressType.instance, true));
          assertInvalidComparison("Found 2 incompatible versions of column c in ks.t: one of type org.apache.cassandra.db.marshal.ListType(org.apache.cassandra.db.marshal.UTF8Type) and one of type org.apache.cassandra.db.marshal.ListType(org.apache.cassandra.db.marshal.InetAddressType) (but both types are incompatible)",
                                  ListType.getInstance(UTF8Type.instance, true),
                                  ListType.getInstance(InetAddressType.instance, true));
          assertInvalidComparison("Found 2 incompatible versions of column c in ks.t: one of type org.apache.cassandra.db.marshal.MapType(org.apache.cassandra.db.marshal.UTF8Type,org.apache.cassandra.db.marshal.IntegerType) and one of type org.apache.cassandra.db.marshal.MapType(org.apache.cassandra.db.marshal.InetAddressType,org.apache.cassandra.db.marshal.IntegerType) (but both types are incompatible)",
                                  MapType.getInstance(UTF8Type.instance, IntegerType.instance, true),
                                  MapType.getInstance(InetAddressType.instance, IntegerType.instance, true));
          assertInvalidComparison("Found 2 incompatible versions of column c in ks.t: one of type org.apache.cassandra.db.marshal.MapType(org.apache.cassandra.db.marshal.IntegerType,org.apache.cassandra.db.marshal.UTF8Type) and one of type org.apache.cassandra.db.marshal.MapType(org.apache.cassandra.db.marshal.IntegerType,org.apache.cassandra.db.marshal.InetAddressType) (but both types are incompatible)",
                                  MapType.getInstance(IntegerType.instance, UTF8Type.instance, true),
                                  MapType.getInstance(IntegerType.instance, InetAddressType.instance, true));
      }
  
      private void assertInvalidComparison(String expectedMessage, AbstractType<?> oldVersion, AbstractType<?> newVersion)
      {
          try
          {
              checkComparisonResults(oldVersion, newVersion);
              fail("comparison doesn't throw expected IllegalArgumentException: " + expectedMessage);
          }
          catch (IllegalArgumentException e)
          {
              System.out.println(e.getMessage());
              assertEquals(expectedMessage, e.getMessage());
          }
      }
  
      private void checkComparisonResults(AbstractType<?> oldVersion, AbstractType<?> newVersion)
      {
          assertEquals(0, compare(oldVersion, oldVersion));
          assertEquals(0, compare(newVersion, newVersion));
          assertEquals(-1, compare(oldVersion, newVersion));
          assertEquals(1, compare(newVersion, oldVersion));
      }
  
      private static int compare(AbstractType<?> left, AbstractType<?> right)
      {
          ColumnMetadata v1 = ColumnMetadata.regularColumn("ks", "t", "c", left, ColumnMetadata.NO_UNIQUE_ID);
          ColumnMetadata v2 = ColumnMetadata.regularColumn("ks", "t", "c", right, ColumnMetadata.NO_UNIQUE_ID);
          return ColumnMetadataVersionComparator.INSTANCE.compare(v1, v2);
      }
  }
  