14 lines
336 B
Go
14 lines
336 B
Go
|
package database
|
||
|
|
|
||
|
|
import "testing"
|
||
|
|
|
||
|
|
func TestUpSection(t *testing.T) {
|
||
|
|
got, err := upSection("-- +goose Up\nCREATE TABLE example(id int);\n-- +goose Down\nDROP TABLE example;")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("upSection() error = %v", err)
|
||
|
|
}
|
||
|
|
if got != "\nCREATE TABLE example(id int);\n" {
|
||
|
|
t.Fatalf("unexpected section: %q", got)
|
||
|
|
}
|
||
|
|
}
|