mirror of
https://github.com/ent/ent.git
synced 2026-05-04 08:30:57 +03:00
6c20ef1d3a
Reviewed By: alexsn Differential Revision: D17200893 fbshipit-source-id: 09ec631b5a52d617d0ee97bda81c52f67486889e
49 lines
1.3 KiB
Go
49 lines
1.3 KiB
Go
package schema
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/facebookincubator/ent/dialect"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestWriteDriver(t *testing.T) {
|
|
b := &bytes.Buffer{}
|
|
w := WriteDriver{Driver: nopDriver{}, Writer: b}
|
|
tx, err := w.Tx(nil)
|
|
require.NoError(t, err)
|
|
err = tx.Query(nil, "SELECT `name` FROM `users`", nil, nil)
|
|
require.NoError(t, err)
|
|
err = tx.Query(nil, "SELECT `name` FROM `users`", nil, nil)
|
|
require.NoError(t, err)
|
|
err = tx.Exec(nil, "ALTER TABLE `users` ADD COLUMN `age` int", nil, nil)
|
|
require.NoError(t, err)
|
|
err = tx.Exec(nil, "ALTER TABLE `users` ADD COLUMN `NAME` varchar(100);", nil, nil)
|
|
require.NoError(t, err)
|
|
err = tx.Query(nil, "SELECT `name` FROM `users`", nil, nil)
|
|
require.NoError(t, err)
|
|
require.NoError(t, tx.Commit())
|
|
lines := strings.Split(b.String(), "\n")
|
|
require.Equal(t, "BEGIN;", lines[0])
|
|
require.Equal(t, "ALTER TABLE `users` ADD COLUMN `age` int;", lines[1])
|
|
require.Equal(t, "ALTER TABLE `users` ADD COLUMN `NAME` varchar(100);", lines[2])
|
|
require.Equal(t, "COMMIT;", lines[3])
|
|
require.Empty(t, lines[4], "file ends with blank line")
|
|
}
|
|
|
|
type nopDriver struct {
|
|
dialect.Driver
|
|
}
|
|
|
|
func (nopDriver) Exec(context.Context, string, interface{}, interface{}) error {
|
|
return nil
|
|
}
|
|
|
|
func (nopDriver) Query(context.Context, string, interface{}, interface{}) error {
|
|
return nil
|
|
}
|