modules/runner/src/integrationTest/java/org/apache/ignite/internal/runner/app/ITTableCreationTest.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.ignite.internal.runner.app;

import java.nio.file.Path;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.UUID;
import org.apache.ignite.app.Ignite;
import org.apache.ignite.app.IgnitionManager;
import org.apache.ignite.internal.testframework.WorkDirectory;
import org.apache.ignite.internal.testframework.WorkDirectoryExtension;
import org.apache.ignite.internal.util.IgniteUtils;
import org.apache.ignite.table.KeyValueBinaryView;
import org.apache.ignite.table.Table;
import org.apache.ignite.table.Tuple;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Disabled;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNull;

/**
 * Ignition interface tests.
 */
@Disabled("https://issues.apache.org/jira/browse/IGNITE-14578")
@ExtendWith(WorkDirectoryExtension.class)
class ITTableCreationTest {
    /** Nodes bootstrap configuration with preconfigured tables. */
    private final LinkedHashMap<String, String> nodesBootstrapCfg = new LinkedHashMap<>() {{
        put("node0", "{\n" +
            "  \"node\": {\n" +
            "    \"name\":node0,\n" +
            "    \"metastorageNodes\":[ \"node0\", \"node1\" ]\n" +
            "  },\n" +
            "  \"network\": {\n" +
            "    \"port\":3344,\n" +
            "    \"netClusterNodes\":[ \"localhost:3344\", \"localhost:3345\", \"localhost:3346\" ]\n" +
            "  },\n" +
            "  \"table\": {\n" +
            "       \"tables\": {\n" +
            "           \"tbl1\": {\n" +
            "               \"partitions\":10,\n" +
            "               \"replicas\":2,\n" +
            "               \"columns\": { \n" +
            "                   \"key\": {\n" +
            "                       \"type\": {" +
            "                           \"type\":UUID\n" +
            "                       },\n" +
            "                       \"nullable\":false\n" +
            "                   },\n" +
            "                   \"affKey\": {\n" +
            "                       \"type\": {" +
            "                           \"type\":INT64\n" +
            "                       },\n" +
            "                       \"nullable\":false\n" +
            "                   },\n" +
            "                   \"valString\": {\n" +
            "                       \"type\": {" +
            "                           \"type\":String\n" +
            "                       },\n" +
            "                       \"nullable\":false\n" +
            "                   },\n" +
            "                   \"valInt\": {\n" +
            "                       \"type\": {" +
            "                           \"type\":INT32\n" +
            "                       },\n" +
            "                       \"nullable\":false\n" +
            "                   },\n" +
            "                   \"valNullable\": {\n" +
            "                       \"type\": {" +
            "                           \"type\":String\n" +
            "                       },\n" +
            "                       \"nullable\":true\n" +
            "                   }\n" +
            "               },\n" + /* Columns. */
            "               \"indices\": {\n" +
            "                   \"PK\": {\n" +
            "                       \"type\":PRIMARY,\n" +
            "                       \"columns\": {\n" +
            "                           \"key\": {\n" +
            "                               \"asc\":true\n" +
            "                           },\n" +
            "                           \"affKey\": {}\n" +
            "                       },\n" + /* Columns. */
            "                       \"affinityColumns\":[ \"affKey\" ]\n" +
            "                   }\n" +
            "               }\n" + /* Indices. */
            "           },\n" + /* Table. */
            "\n" +
            "           \"tbl2\": {\n" + // Table minimal configuration.
            "               \"columns\": { \n" +
            "                   \"key\": {\n" +
            "                       \"type\": {" +
            "                           \"type\":INT64\n" +
            "                       },\n" +
            "                   },\n" +
            "                   \"val\": {\n" +
            "                       \"type\": {" +
            "                           \"type\":INT64\n" +
            "                       },\n" +
            "                   }\n" +
            "               },\n" + /* Columns. */
            "               \"indices\": {\n" +
            "                   \"PK\": {\n" +
            "                       \"type\":PRIMARY,\n" +
            "                       \"columns\": {\n" +
            "                           \"key\": {}\n" +
            "                       },\n" + /* Columns. */
            "                   }\n" +
            "               }\n" + /* Indices. */
            "           }\n" + /* Table. */
            "       }\n" + /* Tables. */
            "  }\n" + /* Root. */
            "}");

        put("node1", "{\n" +
            "  \"node\": {\n" +
            "    \"metastorageNodes\":[ \"node0\", \"node1\" ]\n" +
            "  },\n" +
            "  \"network\": {\n" +
            "    \"port\":3345,\n" +
            "    \"netClusterNodes\":[ \"localhost:3344\", \"localhost:3345\", \"localhost:3346\" ]\n" +
            "  }\n" +
            "}");

        put("node2", "{\n" +
            "  \"node\": {\n" +
            "    \"metastorageNodes\":[ \"node0\", \"node1\" ]\n" +
            "  },\n" +
            "  \"network\": {\n" +
            "    \"port\":3346,\n" +
            "    \"netClusterNodes\":[ \"localhost:3344\", \"localhost:3345\", \"localhost:3346\" ]\n" +
            "  }\n" +
            "}");
    }};

