func TestSimpleJSONDB_InitializesDataFile(t *testing.T) {
firstDataBlock := make([]byte, 10)
blocksBitMapBlock := make([]byte, dbio.DATABLOCK_SIZE)
bTreeRootBlock := make([]byte, 2)
fakeDataFile := utils.NewFakeDataFileWithBlocks([][]byte{
firstDataBlock,
blocksBitMapBlock,
nil,
nil,
bTreeRootBlock,
})
jsondb.NewWithDataFile(fakeDataFile)
if !utils.SlicesEqual(firstDataBlock[0:2], []byte{0x00, 0x03}) {
t.Error("Did not set the next available data block pointer to 3")
}
if !utils.SlicesEqual(firstDataBlock[2:4], []byte{0x00, 0x03}) {
t.Error("Did not set the first record block pointer to 3")
}
blocksBitMap := dbio.NewBitMapFromBytes(blocksBitMapBlock)
for i := 0; i < 4; i++ {
val, err := blocksBitMap.Get(i)
if err != nil {
t.Fatal(err)
}
if !val {
t.Errorf("Expected block %d to be flagged as used", i)
}
}
}
func TestCreateAndRemoveRecords(t *testing.T) {
fakeDataFile := utils.NewFakeDataFile(20)
db, err := jsondb.NewWithDataFile(fakeDataFile)
if err != nil {
t.Fatalf("Unexpected error returned '%s'", err)
}
for i := 0; i < 10; i++ {
id := uint32(i + 1)
data := fmt.Sprintf(`{"a":%d}`, i)
err := db.InsertRecord(id, data)
if err != nil {
t.Fatalf("Unexpected error returned when inserting '%s'", err)
}
err = db.DeleteRecord(id)
if err != nil {
t.Fatalf("Unexpected error returned when removing '%s'", err)
}
if record, err := db.FindRecord(id); err == nil {
t.Errorf("Expected error to be returned when finding %d, got nil and data '%s'", id, string(record.Data))
}
}
}
func TestSearchByTag(t *testing.T) {
fakeDataFile := utils.NewFakeDataFile(20)
db, err := jsondb.NewWithDataFile(fakeDataFile)
if err != nil {
t.Fatalf("Unexpected error returned '%s'", err)
}
// Insert some data
states := []string{"RS", "BA", "SC"}
for i := 0; i < 10; i++ {
id := uint32(i + 1)
data := fmt.Sprintf(`{"id":%d,"state":"%s"}`, id, states[i%len(states)])
err := db.InsertRecord(id, data)
if err != nil {
t.Fatalf("Unexpected error returned when inserting '%s'", err)
}
}
// Search by state
result, err := db.SearchRecords("state", "RS")
if err != nil {
t.Fatal(err)
}
if len(result) != 4 {
t.Errorf("Unexpected results found, expected 4 items, got %d", len(result))
}
for i, record := range result {
document, err := record.ParseJSON()
if err != nil {
t.Fatal(err)
}
if document["state"].(string) != "RS" {
t.Errorf("Invalid document returned with state != `RS`: %v", document)
} else if document["id"].(float64) != float64(i*3)+1 {
t.Errorf("Invalid document returned: %v", document)
}
}
}
func TestCreateAndRetrieveLotsOfRecords(t *testing.T) {
fakeDataFile := utils.NewFakeDataFile(80)
db, err := jsondb.NewWithDataFile(fakeDataFile)
if err != nil {
t.Fatalf("Unexpected error returned '%s'", err)
}
for i := 0; i < 4500; i++ {
id := uint32(i + 1)
data := fmt.Sprintf(`{"a":%d}`, i)
err := db.InsertRecord(id, data)
if err != nil {
t.Fatalf("Unexpected error returned for the %d-th record: '%s'", id, err)
}
record, err := db.FindRecord(id)
if err != nil {
t.Fatalf("Unexpected error returned while reading %d (%s)", id, err)
}
if string(record.Data) != data {
t.Errorf("Unexpected data returned, got %s, expected %s", string(record.Data), data)
}
}
}
func TestCreateAndUpdateRecords(t *testing.T) {
fakeDataFile := utils.NewFakeDataFile(20)
db, err := jsondb.NewWithDataFile(fakeDataFile)
if err != nil {
t.Fatalf("Unexpected error returned '%s'", err)
}
// Insert some data
for i := 0; i < 1000; i++ {
data := fmt.Sprintf(`{"longest":%d}`, i)
id := uint32(i + 1)
err := db.InsertRecord(id, data)
if err != nil {
t.Fatalf("Unexpected error returned when inserting '%s'", err)
}
}
// Shrink records
for i := uint32(0); i < 1000; i++ {
data := fmt.Sprintf(`{"a":%d}`, -int(i))
id := i + 1
err := db.UpdateRecord(id, data)
if err != nil {
t.Errorf("Unexpected error returned when updating record `%d` '%s'", id, err)
continue
}
record, err := db.FindRecord(id)
if err != nil {
t.Errorf("Unexpected error returned while reading %d (%s)", id, err)
continue
}
if string(record.Data) != data {
t.Errorf("Unexpected data returned, got `%s`, expected `%s`", string(record.Data), data)
continue
}
}
// Grow records again
for i := uint32(0); i < 1000; i++ {
data := fmt.Sprintf(`{"waaaaat":%d}`, int(i))
id := i + 1
err := db.UpdateRecord(id, data)
if err != nil {
t.Errorf("Unexpected error returned when updating record `%d` '%s'", id, err)
continue
}
record, err := db.FindRecord(id)
if err != nil {
t.Errorf("Unexpected error returned while reading %d (%s)", id, err)
continue
}
if string(record.Data) != data {
t.Errorf("Unexpected data returned, got `%s`, expected `%s`", string(record.Data), data)
continue
}
}
}