Do not raise Error with multiple registry tables

Author: Tranquility

lib/refile/postgres/backend.rb

@@ -23,7 +23,7 @@ def registry_table
         unless @registry_table_validated
           with_connection do |connection|
             connection.exec_params("SELECT * FROM pg_catalog.pg_tables WHERE tablename = $1::varchar;", [@registry_table]) do |result|
-              if result.count != 1
+              if result.count == 0
                 raise RegistryTableDoesNotExistError.new(%{Please create a table "#{@registry_table}" where backend could store list of attachments})
               end
             end

spec/refile/postgres/backend_spec.rb

@@ -21,6 +21,7 @@
           }.to raise_error(ArgumentError, "When initializing new Refile::Postgres::Backend first argument should be an instance of PG::Connection or a lambda/proc that yields it.")
         end
       end
+
       context "when lambda does yield a PG::Connection" do
         let(:connection_or_proc) { lambda { |&blk| blk.call(connection) } }
         it "is usable in queries" do
@@ -30,6 +31,39 @@
     end
   end
 
+  describe "#registry_table" do
+    context "when no registry table is present" do
+      it "raises an exception" do
+        drop_registry_table
+        expect {
+          Refile::Postgres::Backend.new(test_connection, max_size: 100).registry_table
+        }.to raise_error Refile::Postgres::Backend::RegistryTableDoesNotExistError
+      end
+    end
+
+    context "when registry tables exist in multiple schemas" do
+      before do
+        test_connection.exec %{
+          CREATE SCHEMA other_schema;
+          CREATE TABLE IF NOT EXISTS other_schema.#{Refile::Postgres::Backend::DEFAULT_REGISTRY_TABLE}
+          ( id serial NOT NULL );
+        }
+      end
+
+      after do
+        test_connection.exec %{
+          DROP SCHEMA other_schema CASCADE;
+        }
+      end
+
+      it "does not raise an exception" do
+        expect {
+          Refile::Postgres::Backend.new(test_connection, max_size: 100).registry_table
+        }.not_to raise_error
+      end
+    end
+  end
+
   context "Refile Provided tests" do
     let(:connection_or_proc) { connection }
     it_behaves_like :backend