db_test.go

package yaorm_test

import (
	"context"
	"os"
	"testing"

	"github.com/geoffreybauduin/yaorm"
	_ "github.com/geoffreybauduin/yaorm/testdata"
	"github.com/stretchr/testify/assert"
)

func TestRegisterDB(t *testing.T) {
	defer func() {
		os.Remove("/tmp/test_test.sqlite")
		yaorm.UnregisterDB("test")
	}()
	err := yaorm.RegisterDB(&yaorm.DatabaseConfiguration{
		Name:             "test",
		DSN:              "/tmp/test_test.sqlite",
		System:           yaorm.DatabaseSqlite3,
		AutoCreateTables: true,
	})
	assert.Nil(t, err)
}

func TestRegisterDB_Conflicts(t *testing.T) {
	defer func() {
		os.Remove("/tmp/test_test.sqlite")
		yaorm.UnregisterDB("test")
	}()
	err := yaorm.RegisterDB(&yaorm.DatabaseConfiguration{
		Name:             "test",
		DSN:              "/tmp/test_test.sqlite",
		System:           yaorm.DatabaseSqlite3,
		AutoCreateTables: true,
	})
	assert.Nil(t, err)
	err = yaorm.RegisterDB(&yaorm.DatabaseConfiguration{
		Name:             "test",
		DSN:              "/tmp/test_test.sqlite",
		System:           yaorm.DatabaseSqlite3,
		AutoCreateTables: true,
	})
	assert.Equal(t, yaorm.ErrDatabaseConflict, err)
}

func TestUnregisterDB(t *testing.T) {
	defer func() {
		os.Remove("/tmp/test_test.sqlite")
		yaorm.UnregisterDB("test")
	}()
	err := yaorm.RegisterDB(&yaorm.DatabaseConfiguration{
		Name:             "test",
		DSN:              "/tmp/test_test.sqlite",
		System:           yaorm.DatabaseSqlite3,
		AutoCreateTables: true,
	})
	assert.Nil(t, err)

	err = yaorm.UnregisterDB("test")
	assert.Nil(t, err)
}

func TestUnregisterDB_NotFound(t *testing.T) {
	err := yaorm.UnregisterDB("test")
	assert.Equal(t, yaorm.ErrDbNotFound, err)
}

func TestDb_System(t *testing.T) {
	defer func() {
		os.Remove("/tmp/test_test.sqlite")
		yaorm.UnregisterDB("test")
	}()
	err := yaorm.RegisterDB(&yaorm.DatabaseConfiguration{
		Name:             "test",
		DSN:              "/tmp/test_test.sqlite",
		System:           yaorm.DatabaseSqlite3,
		AutoCreateTables: true,
	})
	assert.Nil(t, err)
	_, err = yaorm.NewDBProvider(context.TODO(), "test")
	assert.Nil(t, err)
	db, err := yaorm.GetDB("test")
	assert.Nil(t, err)
	assert.Equal(t, yaorm.DatabaseSqlite3, db.System())
}

func TestDb_ExecutorHook(t *testing.T) {
	defer func() {
		os.Remove("/tmp/test_test.sqlite")
		yaorm.UnregisterDB("test")
	}()
	err := yaorm.RegisterDB(&yaorm.DatabaseConfiguration{
		Name:             "test",
		DSN:              "/tmp/test_test.sqlite",
		System:           yaorm.DatabaseSqlite3,
		AutoCreateTables: true,
	})
	assert.Nil(t, err)
	_, err = yaorm.NewDBProvider(context.TODO(), "test")
	assert.Nil(t, err)
	db, err := yaorm.GetDB("test")
	assert.Nil(t, err)
	inst := db.ExecutorHook()
	assert.NotNil(t, inst)
	assert.IsType(t, inst, &yaorm.DefaultExecutorHook{})
}

type customExecutorHook struct {
	yaorm.DefaultExecutorHook
}

func (h customExecutorHook) BeforeSelectOne(ctx context.Context, query string, args ...interface{}) {}
func (h customExecutorHook) AfterSelectOne(ctx context.Context, query string, args ...interface{})  {}

func (h customExecutorHook) BeforeSelect(ctx context.Context, query string, args ...interface{}) {}
func (h customExecutorHook) AfterSelect(ctx context.Context, query string, args ...interface{})  {}

func TestDb_ExecutorHook_Custom(t *testing.T) {
	defer func() {
		os.Remove("/tmp/test_test.sqlite")
		yaorm.UnregisterDB("test")
	}()
	err := yaorm.RegisterDB(&yaorm.DatabaseConfiguration{
		Name:             "test",
		DSN:              "/tmp/test_test.sqlite",
		System:           yaorm.DatabaseSqlite3,
		AutoCreateTables: true,
		ExecutorHook:     customExecutorHook{},
	})
	assert.Nil(t, err)
	_, err = yaorm.NewDBProvider(context.TODO(), "test")
	assert.Nil(t, err)
	db, err := yaorm.GetDB("test")
	assert.Nil(t, err)
	inst := db.ExecutorHook()
	assert.NotNil(t, inst)
	assert.IsType(t, inst, &customExecutorHook{})
}