storage_test.go

package sqlite3utils

import (
	"fmt"
	"os"
	"os/exec"
	"path/filepath"
	"regexp"
	"testing"

	"github.com/stretchr/testify/assert"
)

func execSQLite(filename string, queries []string) {
	script, _ := filepath.Abs("./script/sqlite.rb")
	escape := regexp.MustCompile(`'`)
	for _, q := range queries {
		q = escape.ReplaceAllString(q, "\\'")
		//fmt.Print("Query> " + q)
		//out, err := exec.Command("ruby", script, filename, q).Output()
		err := exec.Command("ruby", script, filename, q).Run()
		//fmt.Print("Result> " + string(out))
		if err != nil {
			panic(err)
		}
	}
}

func rmSQLite(filename string) {
	_, err := os.Stat(filename)
	if err == nil {
		err := exec.Command("rm", filename).Run()
		if err != nil {
			panic(err)
		}
	}
}

func TestSimpleLoad(t *testing.T) {
	filename := "/tmp/test.db"
	rmSQLite(filename)

	execSQLite(filename, []string{
		"CREATE TABLE person(id integer, name text, hp integer);",
		"INSERT INTO person VALUES (1, \"hoge\", 10);",
		"INSERT INTO person VALUES (2, \"foo\", 100);",
		"INSERT INTO person VALUES (3, \"bar\", 1000);",
	})

	pages, _ := Load(filename)
	assert.Equal(t, pages.Tables["person"].Entries[0].Datas[0].Value, "1", "Should be same")
	assert.Equal(t, pages.Tables["person"].Entries[0].Datas[1].Value, "hoge", "Should be same")
	assert.Equal(t, pages.Tables["person"].Entries[1].Datas[0].Value, "2", "Should be same")
	assert.Equal(t, pages.Tables["person"].Entries[1].Datas[1].Value, "foo", "Should be same")
	assert.Equal(t, pages.Tables["person"].Entries[2].Datas[0].Value, "3", "Should be same")
	assert.Equal(t, pages.Tables["person"].Entries[2].Datas[1].Value, "bar", "Should be same")

	//pp.Println(pages.Tables["person"])

	rmSQLite(filename)
}

/*
func TestSvn(t *testing.T) {
	//filename := "/home/vagrant/simple.wc.db"
	filename := "/home/vagrant/wc.db"

	//Load(filename)
	pages, _ := Load(filename)

	for _, e := range pages.Tables["NODES"].Entries {
		pp.Println(e.Datas[5].Value)
	}
	//pp.Println(len(pages.Tables["NODES"].Entries))
}
*/

func TestOverflow(t *testing.T) {
	filename := "/tmp/test.db"
	//rmSQLite(filename)

	cmd := []string{"CREATE TABLE person(id integer, name text);"}

	//for i := 0; i < 4096; i++ {
	//for i := 0; i < 128; i++ {
	for i := 1; i < 10000000; i++ {
		cmd = append(cmd, fmt.Sprintf("INSERT INTO person VALUES (%d, \"abc\");", i))
	}

	//execSQLite(filename, cmd)

	Load(filename)
	//pages, _ := Load(filename)
	//pp.Print(pages.Header)
	//pp.Print(pages)

	//rmSQLite(filename)
}