Skip to content
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.

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

add lob test #121

Merged
merged 1 commit into from
Oct 17, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
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
4 changes: 4 additions & 0 deletions test/single/all_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -46,6 +46,8 @@ func setup() {
test.CreateTable(testVarbinaryCreateStatement)
test.CreateTable(testDatetimeCreateStatement)
test.CreateTable(testTimestampCreateStatement)
test.CreateTable(testBlobCreateStatement)
test.CreateTable(testClobCreateStatement)
}

func teardown() {
Expand All @@ -65,6 +67,8 @@ func teardown() {
test.DropTable(testVarbinaryTableName)
test.DropTable(testDatetimeTableName)
test.DropTable(testTimestampTableName)
test.DropTable(testBlobTableName)
test.DropTable(testClobTableName)

test.CloseDB()
}
Expand Down
254 changes: 254 additions & 0 deletions test/single/blob_table_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,254 @@
/*-
* #%L
* OBKV Table Client Framework
* %%
* Copyright (C) 2021 OceanBase
* %%
* OBKV Table Client Framework is licensed under Mulan PSL v2.
* You can use this software according to the terms and conditions of the Mulan PSL v2.
* You may obtain a copy of Mulan PSL v2 at:
* http://license.coscl.org.cn/MulanPSL2
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
* See the Mulan PSL v2 for more details.
* #L%
*/

package single

import (
"context"
"testing"

"github.com/stretchr/testify/assert"

"github.com/oceanbase/obkv-table-client-go/client/option"
"github.com/oceanbase/obkv-table-client-go/table"
"github.com/oceanbase/obkv-table-client-go/test"
)

const (
testBlobTableName = "test_blob"
testBlobCreateStatement = "create table if not exists test_blob(c1 int(12) primary key, c2 int(12) default null, c3 blob not null);"
)

func TestBlob_insert(t *testing.T) {
tableName := testBlobTableName
defer test.DeleteTable(tableName)

// 1. insert
rowKey := []*table.Column{table.NewColumn("c1", int32(0))}
mutateColumns := []*table.Column{table.NewColumn("c3", []byte("abc"))}
affectRows, err := cli.Insert(
context.TODO(),
tableName,
rowKey,
mutateColumns,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, 1, affectRows)

// 2. get
res, err := cli.Get(
context.TODO(),
tableName,
rowKey,
nil,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, 0, res.Value("c1"))
assert.EqualValues(t, []byte("abc"), res.Value("c3"))
}

func TestBlob_delete(t *testing.T) {
tableName := testBlobTableName
defer test.DeleteTable(tableName)

// 1. insert
rowKey := []*table.Column{table.NewColumn("c1", int32(0))}
mutateColumns := []*table.Column{table.NewColumn("c3", []byte("abc"))}
affectRows, err := cli.Insert(
context.TODO(),
tableName,
rowKey,
mutateColumns,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, 1, affectRows)

// 2. delete
affectRows, err = cli.Delete(
context.TODO(),
tableName,
rowKey,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, 1, affectRows)

// 3. get
res, err := cli.Get(
context.TODO(),
tableName,
rowKey,
nil,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, nil, res.Value("c3"))
}

func TestBlob_update(t *testing.T) {
tableName := testBlobTableName
defer test.DeleteTable(tableName)

// 1. insert
rowKey := []*table.Column{table.NewColumn("c1", int32(0))}
mutateColumns := []*table.Column{table.NewColumn("c3", []byte("abc"))}
affectRows, err := cli.Insert(
context.TODO(),
tableName,
rowKey,
mutateColumns,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, 1, affectRows)

// 2. update
mutateColumns = []*table.Column{table.NewColumn("c3", []byte("efg"))}
affectRows, err = cli.Update(
context.TODO(),
tableName,
rowKey,
mutateColumns,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, 1, affectRows)

// 3. get
res, err := cli.Get(
context.TODO(),
tableName,
rowKey,
nil,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, []byte("efg"), res.Value("c3"))
}

func TestBlob_replace(t *testing.T) {
tableName := testBlobTableName
defer test.DeleteTable(tableName)

// 1. insert
rowKey := []*table.Column{table.NewColumn("c1", int32(0))}
mutateColumns := []*table.Column{table.NewColumn("c3", []byte("abc"))}
affectRows, err := cli.Insert(
context.TODO(),
tableName,
rowKey,
mutateColumns,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, 1, affectRows)

// 2. replace
mutateColumns = []*table.Column{table.NewColumn("c3", []byte("efg"))}
affectRows, err = cli.Replace(
context.TODO(),
tableName,
rowKey,
mutateColumns,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, 2, affectRows)

// 3. get
res, err := cli.Get(
context.TODO(),
tableName,
rowKey,
nil,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, []byte("efg"), res.Value("c3"))
}

func TestBlob_insertUp(t *testing.T) {
tableName := testBlobTableName
defer test.DeleteTable(tableName)

// 1. insertUp(insert)
rowKey := []*table.Column{table.NewColumn("c1", int32(0))}
mutateColumns := []*table.Column{table.NewColumn("c3", []byte("abc"))}
affectRows, err := cli.InsertOrUpdate(
context.TODO(),
tableName,
rowKey,
mutateColumns,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, 1, affectRows)

// 2. insertUp(update)
mutateColumns = []*table.Column{table.NewColumn("c3", []byte("efg"))}
affectRows, err = cli.InsertOrUpdate(
context.TODO(),
tableName,
rowKey,
mutateColumns,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, 1, affectRows)

// 3. get
res, err := cli.Get(
context.TODO(),
tableName,
rowKey,
nil,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, 0, res.Value("c1"))
assert.EqualValues(t, []byte("efg"), res.Value("c3"))
}

func TestBlob_append(t *testing.T) {
tableName := testBlobTableName
defer test.DeleteTable(tableName)

// 1. insert
rowKey := []*table.Column{table.NewColumn("c1", int32(0))}
mutateColumns := []*table.Column{table.NewColumn("c3", []byte("abc"))}
affectRows, err := cli.Insert(
context.TODO(),
tableName,
rowKey,
mutateColumns,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, 1, affectRows)

// 2. append, not expired, success
mutateColumns = []*table.Column{table.NewColumn("c3", []byte("efg"))}
res, err := cli.Append(
context.TODO(),
tableName,
rowKey,
mutateColumns,
option.WithReturnAffectedEntity(true),
)
assert.Equal(t, nil, err)
assert.EqualValues(t, 1, affectRows)
assert.EqualValues(t, []byte("abcefg"), res.Value("c3"))

// 3. get
res, err = cli.Get(
context.TODO(),
tableName,
rowKey,
nil,
)
assert.Equal(t, nil, err)
assert.EqualValues(t, 0, res.Value("c1"))
assert.EqualValues(t, []byte("abcefg"), res.Value("c3"))
}
Loading