Mark some module methods as private

lib/gitlab/database/partitioning.rb

@@ -18,29 +18,16 @@ module Gitlab
         end
       end
 
-      def self.register_models(models)
+      class << self
+        def register_models(models)
           registered_models.merge(models)
         end
 
-      def self.register_tables(tables)
+        def register_tables(tables)
           registered_tables.merge(tables)
         end
 
-      def self.registered_models
-        @registered_models ||= Set.new
-      end
-
-      def self.registered_tables
-        @registered_tables ||= Set.new
-      end
-
-      def self.registered_for_sync
-        registered_models + registered_tables.map do |table|
-          TableWithoutModel.new(**table)
-        end
-      end
-
-      def self.sync_partitions(models_to_sync = registered_for_sync)
+        def sync_partitions(models_to_sync = registered_for_sync)
           Gitlab::AppLogger.info(message: 'Syncing dynamic postgres partitions')
 
           Gitlab::Database::EachDatabase.each_model_connection(models_to_sync) do |model|
@@ -50,7 +37,7 @@ module Gitlab
           Gitlab::AppLogger.info(message: 'Finished sync of dynamic postgres partitions')
         end
 
-      def self.report_metrics(models_to_monitor = registered_models)
+        def report_metrics(models_to_monitor = registered_models)
           partition_monitoring = PartitionMonitoring.new
 
           Gitlab::Database::EachDatabase.each_model_connection(models_to_monitor) do |model|
@@ -58,7 +45,7 @@ module Gitlab
           end
         end
 
-      def self.drop_detached_partitions
+        def drop_detached_partitions
           Gitlab::AppLogger.info(message: 'Dropping detached postgres partitions')
 
           Gitlab::Database::EachDatabase.each_database_connection do
@@ -67,6 +54,23 @@ module Gitlab
 
           Gitlab::AppLogger.info(message: 'Finished dropping detached postgres partitions')
         end
+
+        private
+
+        def registered_models
+          @registered_models ||= Set.new
+        end
+
+        def registered_tables
+          @registered_tables ||= Set.new
+        end
+
+        def registered_for_sync
+          registered_models + registered_tables.map do |table|
+            TableWithoutModel.new(**table)
+          end
+        end
+      end
     end
   end
 end

spec/lib/gitlab/database/partitioning_spec.rb

@@ -45,7 +45,7 @@ RSpec.describe Gitlab::Database::Partitioning do
       let(:model) { double('model') }
 
       it 'manages partitions for each registered model' do
-        registered_for_sync = described_class.registered_for_sync
+        registered_for_sync = described_class.__send__(:registered_for_sync)
 
         allow(described_class).to receive(:registered_for_sync)
           .and_return(registered_for_sync)
@@ -76,7 +76,7 @@ RSpec.describe Gitlab::Database::Partitioning do
         end
 
         expect(Gitlab::Database::EachDatabase).to receive(:each_model_connection)
-          .with(described_class.registered_models)
+          .with(described_class.__send__(:registered_models))
           .and_yield(model1)
           .and_yield(model2)
 
@@ -128,7 +128,7 @@ RSpec.describe Gitlab::Database::Partitioning do
 
   context 'ensure that the registered models have partitioning strategy' do
     it 'fails when partitioning_strategy is not specified for the model' do
-      expect(described_class.registered_models).to all(respond_to(:partitioning_strategy))
+      expect(described_class.__send__(:registered_models)).to all(respond_to(:partitioning_strategy))
     end
   end
 end