YugabyteDB (2.13.0.0-b42, bfc6a6643e7399ac8a0e81d06a3ee6d6571b33ab)

//--------------------------------------------------------------------------------------------------

// Copyright (c) YugaByte, 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.

//

//--------------------------------------------------------------------------------------------------

#include "yb/gutil/strings/substitute.h"

#include "yb/yql/cql/ql/test/ql-test-base.h"

using std::string;

using strings::Substitute;

namespace yb {

namespace ql {

class TestQLDeleteTable : public QLTestBase {

 public:

  TestQLDeleteTable() : QLTestBase() {

  }

};

TEST_F(TestQLDeleteTable, TestQLDeleteTableSimple) {

  // Init the simulated cluster.

  ASSERT_NO_FATALS(CreateSimulatedCluster());

  // Get a processor.

  TestQLProcessor *processor = GetQLProcessor();

  // -----------------------------------------------------------------------------------------------

  // Create the table.

  const char *create_stmt =

    "CREATE TABLE test_table(h1 int, h2 varchar, "

                            "r1 int, r2 varchar, "

                            "v1 int, v2 varchar, "

                            "primary key((h1, h2), r1, r2));";

  CHECK_VALID_STMT(create_stmt);

  // -----------------------------------------------------------------------------------------------

  // Unknown table.

  CHECK_INVALID_STMT("DELETE from test_table_unknown WHERE h1 = 0 AND h2 = 'zero';");

  // Missing hash key.

  CHECK_INVALID_STMT("DELETE from test_table;");

  CHECK_INVALID_STMT("DELETE from test_table WHERE h1 = 0;");

  // Wrong operator on hash key.

  CHECK_INVALID_STMT("DELETE from test_table WHERE h1 > 0 AND h2 = 'zero';");

  // -----------------------------------------------------------------------------------------------

  // TESTCASE: DELETE statement using only hash key (partition key).

  // Insert 100 rows into the table.

  static const int kNumRows = 100;

  string select_stmt;

  for (int idx = 0; idx < kNumRows; idx++) {

    // INSERT: Valid statement with column list.

    string stmt = Substitute("INSERT INTO test_table(h1, h2, r1, r2, v1, v2) "

                             "VALUES($0, 'h$1', $2, 'r$3', $4, 'v$5');",

                             idx, idx, idx+100, idx+100, idx+1000, idx+1000);

    CHECK_VALID_STMT(stmt);

  }

  // Testing DELETE one row.

  for (int idx = 0; idx < kNumRows; idx++) {

    // SELECT an entry to make sure it's there.

    select_stmt = Substitute("SELECT * FROM test_table"

                             "  WHERE h1 = $0 AND h2 = 'h$1' AND r1 = $2 AND r2 = 'r$3';",

                             idx, idx, idx+100, idx+100);

    CHECK_VALID_STMT(select_stmt);

    std::shared_ptr<QLRowBlock> one_row_block = processor->row_block();

    CHECK_EQ(one_row_block->row_count(), 1);

    // DELETE the entry.

    CHECK_VALID_STMT(Substitute("DELETE FROM test_table"

                                "  WHERE h1 = $0 AND h2 = 'h$1' AND r1 = $2 AND r2 = 'r$3';",

                                idx, idx, idx+100, idx+100));

    // SELECT the same entry to make sure it's no longer there.

    CHECK_VALID_STMT(select_stmt);

    std::shared_ptr<QLRowBlock> empty_row_block = processor->row_block();

    CHECK_EQ(empty_row_block->row_count(), 0);

  }

#if 0

  // -----------------------------------------------------------------------------------------------

  // TESTCASE: DELETE statement using range key.

  // Insert 100 rows into the table that share the same partition key.

  for (int idx = 0; idx < kNumRows; idx++) {

    // INSERT: Valid statement with column list.

    string stmt = Substitute("INSERT INTO test_table(h1, h2, r1, r2, v1, v2)"

                             "  VALUES($0, 'h$1', $2, 'r$3', $4, 'v$5');",

                             9999, 9999, idx+100, idx+100, idx+1000, idx+1000);

    CHECK_VALID_STMT(stmt);

  }

  LOG(INFO) << kNumRows << "rows were inserted";

  // Delete the first half of the table and check.

  // SELECT entries to make sure they are there.

  select_stmt = Substitute("SELECT * FROM test_table"

                           "  WHERE h1 = $0 AND h2 = 'h$1' AND r1 < $2 AND r2 < 'r$3';",

                           9999, 9999, 150, 150);

  CHECK_VALID_STMT(select_stmt);

  std::shared_ptr<QLRowBlock> row_block = processor->row_block();

  CHECK_EQ(row_block->row_count(), 50);

  LOG(INFO) << "50 rows were selected";

  // DELETE the entry.

  CHECK_VALID_STMT(Substitute("DELETE FROM test_table"

                              "  WHERE h1 = $0 AND h2 = 'h$1' AND r1 < $2 AND r2 < 'r$3';",

                              9999, 9999, 150, 150));

  LOG(INFO) << "Expecting that 50 rows were deleted";

  // SELECT the same entries to make sure they are no longer there.

  CHECK_VALID_STMT(select_stmt);

  row_block = processor->row_block();

  if (row_block->row_count() > 0) {

    LOG(WARNING) << "Feature not yet supported. Not all rows are deleted";

  } else {

    LOG(INFO) << "50 rows were deleted";

  }

  // Delete the rest of the table and check.

  // SELECT entries to make sure they are there.

  select_stmt = Substitute("SELECT * FROM test_table"

                           "  WHERE h1 = $0 AND h2 = 'h$1' AND r1 > $2 AND r2 > 'r$3';",

                           9999, 9999, 149, 149);

  CHECK_VALID_STMT(select_stmt);

  row_block = processor->row_block();

  CHECK_EQ(row_block->row_count(), 50);

  LOG(INFO) << "50 rows were selected";

  // DELETE the entry.

  CHECK_VALID_STMT(Substitute("DELETE FROM test_table"

                              "  WHERE h1 = $0 AND h2 = 'h$1' AND r1 > $2 AND r2 > 'r$3';",

                              9999, 9999, 149, 149));

  LOG(INFO) << "Expecting that 50 rows were deleted";

  // SELECT the same entries to make sure they are no longer there.

  CHECK_VALID_STMT(select_stmt);

  row_block = processor->row_block();

  if (row_block->row_count() > 0) {

    LOG(WARNING) << "Feature not yet supported. Not all rows are deleted";

  } else {

    LOG(INFO) << "50 rows were deleted";

  }

#endif

}

} // namespace ql

} // namespace yb