mirror of
https://github.com/status-im/migrate.git
synced 2025-02-23 16:28:08 +00:00
6c96ef02df
* Enabled maligned check
* Enabled staticcheck
* Fixes for golint
* Enabled errcheck linter
* Added fixes for error check
* Added errcheck for tests
* Fixed test
* Increased golangci-lint deadline for travis
* Increased golangci-lint deadline for travis
* Decreased golangci-lint deadline for travis
* Revert for backward compatibility
* Using log.Println() instead of fmt.Println()
* Handling os.RemoveAll() errors
* Using t.Error(error) instead of t.Errorf("%v", err)
* Using t.Fatal(error) instead of t.Fatalf("%v", err)
* Using fmt.Sprint(sum) instead of t.Srintf("%v", sum)
* Refactoring
* Revert for backward compatibility
* Revert
* go mod tidy
* Added error logging
* Added error logging
* Added error handling
* Added error handling
* Added error logging
* Fix error logging
* Added error handling
* Fix
* Added logging for migr.Buffer()
* Fixes
* Firebird test disabled
* Fixed nolint comment
* Updated firebird docker image version
* Disabled test for firebird 2.5
* Fixed // nolint
107 lines
2.6 KiB
Go
107 lines
2.6 KiB
Go
package cassandra
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/golang-migrate/migrate/v4"
|
|
"strconv"
|
|
"testing"
|
|
)
|
|
|
|
import (
|
|
"github.com/dhui/dktest"
|
|
"github.com/gocql/gocql"
|
|
)
|
|
|
|
import (
|
|
dt "github.com/golang-migrate/migrate/v4/database/testing"
|
|
"github.com/golang-migrate/migrate/v4/dktesting"
|
|
_ "github.com/golang-migrate/migrate/v4/source/file"
|
|
)
|
|
|
|
var (
|
|
opts = dktest.Options{PortRequired: true, ReadyFunc: isReady}
|
|
// Supported versions: http://cassandra.apache.org/download/
|
|
// Although Cassandra 2.x is supported by the Apache Foundation,
|
|
// the migrate db driver only supports Cassandra 3.x since it uses
|
|
// the system_schema keyspace.
|
|
specs = []dktesting.ContainerSpec{
|
|
{ImageName: "cassandra:3.0", Options: opts},
|
|
{ImageName: "cassandra:3.11", Options: opts},
|
|
}
|
|
)
|
|
|
|
func isReady(ctx context.Context, c dktest.ContainerInfo) bool {
|
|
// Cassandra exposes 5 ports (7000, 7001, 7199, 9042 & 9160)
|
|
// We only need the port bound to 9042
|
|
ip, portStr, err := c.Port(9042)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
port, err := strconv.Atoi(portStr)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
|
|
cluster := gocql.NewCluster(ip)
|
|
cluster.Port = port
|
|
cluster.Consistency = gocql.All
|
|
p, err := cluster.CreateSession()
|
|
if err != nil {
|
|
return false
|
|
}
|
|
defer p.Close()
|
|
// Create keyspace for tests
|
|
if err = p.Query("CREATE KEYSPACE testks WITH REPLICATION = {'class': 'SimpleStrategy', 'replication_factor':1}").Exec(); err != nil {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func Test(t *testing.T) {
|
|
dktesting.ParallelTest(t, specs, func(t *testing.T, c dktest.ContainerInfo) {
|
|
ip, port, err := c.Port(9042)
|
|
if err != nil {
|
|
t.Fatal("Unable to get mapped port:", err)
|
|
}
|
|
addr := fmt.Sprintf("cassandra://%v:%v/testks", ip, port)
|
|
p := &Cassandra{}
|
|
d, err := p.Open(addr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer func() {
|
|
if err := d.Close(); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}()
|
|
dt.Test(t, d, []byte("SELECT table_name from system_schema.tables"))
|
|
})
|
|
}
|
|
|
|
func TestMigrate(t *testing.T) {
|
|
dktesting.ParallelTest(t, specs, func(t *testing.T, c dktest.ContainerInfo) {
|
|
ip, port, err := c.Port(9042)
|
|
if err != nil {
|
|
t.Fatal("Unable to get mapped port:", err)
|
|
}
|
|
addr := fmt.Sprintf("cassandra://%v:%v/testks", ip, port)
|
|
p := &Cassandra{}
|
|
d, err := p.Open(addr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer func() {
|
|
if err := d.Close(); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}()
|
|
|
|
m, err := migrate.NewWithDatabaseInstance("file://./examples/migrations", "testks", d)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
dt.TestMigrate(t, m, []byte("SELECT table_name from system_schema.tables"))
|
|
})
|
|
}
|