    /** */
    private final List<Ignite> clusterNodes = new ArrayList<>();

    /** */
    @WorkDirectory
    private Path workDir;

    /** */
    @AfterEach
    void tearDown() throws Exception {
        IgniteUtils.closeAll(clusterNodes);
    }

    /**
     * Check table creation via bootstrap configuration with pre-configured table.
     */
    @Test
    void testInitialSimpleTableConfiguration() {
        nodesBootstrapCfg.forEach((nodeName, configStr) ->
            clusterNodes.add(IgnitionManager.start(nodeName, configStr, workDir.resolve(nodeName)))
        );

        assertEquals(3, clusterNodes.size());

        clusterNodes.forEach(Assertions::assertNotNull);

        { /* Table 1. */
            Table tbl1 = clusterNodes.get(1).tables().table("tbl1");
            KeyValueBinaryView kvView1 = tbl1.kvView();

            tbl1.insert(Tuple.create().set("key", 1L).set("val", 111));
            kvView1.put(Tuple.create().set("key", 2L), Tuple.create().set("val", 222));

            Table tbl2 = clusterNodes.get(2).tables().table("tbl1");
            KeyValueBinaryView kvView2 = tbl2.kvView();

            final Tuple keyTuple1 = Tuple.create().set("key", 1L);
            final Tuple keyTuple2 = Tuple.create().set("key", 2L);

            assertEquals(111, (Integer)tbl2.get(keyTuple1).value("val"));
            assertEquals(111, (Integer)kvView1.get(keyTuple1).value("val"));
            assertEquals(222, (Integer)tbl2.get(keyTuple2).value("val"));
            assertEquals(222, (Integer)kvView1.get(keyTuple2).value("val"));
        }

        { /* Table 2. */
            final UUID uuid = UUID.randomUUID();
            final UUID uuid2 = UUID.randomUUID();

            // Put data on node 1.
            Table tbl1 = clusterNodes.get(1).tables().table("tbl1");
            KeyValueBinaryView kvView1 = tbl1.kvView();

            tbl1.insert(Tuple.create().set("key", uuid).set("affKey", 42L)
                .set("valStr", "String value").set("valInt", 73).set("valNullable", null));

            kvView1.put(Tuple.create().set("key", uuid2).set("affKey", 4242L),
                Tuple.create().set("valStr", "String value 2").set("valInt", 7373).set("valNullable", null));

            // Get data on node 2.
            Table tbl2 = clusterNodes.get(2).tables().table("tbl1");
            KeyValueBinaryView kvView2 = tbl2.kvView();

            final Tuple keyTuple1 = Tuple.create().set("key", uuid).set("affKey", 42L);
            final Tuple keyTuple2 = Tuple.create().set("key", uuid2).set("affKey", 4242L);

            assertEquals("String value", tbl2.get(keyTuple1).value("valStr"));
            assertEquals("String value", kvView2.get(keyTuple1).value("valStr"));
            assertEquals("String value 2", tbl2.get(keyTuple2).value("valStr"));
            assertEquals("String value 2", kvView2.get(keyTuple2).value("valStr"));
            assertEquals(73, (Integer)tbl2.get(keyTuple1).value("valInt"));
            assertEquals(73, (Integer)kvView2.get(keyTuple1).value("valInt"));
            assertEquals(7373, (Integer)tbl2.get(keyTuple2).value("valInt"));
            assertEquals(7373, (Integer)kvView2.get(keyTuple2).value("valInt"));
            assertNull(tbl2.get(keyTuple1).value("valNullable"));
            assertNull(kvView2.get(keyTuple1).value("valNullable"));
            assertNull(tbl2.get(keyTuple2).value("valNullable"));
            assertNull(kvView2.get(keyTuple2).value("valNullable"));
        }
    }
}