SWAMPy

✎ CARVE FLESH

EDITING: migration_spec.rb

# frozen_string_literal: true

#
# Copyright (C) 2014 - present Instructure, Inc.
#
# This file is part of Canvas.
#
# Canvas is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the Free
# Software Foundation, version 3 of the License.
#
# Canvas is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
# A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
# details.
#
# You should have received a copy of the GNU Affero General Public License along
# with this program. If not, see <http://www.gnu.org/licenses/>.

describe CanvasPartman::Migration do
  require "fixtures/db/20141103000000_add_foo_to_partman_animals"
  require "fixtures/db/20141103000001_add_bar_to_partman_animals"
  require "fixtures/db/20141103000002_remove_foo_from_partman_animals"
  require "fixtures/db/20141103000003_add_another_thing_to_partman_animals"
  require "fixtures/db/20141103000004_add_race_index_to_partman_animals"

it "does nothing with no partitions" do
    Animal.transaction do
      AddFooToPartmanAnimals.new.migrate(:up)

expect(
        connection.column_exists?("partman_animals", "foo")
      ).to be_falsy

AddFooToPartmanAnimals.new.migrate(:down)
    end
  end

it "applies a migration on all partition tables" do
    partman = CanvasPartman::PartitionManager.create(Animal)
    partman.create_partition(Time.new(2014, 11))

Animal.transaction do
      AddFooToPartmanAnimals.new.migrate(:up)

expect(
        connection.column_exists?("partman_animals_2014_11", "foo")
      ).to be_truthy

AddFooToPartmanAnimals.new.migrate(:down)
    end
  end

it "applies multiple migrations" do
    partman = CanvasPartman::PartitionManager.create(Animal)
    partman.create_partition(Time.new(2014, 11))

Animal.transaction do
      AddFooToPartmanAnimals.new.migrate(:up)
      expect(
        connection.column_exists?("partman_animals_2014_11", "foo")
      ).to be_truthy

AddBarToPartmanAnimals.new.migrate(:up)
      expect(
        connection.column_exists?("partman_animals_2014_11", "bar")
      ).to be_truthy

RemoveFooFromPartmanAnimals.new.migrate(:up)
      expect(
        connection.column_exists?("partman_animals_2014_11", "foo")
      ).to be_falsy

RemoveFooFromPartmanAnimals.new.migrate(:down)
      expect(
        connection.column_exists?("partman_animals_2014_11", "foo")
      ).to be_truthy

AddBarToPartmanAnimals.new.migrate(:down)
      expect(
        connection.column_exists?("partman_animals_2014_11", "bar")
      ).to be_falsy

AddFooToPartmanAnimals.new.migrate(:down)
      expect(
        connection.column_exists?("partman_animals_2014_11", "foo")
      ).to be_falsy
    end
  end

it "applies multiple migrations on multiple partitions" do
    partman = CanvasPartman::PartitionManager.create(Animal)
    partman.create_partition(Time.new(2014, 11))
    partman.create_partition(Time.new(2014, 12))

Animal.transaction do
      AddFooToPartmanAnimals.new.migrate(:up)
      expect(
        connection.column_exists?("partman_animals_2014_11", "foo")
      ).to be_truthy
      expect(
        connection.column_exists?("partman_animals_2014_12", "foo")
      ).to be_truthy

AddBarToPartmanAnimals.new.migrate(:up)
      expect(
        connection.column_exists?("partman_animals_2014_11", "bar")
      ).to be_truthy
      expect(
        connection.column_exists?("partman_animals_2014_12", "bar")
      ).to be_truthy

RemoveFooFromPartmanAnimals.new.migrate(:up)
      expect(
        connection.column_exists?("partman_animals_2014_11", "foo")
      ).to be_falsy
      expect(
        connection.column_exists?("partman_animals_2014_12", "foo")
      ).to be_falsy

RemoveFooFromPartmanAnimals.new.migrate(:down)
      expect(
        connection.column_exists?("partman_animals_2014_11", "foo")
      ).to be_truthy
      expect(
        connection.column_exists?("partman_animals_2014_12", "foo")
      ).to be_truthy

AddBarToPartmanAnimals.new.migrate(:down)
      expect(
        connection.column_exists?("partman_animals_2014_11", "bar")
      ).to be_falsy
      expect(
        connection.column_exists?("partman_animals_2014_12", "bar")
      ).to be_falsy

AddFooToPartmanAnimals.new.migrate(:down)
      expect(
        connection.column_exists?("partman_animals_2014_11", "foo")
      ).to be_falsy
      expect(
        connection.column_exists?("partman_animals_2014_12", "foo")
      ).to be_falsy
    end
  end

it "accepts an explicitly specified base class" do
    partman = CanvasPartman::PartitionManager.create(CanvasPartmanTest::AnimalAlias)
    partman.create_partition(Time.new(2014, 11))

Animal.transaction do
      AddAnotherThingToPartmanAnimals.new.migrate(:up)

expect(
        connection.column_exists?("partman_animals_2014_11", "another_thing")
      ).to be_truthy

AddAnotherThingToPartmanAnimals.new.migrate(:down)
    end
  end

it "add/removes indices just fine" do
    partman = CanvasPartman::PartitionManager.create(Animal)
    partman.create_partition(Time.new(2014, 11))

Animal.transaction do
      AddRaceIndexToPartmanAnimals.new.migrate(:up)

expect(
        connection.index_exists?("partman_animals_2014_11", :race)
      ).to be_truthy

AddRaceIndexToPartmanAnimals.new.migrate(:down)
    end
  end
end

CANCEL

Name	Type	Size	Modified	Actions
↩ ..	DIR	—	—
📁 concerns/	DIR	—	2023-09-24 03:39
📄 migration_spec.rb	RB	5.4 KB	2023-09-24 03:36	EDIT
📄 partition_manager_spec.rb	RB	9.3 KB	2023-09-24 03:36	EDIT