Adding the challenge 0402

0402/table_test.go

@@ -0,0 +1,126 @@
+package db0402
+
+import (
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestTableByPKey(t *testing.T) {
+	db := DB{}
+	db.KV.log.FileName = ".test_db"
+	defer os.Remove(db.KV.log.FileName)
+
+	os.Remove(db.KV.log.FileName)
+	err := db.Open()
+	assert.Nil(t, err)
+	defer db.Close()
+
+	schema := &Schema{
+		Table: "link",
+		Cols: []Column{
+			{Name: "time", Type: TypeI64},
+			{Name: "src", Type: TypeStr},
+			{Name: "dst", Type: TypeStr},
+		},
+		PKey: []int{1, 2}, // (src, dst)
+	}
+
+	row := Row{
+		Cell{Type: TypeI64, I64: 123},
+		Cell{Type: TypeStr, Str: []byte("a")},
+		Cell{Type: TypeStr, Str: []byte("b")},
+	}
+	ok, err := db.Select(schema, row)
+	assert.True(t, !ok && err == nil)
+
+	updated, err := db.Insert(schema, row)
+	assert.True(t, updated && err == nil)
+
+	out := Row{
+		Cell{},
+		Cell{Type: TypeStr, Str: []byte("a")},
+		Cell{Type: TypeStr, Str: []byte("b")},
+	}
+	ok, err = db.Select(schema, out)
+	assert.True(t, ok && err == nil)
+	assert.Equal(t, row, out)
+
+	row[0].I64 = 456
+	updated, err = db.Update(schema, row)
+	assert.True(t, updated && err == nil)
+
+	ok, err = db.Select(schema, out)
+	assert.True(t, ok && err == nil)
+	assert.Equal(t, row, out)
+
+	deleted, err := db.Delete(schema, row)
+	assert.True(t, deleted && err == nil)
+
+	ok, err = db.Select(schema, row)
+	assert.True(t, !ok && err == nil)
+}
+
+func parseStmt(t *testing.T, s string) interface{} {
+	p := NewParser(s)
+	stmt, err := p.parseStmt()
+	require.Nil(t, err)
+	return stmt
+}
+
+func TestSQLByPKey(t *testing.T) {
+	db := DB{}
+	db.KV.log.FileName = ".test_db"
+	defer os.Remove(db.KV.log.FileName)
+
+	os.Remove(db.KV.log.FileName)
+	err := db.Open()
+	assert.Nil(t, err)
+	defer db.Close()
+
+	s := "create table link (time int64, src string, dst string, primary key (src, dst));"
+	_, err = db.ExecStmt(parseStmt(t, s))
+	require.Nil(t, err)
+
+	s = "insert into link values (123, 'bob', 'alice');"
+	r, err := db.ExecStmt(parseStmt(t, s))
+	require.Nil(t, err)
+	require.Equal(t, 1, r.Updated)
+
+	s = "select time from link where dst = 'alice' and src = 'bob';"
+	r, err = db.ExecStmt(parseStmt(t, s))
+	require.Nil(t, err)
+	require.Equal(t, []Row{{Cell{Type: TypeI64, I64: 123}}}, r.Values)
+
+	s = "update link set time = 456 where dst = 'alice' and src = 'bob';"
+	r, err = db.ExecStmt(parseStmt(t, s))
+	require.Nil(t, err)
+	require.Equal(t, 1, r.Updated)
+
+	s = "select time from link where dst = 'alice' and src = 'bob';"
+	r, err = db.ExecStmt(parseStmt(t, s))
+	require.Nil(t, err)
+	require.Equal(t, []Row{{Cell{Type: TypeI64, I64: 456}}}, r.Values)
+
+	// reopen
+	err = db.Close()
+	require.Nil(t, err)
+	db = DB{}
+	db.KV.log.FileName = ".test_db"
+	err = db.Open()
+	require.Nil(t, err)
+
+	s = "delete from link where src = 'bob' and dst = 'alice';"
+	r, err = db.ExecStmt(parseStmt(t, s))
+	require.Nil(t, err)
+	require.Equal(t, 1, r.Updated)
+
+	s = "select time from link where dst = 'alice' and src = 'bob';"
+	r, err = db.ExecStmt(parseStmt(t, s))
+	require.Nil(t, err)
+	require.Equal(t, 0, len(r.Values))
+}
+
+// QzBQWVJJOUhU https://trialofcode.org/