driver-mapping/src/test/java/com/datastax/driver/mapping/MapperCompositeKeyTest.java

/*
 * Copyright (C) 2012-2017 DataStax Inc.
 *
 * 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 com.datastax.driver.mapping;

import com.datastax.driver.core.CCMTestsSupport;
import com.datastax.driver.mapping.annotations.ClusteringColumn;
import com.datastax.driver.mapping.annotations.PartitionKey;
import com.datastax.driver.mapping.annotations.Table;
import org.testng.annotations.Test;

/**
 * Tests for the mapper with composite partition keys and multiple clustering columns.
 */
public class MapperCompositeKeyTest extends CCMTestsSupport {

    @Override
    public void onTestContextInitialized() {
        execute("CREATE TABLE test_table (pk1 int, pk2 int, cc1 int, cc2 int, PRIMARY KEY ((pk1, pk2), cc1, cc2))");
    }

    @SuppressWarnings("unused")
    @Table(name = "test_table")
    public static class TestTable {
        @PartitionKey(0)
        private int pk1;

        @PartitionKey(1)
        private int pk2;

        @ClusteringColumn(0)
        private int cc1;

        @ClusteringColumn(1)
        private int cc2;

        public int getPk1() {
            return pk1;
        }

        public void setPk1(int pk1) {
            this.pk1 = pk1;
        }

        public int getPk2() {
            return pk2;
        }

        public void setPk2(int pk2) {
            this.pk2 = pk2;
        }

        public int getCc1() {
            return cc1;
        }

        public void setCc1(int cc1) {
            this.cc1 = cc1;
        }

        public int getCc2() {
            return cc2;
        }

        public void setCc2(int cc2) {
            this.cc2 = cc2;
        }
    }

    @Test(groups = "short")
    public void testCreateMapper() throws Exception {
        new MappingManager(session()).mapper(TestTable.class);
    }
}