mirror of
https://github.com/dolthub/dolt.git
synced 2026-05-04 03:11:52 -05:00
smoke test working
This commit is contained in:
@@ -15,11 +15,25 @@
|
||||
package transactions
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
_ "github.com/go-sql-driver/mysql"
|
||||
"github.com/gocraft/dbr/v2"
|
||||
)
|
||||
|
||||
var defaultConfig = ServerConfig{
|
||||
database: "mysql",
|
||||
host: "127.0.0.1",
|
||||
port: 3316,
|
||||
user: "root",
|
||||
password: "toor",
|
||||
}
|
||||
|
||||
func TestConcurrentTransactions(t *testing.T) {
|
||||
for _, test := range txTests {
|
||||
t.Run(test.name, func(t *testing.T) {
|
||||
@@ -29,13 +43,128 @@ func TestConcurrentTransactions(t *testing.T) {
|
||||
}
|
||||
|
||||
type ConcurrentTxTest struct {
|
||||
name string
|
||||
name string
|
||||
queries []concurrentQuery
|
||||
}
|
||||
|
||||
type concurrentQuery struct {
|
||||
conn string
|
||||
write string
|
||||
query selector
|
||||
expected []testRow
|
||||
}
|
||||
|
||||
type selector func(s *dbr.Session) *dbr.SelectStmt
|
||||
|
||||
type testRow struct {
|
||||
Pk, C0 int
|
||||
}
|
||||
|
||||
const (
|
||||
one = "one"
|
||||
two = "two"
|
||||
)
|
||||
|
||||
var txTests = []ConcurrentTxTest{
|
||||
{name: "todo"},
|
||||
{
|
||||
name: "smoke test",
|
||||
queries: []concurrentQuery{
|
||||
{
|
||||
conn: one,
|
||||
query: func(s *dbr.Session) *dbr.SelectStmt {
|
||||
return s.Select("*").From("data")
|
||||
},
|
||||
expected: []testRow{
|
||||
{1, 1},
|
||||
{2, 2},
|
||||
{3, 3},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
func setupCommon(sess *dbr.Session) (err error) {
|
||||
queries := []string{
|
||||
"CREATE DATABASE IF NOT EXISTS tx;",
|
||||
"USE tx;",
|
||||
"CREATE TABLE data (pk int primary key, c0 int);",
|
||||
"INSERT INTO data VALUES (1,1),(2,2),(3,3);",
|
||||
}
|
||||
|
||||
for _, q := range queries {
|
||||
if _, err = sess.Exec(q); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func testConcurrentTx(t *testing.T, test ConcurrentTxTest) {
|
||||
assert.True(t, true)
|
||||
conns, err := createNamedConnections(defaultConfig, one, two)
|
||||
require.NoError(t, err)
|
||||
defer func() { require.NoError(t, closeNamedConnections(conns)) }()
|
||||
|
||||
err = setupCommon(conns[one])
|
||||
defer func() { require.NoError(t, teardownCommon(conns[one])) }()
|
||||
|
||||
for _, q := range test.queries {
|
||||
conn := conns[q.conn]
|
||||
if q.write != "" {
|
||||
_, err = conn.Query(q.write)
|
||||
require.NoError(t, err)
|
||||
}
|
||||
|
||||
var actual []testRow
|
||||
_, err = q.query(conn).Load(&actual)
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, q.expected, actual)
|
||||
}
|
||||
}
|
||||
|
||||
func teardownCommon(sess *dbr.Session) (err error) {
|
||||
_, err = sess.Exec("DROP DATABASE tx;")
|
||||
return
|
||||
}
|
||||
|
||||
type ServerConfig struct {
|
||||
database string
|
||||
host string
|
||||
port int
|
||||
user string
|
||||
password string
|
||||
}
|
||||
|
||||
type namedConnections map[string]*dbr.Session
|
||||
|
||||
// ConnectionString returns a Data Source Name (DSN) to be used by go clients for connecting to a running server.
|
||||
func ConnectionString(config ServerConfig) string {
|
||||
return fmt.Sprintf("%v:%v@tcp(%v:%v)/%s",
|
||||
config.user,
|
||||
config.password,
|
||||
config.host,
|
||||
config.port,
|
||||
config.database,
|
||||
)
|
||||
}
|
||||
|
||||
func createNamedConnections(config ServerConfig, names ...string) (nc namedConnections, err error) {
|
||||
nc = make(namedConnections, len(names))
|
||||
for _, name := range names {
|
||||
var c *dbr.Connection
|
||||
if c, err = dbr.Open("mysql", ConnectionString(config), nil); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
nc[name] = c.NewSession(nil)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func closeNamedConnections(nc namedConnections) (err error) {
|
||||
for _, conn := range nc {
|
||||
if err = conn.Close(); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
@@ -6,6 +6,7 @@ require github.com/go-sql-driver/mysql v1.6.0
|
||||
|
||||
require (
|
||||
github.com/davecgh/go-spew v1.1.0 // indirect
|
||||
github.com/gocraft/dbr/v2 v2.7.3 // indirect
|
||||
github.com/pmezard/go-difflib v1.0.0 // indirect
|
||||
github.com/stretchr/testify v1.7.0 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c // indirect
|
||||
|
||||
@@ -2,6 +2,8 @@ github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8
|
||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/go-sql-driver/mysql v1.6.0 h1:BCTh4TKNUYmOmMUcQ3IipzF5prigylS7XXjEkfCHuOE=
|
||||
github.com/go-sql-driver/mysql v1.6.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
|
||||
github.com/gocraft/dbr/v2 v2.7.3 h1:5/PTRiBkdD2FoHpnrCMoEUw5Wf/Cl3l3PjJ02Wm+pwM=
|
||||
github.com/gocraft/dbr/v2 v2.7.3/go.mod h1:8IH98S8M8J0JSEiYk0MPH26ZDUKemiQ/GvmXL5jo+Uw=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
|
||||
Reference in New Issue
Block a user