ezdb/ezleveldb/leveldb_test.go

package ezleveldb

import (
	"fmt"
	"testing"

	"github.com/annybs/ezdb"
)

func TestLevelDB(t *testing.T) {
	type Student struct {
		Name string `json:"name"`
		Age  int    `json:"age"`
	}

	path := ".leveldb/test_json"

	marshaler := ezdb.JSON(func() *Student {
		return &Student{}
	})

	collection := ezdb.LevelDB(path, marshaler, nil)

	cmp := func(expected, actual *Student) error {
		if expected == nil && actual != nil {
			return fmt.Errorf("expected no student (got %v)", actual)
		} else if expected != nil && actual == nil {
			return fmt.Errorf("expected student %v (got nil)", expected)
		}
		if actual.Name != expected.Name {
			return fmt.Errorf("student has wrong name (expected '%s', got '%s')", expected.Name, actual.Name)
		} else if actual.Age != expected.Age {
			return fmt.Errorf("student has wrong age (expected '%d', got '%d')", expected.Age, actual.Age)
		}
		return nil
	}

	tester := &ezdb.CollectionTester[*Student]{
		C: collection,

		Cmp: cmp,
		Data: map[string]*Student{
			"annie": {Name: "Annie", Age: 32},
			"ben":   {Name: "Ben", Age: 50},
			"clive": {Name: "Clive", Age: 21},
		},
	}

	named := func(name string, do func() error) func() {
		return func() {
			if err := do(); err != nil {
				t.Errorf("failed test %q: %v", name, err)
			} else {
				t.Logf("passed test %q", name)
			}
		}
	}

	destroy := func() error {
		return collection.Destroy()
	}

	sequence := []func(){
		named("init 1", tester.Init),
		named("has", tester.Has),
		named("get", tester.Get),
		named("getAll 1", tester.GetAll),
		named("deleteOne 1", tester.DeleteOne),
		named("deleteOne 2", tester.DeleteOne),
		named("init 2", tester.Init),
		named("close 1", tester.Close),
		named("open", tester.Open),
		named("getAll 2", tester.GetAll),
		named("iterCount", tester.IterCount),
		named("iterSortKeys", tester.IterSortKeys),
		named("deleteAll", tester.DeleteAll),
		named("close", tester.Close),
		named("destroy", destroy),
	}
	for _, do := range sequence {
		do()
	}
}