Skip to content

add test cases for 4377 error #129

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Apr 23, 2024
Merged

add test cases for 4377 error #129

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion src/test/java/com/alipay/oceanbase/rpc/ObTableErrMsgTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -234,6 +234,6 @@ public void testUpdateVirtualColumnNotSupport() {
);

System.out.println(thrown.getMessage());
assertTrue(thrown.getMessage().contains("[-4007][OB_NOT_SUPPORTED][assign virtual generated column not supported]"));
assertTrue(thrown.getMessage().contains("[-4007][OB_NOT_SUPPORTED][The specified value for generated column is not allowed]"));
}
}
215 changes: 215 additions & 0 deletions src/test/java/com/alipay/oceanbase/rpc/ObTableIndexWithCalcColumn.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@

import com.alipay.oceanbase.rpc.ObTableClient;
import com.alipay.oceanbase.rpc.mutation.Row;
import com.alipay.oceanbase.rpc.mutation.result.MutationResult;
import com.alipay.oceanbase.rpc.stream.QueryResultSet;
import com.alipay.oceanbase.rpc.util.ObTableClientTestUtil;
import org.junit.After;
Expand All @@ -31,6 +32,7 @@

import static com.alipay.oceanbase.rpc.mutation.MutationFactory.colVal;
import static com.alipay.oceanbase.rpc.mutation.MutationFactory.row;
import static org.junit.Assert.*;

public class ObTableIndexWithCalcColumn {

Expand Down Expand Up @@ -407,4 +409,217 @@ public void test_query(boolean is_expire) throws Exception {
Assert.assertEquals(resultSet2.cacheSize(), recordCount);
}
}


@Test
public void testLocalIndexHasGenerateColumn() throws Exception {
/*
CREATE TABLE `test_local_index_with_vgen_col` (
`name` varchar(512) NOT NULL DEFAULT '',
`pk` varchar(512) NOT NULL,
`adiu` varchar(512) NOT NULL DEFAULT '',
`id` bigint(20) NOT NULL DEFAULT 0,
`name_v` varchar(20) GENERATED ALWAYS AS (substr(`name`,1,5)) VIRTUAL,
`gmt_create` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
PRIMARY KEY (`adiu`, `pk`, `gmt_create`),
KEY `idx_adiu_v_name` (`adiu`, `name_v`) BLOCK_SIZE 16384 LOCAL
) TTL (gmt_create + INTERVAL 300 SECOND) partition by key(adiu) partitions 8;
*/
String TABLE_NAME = "test_local_index_with_vgen_col";

}

@Test
public void testGlobalIndexHasGenerateColumn() throws Exception {
/*
CREATE TABLE `test_global_index_with_vgen_col` (
`name` varchar(512) NOT NULL DEFAULT '',
`pk` varchar(512) NOT NULL,
`adiu` varchar(512) NOT NULL DEFAULT '',
`id` bigint(20) NOT NULL DEFAULT 0,
`name_v` varchar(20) GENERATED ALWAYS AS (substr(`name`,1,5)) VIRTUAL,
`gmt_create` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
PRIMARY KEY (`adiu`, `pk`, `gmt_create`),
KEY `idx_adiu_v_name` (`adiu`) global
) TTL (gmt_create + INTERVAL 300 SECOND) partition by key(adiu) partitions 8;
*/
String tableName = "test_global_index_with_vgen_col";
testIndexHasGenerateColumn(tableName);
}

