mirror of https://github.com/VinGarcia/ksql.git
101 lines
1.9 KiB
Go
101 lines
1.9 KiB
Go
package ksql
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"testing"
|
|
|
|
tt "github.com/vingarcia/ksql/internal/testtools"
|
|
"github.com/vingarcia/ksql/sqldialect"
|
|
)
|
|
|
|
func TestScanArgError(t *testing.T) {
|
|
err := ScanArgError{
|
|
ColumnIndex: 12,
|
|
Err: io.EOF,
|
|
}
|
|
|
|
tt.AssertErrContains(t, err, "input attribute", "index 12", "EOF")
|
|
}
|
|
|
|
func TestConfigSetDefaultValues(t *testing.T) {
|
|
config := Config{}
|
|
config.SetDefaultValues()
|
|
|
|
tt.AssertEqual(t, config, Config{
|
|
MaxOpenConns: 1,
|
|
})
|
|
}
|
|
|
|
func TestNewAdapterWith(t *testing.T) {
|
|
t.Run("should build new instances correctly", func(t *testing.T) {
|
|
for _, dialect := range sqldialect.SupportedDialects {
|
|
db, err := NewWithAdapter(
|
|
DBAdapter(nil),
|
|
dialect,
|
|
)
|
|
|
|
tt.AssertNoErr(t, err)
|
|
tt.AssertEqual(t, db.dialect, dialect)
|
|
}
|
|
})
|
|
|
|
t.Run("should report invalid dialects correctly", func(t *testing.T) {
|
|
_, err := NewWithAdapter(
|
|
DBAdapter(nil),
|
|
nil,
|
|
)
|
|
|
|
tt.AssertErrContains(t, err, "expected a valid", "Provider", "nil")
|
|
})
|
|
}
|
|
|
|
func TestClose(t *testing.T) {
|
|
t.Run("should close the adapter if it implements the io.Closer interface", func(t *testing.T) {
|
|
c := DB{
|
|
db: struct {
|
|
DBAdapter
|
|
io.Closer
|
|
}{
|
|
DBAdapter: mockDBAdapter{},
|
|
Closer: mockCloser{
|
|
CloseFn: func() error {
|
|
return nil
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
err := c.Close()
|
|
tt.AssertNoErr(t, err)
|
|
})
|
|
|
|
t.Run("should exit normally if the adapter does not implement the io.Closer interface", func(t *testing.T) {
|
|
c := DB{
|
|
db: mockDBAdapter{},
|
|
}
|
|
|
|
err := c.Close()
|
|
tt.AssertNoErr(t, err)
|
|
})
|
|
|
|
t.Run("should report an error if the adapter.Close() returns one", func(t *testing.T) {
|
|
c := DB{
|
|
db: struct {
|
|
DBAdapter
|
|
io.Closer
|
|
}{
|
|
DBAdapter: mockDBAdapter{},
|
|
Closer: mockCloser{
|
|
CloseFn: func() error {
|
|
return fmt.Errorf("fakeCloseErrMsg")
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
err := c.Close()
|
|
tt.AssertErrContains(t, err, "fakeCloseErrMsg")
|
|
})
|
|
}
|