summaryrefslogtreecommitdiff
path: root/db/migrate/20210908061132_insert_dast_profile_schedules_plan_limits.rb
blob: b9980f65c9859e8f00ccb550a7a94ce67289d5e5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
# frozen_string_literal: true

class InsertDastProfileSchedulesPlanLimits < Gitlab::Database::Migration[1.0]
  def up
    create_or_update_plan_limit('dast_profile_schedules', 'default', 0)
    create_or_update_plan_limit('dast_profile_schedules', 'free', 1)
    create_or_update_plan_limit('dast_profile_schedules', 'bronze', 1)
    create_or_update_plan_limit('dast_profile_schedules', 'silver', 1)
    create_or_update_plan_limit('dast_profile_schedules', 'premium', 1)
    create_or_update_plan_limit('dast_profile_schedules', 'premium_trial', 1)
    create_or_update_plan_limit('dast_profile_schedules', 'gold', 20)
    create_or_update_plan_limit('dast_profile_schedules', 'ultimate', 20)
    create_or_update_plan_limit('dast_profile_schedules', 'ultimate_trial', 20)
  end

  def down
    create_or_update_plan_limit('dast_profile_schedules', 'default', 0)
    create_or_update_plan_limit('dast_profile_schedules', 'free', 0)
    create_or_update_plan_limit('dast_profile_schedules', 'bronze', 0)
    create_or_update_plan_limit('dast_profile_schedules', 'silver', 0)
    create_or_update_plan_limit('dast_profile_schedules', 'premium', 0)
    create_or_update_plan_limit('dast_profile_schedules', 'premium_trial', 0)
    create_or_update_plan_limit('dast_profile_schedules', 'gold', 0)
    create_or_update_plan_limit('dast_profile_schedules', 'ultimate', 0)
    create_or_update_plan_limit('dast_profile_schedules', 'ultimate_trial', 0)
  end
end