1 files changed, 32 insertions, 0 deletions
diff --git a/spec/migrations/add_new_post_eoa_plans_spec.rb b/spec/migrations/add_new_post_eoa_plans_spec.rb
new file mode 100644
index 00000000000..5ae227a6617
--- /dev/null
+++ b/spec/migrations/add_new_post_eoa_plans_spec.rb
@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+require Rails.root.join('db', 'post_migrate', '20210205104425_add_new_post_eoa_plans.rb')
+
+RSpec.describe AddNewPostEoaPlans do
+  let(:plans) { table(:plans) }
+
+  subject(:migration) { described_class.new }
+
+  describe '#up' do
+    it 'creates the two new records' do
+      expect { migration.up }.to change { plans.count }.by(2)
+
+      new_plans = plans.last(2)
+      expect(new_plans.map(&:name)).to contain_exactly('premium', 'ultimate')
+    end
+  end
+
+  describe '#down' do
+    it 'removes these two new records' do
+      plans.create!(name: 'premium', title: 'Premium (Formerly Silver)')
+      plans.create!(name: 'ultimate', title: 'Ultimate (Formerly Gold)')
+
+      expect { migration.down }.to change { plans.count }.by(-2)
+
+      expect(plans.find_by(name: 'premium')).to be_nil
+      expect(plans.find_by(name: 'ultimate')).to be_nil
+    end
+  end
+end