Current section

Files

Jump to
ecto integration_test cases migrator.exs
Raw

integration_test/cases/migrator.exs

Code.require_file "../support/file_helpers.exs", __DIR__
defmodule Ecto.Integration.MigratorTest do
use Ecto.Integration.Case
import Support.FileHelpers
import Ecto.Migrator, only: [migrated_versions: 1]
alias Ecto.Integration.Barebone
alias Ecto.Integration.TestRepo
alias Ecto.Migration.SchemaMigration
setup do
TestRepo.delete_all(SchemaMigration)
:ok
end
defmodule GoodMigration do
use Ecto.Migration
def up do
:ok
end
def down do
:ok
end
end
defmodule BadMigration do
use Ecto.Migration
def change do
execute "CREATE WHAT"
end
end
import Ecto.Migrator
test "schema migration" do
up(TestRepo, 20080906120000, GoodMigration, log: false)
[migration] = TestRepo.all(SchemaMigration)
assert migration.version == 20080906120000
assert migration.inserted_at
end
test "migrations up and down" do
assert migrated_versions(TestRepo) == []
assert up(TestRepo, 20080906120000, GoodMigration, log: false) == :ok
assert migrated_versions(TestRepo) == [20080906120000]
assert up(TestRepo, 20080906120000, GoodMigration, log: false) == :already_up
assert migrated_versions(TestRepo) == [20080906120000]
assert down(TestRepo, 20080906120001, GoodMigration, log: false) == :already_down
assert migrated_versions(TestRepo) == [20080906120000]
assert down(TestRepo, 20080906120000, GoodMigration, log: false) == :ok
assert migrated_versions(TestRepo) == []
end
test "bad migration" do
assert catch_error(up(TestRepo, 20080906120000, BadMigration, log: false))
end
test "run up to/step migration" do
in_tmp fn path ->
create_migration(47)
create_migration(48)
assert [47] = run(TestRepo, path, :up, step: 1, log: false)
assert count_entries() == 1
assert [48] = run(TestRepo, path, :up, to: 48, log: false)
end
end
test "run down to/step migration" do
in_tmp fn path ->
migrations = [
create_migration(49),
create_migration(50),
]
assert [49, 50] = run(TestRepo, path, :up, all: true, log: false)
purge migrations
assert [50] = run(TestRepo, path, :down, step: 1, log: false)
purge migrations
assert count_entries() == 1
assert [50] = run(TestRepo, path, :up, to: 50, log: false)
end
end
test "runs all migrations" do
in_tmp fn path ->
migrations = [
create_migration(53),
create_migration(54),
]
assert [53, 54] = run(TestRepo, path, :up, all: true, log: false)
assert [] = run(TestRepo, path, :up, all: true, log: false)
purge migrations
assert [54, 53] = run(TestRepo, path, :down, all: true, log: false)
purge migrations
assert count_entries() == 0
assert [53, 54] = run(TestRepo, path, :up, all: true, log: false)
end
end
defp count_entries() do
length TestRepo.all Barebone
end
defp create_migration(num) do
module = Module.concat(__MODULE__, "Migration#{num}")
File.write! "#{num}_migration_#{num}.exs", """
defmodule #{module} do
use Ecto.Migration
import Ecto.Query, only: [from: 2]
alias Ecto.Integration.TestRepo
alias Ecto.Integration.Barebone
def up do
TestRepo.insert!(%Barebone{num: #{num}})
end
def down do
TestRepo.delete_all from b in Barebone, where: b.num == #{num}
end
end
"""
module
end
defp purge(modules) do
Enum.each(List.wrap(modules), fn m ->
:code.delete m
:code.purge m
end)
end
end