public void testIndexHasGenerateColumn(String tableName) throws SQLException {
long cur_ts = System.currentTimeMillis();
Timestamp cur_time = new Timestamp(cur_ts);
cur_time.setNanos(0);
Timestamp expire_time = new Timestamp(cur_ts - 301000);
expire_time.setNanos(0);
try {
Row rowKey1 = row(colVal("adiu", "adiu111"), colVal("pk", "pk1"),
colVal("gmt_create", cur_time));
Row rowKey2 = row(colVal("adiu", "adiu222"), colVal("pk", "pk2"),
colVal("gmt_create", expire_time));
Row mutateRow1 = row(colVal("id", 1L), colVal("name", "name1111111"));
Row mutateRow2 = row(colVal("id", 2L), colVal("name", "name2222222"));

// prepare data
MutationResult res = client.insert(tableName).setRowKey(rowKey1).addMutateRow(mutateRow1).execute();
assertEquals(1, res.getAffectedRows());
res = client.insert(tableName).setRowKey(rowKey2).addMutateRow(mutateRow2).execute();
assertEquals(1, res.getAffectedRows());
// get
// not expired row
Map<String, Object> getRes = client.get(tableName, rowKey1.getValues(),null);
assertTrue(!getRes.isEmpty());
assertEquals("adiu111",getRes.get("adiu"));
assertEquals("pk1", getRes.get("pk"));
assertEquals(cur_time, getRes.get("gmt_create"));
assertEquals(1L, getRes.get("id"));
assertEquals("name1111111", getRes.get("name"));
assertEquals("name1", getRes.get("name_v"));
// expired row
getRes = client.get(tableName, rowKey2.getValues(),null);
assertTrue(getRes.isEmpty());

// insertup
// not expired: update
res = client.insertOrUpdate(tableName).setRowKey(rowKey1).addMutateRow(mutateRow2).execute();
assertEquals(1, res.getAffectedRows());
// expired: delete + insert
res = client.insertOrUpdate(tableName).setRowKey(rowKey2).addMutateRow(mutateRow1).execute();
assertEquals(1, res.getAffectedRows());
// get
getRes = client.get(tableName, rowKey1.getValues(),null);
assertTrue(!getRes.isEmpty());
assertEquals("adiu111",getRes.get("adiu"));
assertEquals("pk1", getRes.get("pk"));
assertEquals(cur_time, getRes.get("gmt_create"));
assertEquals(2L, getRes.get("id"));
assertEquals("name2222222", getRes.get("name"));
assertEquals("name2", getRes.get("name_v"));
// expired row
getRes = client.get(tableName, rowKey2.getValues(),null);
assertTrue(getRes.isEmpty());

// update
res = client.update(tableName).setRowKey(rowKey1).addMutateRow(mutateRow1).execute();
assertEquals(1, res.getAffectedRows());
getRes = client.get(tableName, rowKey1.getValues(),null);
assertTrue(!getRes.isEmpty());
assertEquals("adiu111",getRes.get("adiu"));
assertEquals("pk1", getRes.get("pk"));
assertEquals(cur_time, getRes.get("gmt_create"));
assertEquals(1L, getRes.get("id"));
assertEquals("name1111111", getRes.get("name"));
assertEquals("name1", getRes.get("name_v"));

// replace
res = client.replace(tableName).setRowKey(rowKey1).addMutateRow(mutateRow2).execute();
assertEquals(2, res.getAffectedRows());
getRes = client.get(tableName, rowKey1.getValues(),null);
assertTrue(!getRes.isEmpty());
assertEquals("adiu111",getRes.get("adiu"));
assertEquals("pk1", getRes.get("pk"));
assertEquals(cur_time, getRes.get("gmt_create"));
assertEquals(2L, getRes.get("id"));
assertEquals("name2222222", getRes.get("name"));
assertEquals("name2", getRes.get("name_v"));

// increment
Row incrMutateRow = row(colVal("id", 100L));
res = client.increment(tableName).setRowKey(rowKey1).addMutateRow(incrMutateRow).execute();
assertEquals(1, res.getAffectedRows());
getRes = client.get(tableName, rowKey1.getValues(),null);
assertTrue(!getRes.isEmpty());
assertEquals("adiu111",getRes.get("adiu"));
assertEquals("pk1", getRes.get("pk"));
assertEquals(cur_time, getRes.get("gmt_create"));
assertEquals(102L, getRes.get("id"));
assertEquals("name2222222", getRes.get("name"));
assertEquals("name2", getRes.get("name_v"));

// append
Row appendMutateRow = row(colVal("name", "ssssss"));
res = client.append(tableName).setRowKey(rowKey1).addMutateRow(appendMutateRow).execute();
assertEquals(1, res.getAffectedRows());
getRes = client.get(tableName, rowKey1.getValues(),null);
assertTrue(!getRes.isEmpty());
assertEquals("adiu111",getRes.get("adiu"));
assertEquals("pk1", getRes.get("pk"));
assertEquals(cur_time, getRes.get("gmt_create"));
assertEquals(102L, getRes.get("id"));
assertEquals("name2222222ssssss", getRes.get("name"));
assertEquals("name2", getRes.get("name_v"));

// delete
res = client.delete(tableName).setRowKey(rowKey1).execute();
assertEquals(1, res.getAffectedRows());
client.delete(tableName).setRowKey(rowKey2).execute();
assertEquals(1, res.getAffectedRows());
} catch (Exception ex) {
ex.printStackTrace();
assertFalse(true);
} finally {
ObTableClientTestUtil.getConnection().createStatement().execute("truncate table "+ tableName);
}
}

