adapter_test.go

package gdbc

import (
	"database/sql"
	"database/sql/driver"
	"errors"
	"github.com/stretchr/testify/assert"
	"testing"
)

func init() {
	sql.Register("testDriver1", &testDriver1{})
	sql.Register("testDriver2", &testDriver2{})
	Register("testDriver1", "driver1", &testDriver1DataSourceNameAdapter{})
	Register("testDriver2", "driver2", &testDriver2DataSourceNameAdapter{})
}

type testConnection struct {
}

func (conn testConnection) Prepare(query string) (driver.Stmt, error) {
	return nil, nil
}

func (conn testConnection) Close() error {
	return nil
}

func (conn testConnection) Begin() (driver.Tx, error) {
	return nil, nil
}

type testDriver1 struct {
}

func (driver testDriver1) Open(name string) (driver.Conn, error) {
	return &testConnection{}, nil
}

type testDriver1DataSourceNameAdapter struct {
}

func (adapter testDriver1DataSourceNameAdapter) GetDataSourceName(dataSource DataSource) (string, error) {
	return "test", nil
}

type testDriver2 struct {
}

func (driver testDriver2) Open(name string) (driver.Conn, error) {
	return &testConnection{}, nil
}

type testDriver2DataSourceNameAdapter struct {
}

func (adapter testDriver2DataSourceNameAdapter) GetDataSourceName(dataSource DataSource) (string, error) {
	return "", errors.New("test error")
}

func TestRegisterWithNilAdapter(t *testing.T) {
	assert.Panics(t, func() {
		Register("testDriver", "test", nil)
	})
}

func TestRegisterWithSameAliasName(t *testing.T) {
	assert.Panics(t, func() {
		Register("testDriver3", "test", testDriver1DataSourceNameAdapter{})
		Register("testDriver4", "test", testDriver2DataSourceNameAdapter{})
	})
}

func TestRegisterWithSameDriverName(t *testing.T) {
	assert.Panics(t, func() {
		Register("testDriver", "test1", testDriver1DataSourceNameAdapter{})
		Register("testDriver", "test2", testDriver2DataSourceNameAdapter{})
	})
}

func TestRegisterWithEmptyDriverOrDriverAliasName(t *testing.T) {
	assert.Panics(t, func() {
		Register("", "test1", nil)
		Register("testDriver", "", nil)
	})
}

func TestGetDriverName(t *testing.T) {
	driverName, ok := GetDriverName("driver1")
	assert.True(t, ok)
	assert.Equal(t, "testDriver1", driverName)

	driverName, ok = GetDriverName("driver3")
	assert.False(t, ok)
	assert.Equal(t, "", driverName)
}