mirror of
https://github.com/chaisql/chai.git
synced 2025-10-24 16:10:30 +08:00
Test Insert method
This commit is contained in:
66
db_test.go
Normal file
66
db_test.go
Normal file
@@ -0,0 +1,66 @@
|
|||||||
|
package genji
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/asdine/genji/engine/memory"
|
||||||
|
"github.com/asdine/genji/field"
|
||||||
|
"github.com/asdine/genji/record"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestTable(t *testing.T) {
|
||||||
|
db, err := Open(memory.NewEngine())
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
t.Run("Table/Insert/NoIndex", func(t *testing.T) {
|
||||||
|
tx, err := db.Begin(true)
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer tx.Rollback()
|
||||||
|
|
||||||
|
tb, err := tx.CreateTable("test")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
rowid, err := tb.Insert(record.FieldBuffer([]field.Field{
|
||||||
|
field.NewString("name", "John"),
|
||||||
|
field.NewInt64("age", 10),
|
||||||
|
}))
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.NotNil(t, rowid)
|
||||||
|
|
||||||
|
m, err := tx.Indexes("test")
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Empty(t, m)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("Table/Insert/WithIndex", func(t *testing.T) {
|
||||||
|
tx, err := db.Begin(true)
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer tx.Rollback()
|
||||||
|
|
||||||
|
tb, err := tx.CreateTable("test")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
_, err = tx.CreateIndex("test", "name")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
rowid, err := tb.Insert(record.FieldBuffer([]field.Field{
|
||||||
|
field.NewString("name", "John"),
|
||||||
|
field.NewInt64("age", 10),
|
||||||
|
}))
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.NotNil(t, rowid)
|
||||||
|
|
||||||
|
m, err := tx.Indexes("test")
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.NotEmpty(t, m)
|
||||||
|
|
||||||
|
c := m["name"].Cursor()
|
||||||
|
v, rid := c.Seek([]byte("John"))
|
||||||
|
require.Equal(t, []byte("John"), v)
|
||||||
|
require.Equal(t, rowid, rid)
|
||||||
|
})
|
||||||
|
|
||||||
|
err = db.Close()
|
||||||
|
require.NoError(t, err)
|
||||||
|
}
|
||||||
@@ -24,5 +24,5 @@ type Transaction interface {
|
|||||||
CreateTable(name string) (table.Table, error)
|
CreateTable(name string) (table.Table, error)
|
||||||
Index(table, name string) (index.Index, error)
|
Index(table, name string) (index.Index, error)
|
||||||
Indexes(table string) (map[string]index.Index, error)
|
Indexes(table string) (map[string]index.Index, error)
|
||||||
CreateIndex(table, name string) (index.Index, error)
|
CreateIndex(table, field string) (index.Index, error)
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user