CassandraPrimaryKeyColumnAnnotationComparatorUnitTests.java

/*
 *  Copyright 2013-2017 the original author or authors
 *
 *  Licensed 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.springframework.data.cassandra.mapping;

import static org.assertj.core.api.Assertions.*;
import static org.springframework.data.cassandra.mapping.CassandraPrimaryKeyColumnAnnotationComparator.*;

import java.sql.Timestamp;
import java.util.UUID;

import org.junit.BeforeClass;
import org.junit.Test;
import org.springframework.cassandra.core.Ordering;
import org.springframework.cassandra.core.PrimaryKeyType;

/**
 * The CassandraPrimaryKeyColumnAnnotationComparatorUnitTests class is a test suite of test cases testing the contract
 * and functionality of the {@link CassandraPrimaryKeyColumnAnnotationComparator} class.
 *
 * @author John Blum
 * @author Mark Paluch
 */
public class CassandraPrimaryKeyColumnAnnotationComparatorUnitTests {

	static PrimaryKeyColumn entityOne;
	static PrimaryKeyColumn entityTwo;
	static PrimaryKeyColumn entityThree;
	static PrimaryKeyColumn entityFour;
	static PrimaryKeyColumn entityFive;

	@BeforeClass
	public static void setup() throws Exception {

		entityOne = EntityOne.class.getDeclaredField("id").getAnnotation(PrimaryKeyColumn.class);
		entityTwo = EntityTwo.class.getDeclaredField("id").getAnnotation(PrimaryKeyColumn.class);
		entityThree = EntityThree.class.getDeclaredField("id").getAnnotation(PrimaryKeyColumn.class);
		entityFour = EntityFour.class.getDeclaredField("id").getAnnotation(PrimaryKeyColumn.class);
		entityFive = EntityFive.class.getDeclaredField("id").getAnnotation(PrimaryKeyColumn.class);
	}

	@Test // DATACASS-248
	public void compareTypes() {

		assertThat(INSTANCE.compare(entityOne, entityTwo)).isEqualTo(1);
		assertThat(INSTANCE.compare(entityTwo, entityTwo)).isEqualTo(0);
		assertThat(INSTANCE.compare(entityTwo, entityOne)).isEqualTo(-1);
	}

	@Test // DATACASS-248
	public void compareOrdinals() {

		assertThat(INSTANCE.compare(entityOne, entityThree)).isEqualTo(-1);
		assertThat(INSTANCE.compare(entityThree, entityThree)).isEqualTo(0);
		assertThat(INSTANCE.compare(entityThree, entityOne)).isEqualTo(1);
	}

	@Test // DATACASS-248
	public void compareName() {

		assertThat(INSTANCE.compare(entityOne, entityFour)).isEqualTo(-1);
		assertThat(INSTANCE.compare(entityFour, entityFour)).isEqualTo(0);
		assertThat(INSTANCE.compare(entityFour, entityOne)).isEqualTo(1);
	}

	@Test // DATACASS-248
	public void compareOrdering() {

		assertThat(INSTANCE.compare(entityOne, entityFive)).isEqualTo(-1);
		assertThat(INSTANCE.compare(entityFive, entityFive)).isEqualTo(0);
		assertThat(INSTANCE.compare(entityFive, entityOne)).isEqualTo(1);
	}

	static class EntityOne {
		@PrimaryKeyColumn(type = PrimaryKeyType.CLUSTERED, ordinal = 1, name = "A",
				ordering = Ordering.ASCENDING) Integer id;
	}

	static class EntityTwo {
		@PrimaryKeyColumn(type = PrimaryKeyType.PARTITIONED, ordinal = 1, name = "A",
				ordering = Ordering.ASCENDING) Long id;
	}

	static class EntityThree {
		@PrimaryKeyColumn(type = PrimaryKeyType.CLUSTERED, ordinal = 2, name = "A",
				ordering = Ordering.ASCENDING) String id;
	}

	static class EntityFour {
		@PrimaryKeyColumn(type = PrimaryKeyType.CLUSTERED, ordinal = 1, name = "B",
				ordering = Ordering.ASCENDING) Timestamp id;
	}

	static class EntityFive {
		@PrimaryKeyColumn(type = PrimaryKeyType.CLUSTERED, ordinal = 1, name = "A", ordering = Ordering.DESCENDING) UUID id;
	}
}