diff --git a/app/models/account.rb b/app/models/account.rb index 6049a4d80..3b2004cfa 100644 --- a/app/models/account.rb +++ b/app/models/account.rb @@ -97,6 +97,9 @@ class Account < ApplicationRecord has_many :api_channels, dependent: :destroy_async, class_name: '::Channel::Api' has_many :articles, dependent: :destroy_async, class_name: '::Article' has_many :captain_reservations, class_name: 'Captain::Reservation', dependent: :destroy + has_many :captain_lifecycle_rules, class_name: 'Captain::Lifecycle::Rule', dependent: :destroy + has_many :captain_lifecycle_deliveries, class_name: 'Captain::Lifecycle::Delivery', dependent: :destroy + has_one :captain_lifecycle_config, class_name: 'Captain::Lifecycle::Config', dependent: :destroy has_many :captain_units, class_name: 'Captain::Unit', dependent: :destroy has_many :captain_gallery_items, class_name: 'Captain::GalleryItem', dependent: :destroy has_many :assignment_policies, dependent: :destroy_async diff --git a/spec/models/account_spec.rb b/spec/models/account_spec.rb index 5eced364d..d86d0a364 100644 --- a/spec/models/account_spec.rb +++ b/spec/models/account_spec.rb @@ -273,4 +273,10 @@ RSpec.describe Account do end end end + + describe 'captain lifecycle associations' do + it { is_expected.to have_many(:captain_lifecycle_rules).class_name('Captain::Lifecycle::Rule') } + it { is_expected.to have_many(:captain_lifecycle_deliveries).class_name('Captain::Lifecycle::Delivery') } + it { is_expected.to have_one(:captain_lifecycle_config).class_name('Captain::Lifecycle::Config') } + end end