1 files changed, 19 insertions, 2 deletions
diff --git a/spec/models/concerns/ci/partitionable_spec.rb b/spec/models/concerns/ci/partitionable_spec.rb
index d53501ccc3d..f3d33c971c7 100644
--- a/spec/models/concerns/ci/partitionable_spec.rb
+++ b/spec/models/concerns/ci/partitionable_spec.rb
@@ -3,9 +3,9 @@
 require 'spec_helper'
 
 RSpec.describe Ci::Partitionable do
-  describe 'partitionable models inclusion' do
-    let(:ci_model) { Class.new(Ci::ApplicationRecord) }
+  let(:ci_model) { Class.new(Ci::ApplicationRecord) }
 
+  describe 'partitionable models inclusion' do
     subject { ci_model.include(described_class) }
 
     it 'raises an exception' do
@@ -23,4 +23,21 @@ RSpec.describe Ci::Partitionable do
       end
     end
   end
+
+  context 'with through options' do
+    before do
+      allow(ActiveSupport::DescendantsTracker).to receive(:store_inherited)
+      stub_const("#{described_class}::Testing::PARTITIONABLE_MODELS", [ci_model.name])
+
+      ci_model.include(described_class)
+      ci_model.partitionable scope: ->(r) { 1 },
+                             through: { table: :_test_table_name, flag: :some_flag }
+    end
+
+    it { expect(ci_model.routing_table_name).to eq(:_test_table_name) }
+
+    it { expect(ci_model.routing_table_name_flag).to eq(:some_flag) }
+
+    it { expect(ci_model.ancestors).to include(described_class::Switch) }
+  end
 end