Rubocoping

Dandush03 · Jan 27, 2025 · 90bbde2 · 90bbde2
1 parent 9d6aca4
commit 90bbde2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/lib/pg_rls/admin.rb b/lib/pg_rls/admin.rb
@@ -1,7 +1,8 @@
 # frozen_string_literal: true
 
 module PgRls
-  # Admin Module
+  # The admin module provides the admin_execute method to execute
+  # queries on the admin shard.
   module Admin
     def self.admin_execute(sql = nil)
       PgRls::Record.connected_to(shard: :admin) do

diff --git a/test/pg_rls/admin_test.rb b/test/pg_rls/admin_test.rb
@@ -9,7 +9,8 @@ class PgRlsAdminTest < ActiveSupport::TestCase
 
   test "admin_execute configures the connection to use the admin shard" do
     PgRls.stub :connects_to, { shards: { admin: { writing: :admin, reading: :admin } } } do
-      assert_equal "primary", PgRls::Admin.admin_execute { ::ActiveRecord::Base.connection_db_config.name }
+      result = PgRls::Admin.admin_execute { ::ActiveRecord::Base.connection_db_config.name }
+      assert_equal "primary", result
     end
   end
 end