mobi_test.go

package mobi_test

import (
	"bytes"
	"encoding/binary"
	"testing"
	"time"

	"github.com/leotaku/mobi/pdb"
)

func TestPDBHeaderLength(t *testing.T) {
	length := measure(pdb.PalmDBHeader{})
	assertEq(t, length, pdb.PalmDBHeaderLength)
}

func TestRecordHeaderLength(t *testing.T) {
	length := measure(pdb.RecordHeader{})
	assertEq(t, length, pdb.RecordHeaderLength)
}

func TestReadWrite(t *testing.T) {
	// Write
	w := bytes.NewBuffer(nil)
	db := pdb.NewDatabase("Test Book", time.Unix(0, 0))
	db.AddRecord(pdb.RawRecord("dog"))
	db.AddRecord(pdb.RawRecord("cat"))
	db.AddRecord(pdb.RawRecord("fish"))
	db.AddRecord(pdb.RawRecord("penguin"))
	err := db.Write(w)
	if err != nil {
		t.Fatal(err)
	}

	// Read
	r := bytes.NewReader(w.Bytes())
	rdb, _ := pdb.ReadDatabase(r)

	// Compare
	assertEq(t, db.Name, rdb.Name)
	assertEq(t, len(db.Records), len(rdb.Records))
	for i := 0; i < len(db.Records); i++ {
		bs := writeRecord(db.Records[i])
		rbs := writeRecord(rdb.Records[i])
		assertEq(t, string(bs), string(rbs))
	}
}

func assertEq(t *testing.T, v1 interface{}, v2 interface{}) {
	if v1 != v2 {
		t.Errorf("Not equal: %v, %v", v1, v2)
	}
}

func writeRecord(r pdb.Record) []byte {
	buf := bytes.NewBuffer(nil)
	err := r.Write(buf)
	if err != nil {
		panic(err)
	}

	return buf.Bytes()
}

func measure(v interface{}) int {
	buf := bytes.NewBuffer(nil)
	err := binary.Write(buf, pdb.Endian, v)
	if err != nil {
		panic(err)
	}

	return len(buf.Bytes())
}