@Test
public void test_current_ts() throws Exception {
String tableName = "test_current_timestamp";
try {
long cur_ts = System.currentTimeMillis();
Timestamp ts_1 = new Timestamp(cur_ts + 1000000);
ts_1.setNanos(0);
Timestamp ts_2 = new Timestamp(cur_ts - 1000000);
ts_2.setNanos(0);
Row rowKey1 = row(colVal("c1", 1));
Row mutateRow1 = row(colVal("c2", "c2_val"), colVal("c3", ts_1));
Row mutateRow2 = row(colVal("c2", "c2_val_1"), colVal("c3", ts_2));

// insert
MutationResult res = client.insert(tableName).setRowKey(rowKey1).addMutateRow(mutateRow1).execute();
assertEquals(1, res.getAffectedRows());
// get
Map<String, Object> getRes = client.get(tableName, rowKey1.getValues(), null);
assertTrue(!getRes.isEmpty());
assertEquals(1, getRes.get("c1"));
assertEquals("c2_val", getRes.get("c2"));
assertEquals(ts_1, getRes.get("c3"));

// insertup: update
res = client.insertOrUpdate(tableName).setRowKey(rowKey1).addMutateRow(mutateRow2).execute();
assertEquals(1, res.getAffectedRows());
// get
getRes = client.get(tableName, rowKey1.getValues(), null);
assertTrue(!getRes.isEmpty());
assertEquals(1, getRes.get("c1"));
assertEquals("c2_val_1", getRes.get("c2"));
assertEquals(ts_2, getRes.get("c3"));

// replace: update
res = client.replace(tableName).setRowKey(rowKey1).addMutateRow(mutateRow1).execute();
assertEquals(2, res.getAffectedRows());
// get
getRes = client.get(tableName, rowKey1.getValues(), null);
assertTrue(!getRes.isEmpty());
assertEquals(1, getRes.get("c1"));
assertEquals("c2_val", getRes.get("c2"));
assertEquals(ts_1, getRes.get("c3"));

// update
res = client.update(tableName).setRowKey(rowKey1).addMutateRow(mutateRow2).execute();
assertEquals(1, res.getAffectedRows());
// get
getRes = client.get(tableName, rowKey1.getValues(), null);
assertTrue(!getRes.isEmpty());
assertEquals(1, getRes.get("c1"));
assertEquals("c2_val_1", getRes.get("c2"));
assertEquals(ts_2, getRes.get("c3"));
} catch (Exception ex) {
ex.printStackTrace();
assertTrue(false);
} finally {
ObTableClientTestUtil.getConnection().createStatement().execute("truncate table "+ tableName);
}
}
}
30 changes: 30 additions & 0 deletions src/test/resources/ci.sql
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -535,4 +535,34 @@ CREATE TABLE `test_batch_get` (
PRIMARY KEY (`c1`, `c2`)
) partition by key(`c2`) partitions 3;

CREATE TABLE `test_local_index_with_vgen_col` (
`name` varchar(512) NOT NULL DEFAULT '',
`pk` varchar(512) NOT NULL,
`adiu` varchar(512) NOT NULL DEFAULT '',
`id` bigint(20) NOT NULL DEFAULT 0,
`name_v` varchar(20) GENERATED ALWAYS AS (substr(`name`,1,5)) VIRTUAL,
`gmt_create` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
PRIMARY KEY (`adiu`, `pk`, `gmt_create`),
KEY `idx_adiu_v_name` (`adiu`, `name_v`) BLOCK_SIZE 16384 LOCAL
) TTL (gmt_create + INTERVAL 300 SECOND) partition by key(adiu) partitions 8;

CREATE TABLE `test_global_index_with_vgen_col` (
`name` varchar(512) NOT NULL DEFAULT '',
`pk` varchar(512) NOT NULL,
`adiu` varchar(512) NOT NULL DEFAULT '',
`id` bigint(20) NOT NULL DEFAULT 0,
`name_v` varchar(20) GENERATED ALWAYS AS (substr(`name`,1,5)) VIRTUAL,
`gmt_create` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
PRIMARY KEY (`adiu`, `pk`, `gmt_create`),
KEY `idx_adiu_v_name` (`adiu`) global
) TTL (gmt_create + INTERVAL 300 SECOND) partition by key(adiu) partitions 8;

CREATE TABLE `test_current_timestamp` (
`c1` int not null,
`c2` varchar(255),
`c3` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
PRIMARY KEY (`c1`),
KEY `idx_adiu_v_name` (`c2`, `c3`) global partition by key(`c2`) partitions 5
) partition by key(`c1`) partitions 8;

alter system set kv_hotkey_throttle_threshold = 50;