Support multiple db/views paths

lib/scenic/definition.rb

@@ -7,13 +7,25 @@ def initialize(name, version)
     end
 
     def to_sql
-      File.read(full_path).tap do |content|
+      File.read(find_definition).tap do |content|
         if content.empty?
           raise "Define view query in #{path} before migrating."
         end
       end
     end
 
+    def find_definition
+      definition = views_paths.flat_map do |directory|
+        Dir.glob("#{directory}/**/#{filename}")
+      end.first
+
+      unless definition
+        raise "Unable to locate #{filename} in #{views_paths}"
+      end
+
+      definition
+    end
+
     def full_path
       Rails.root.join(path)
     end
@@ -28,6 +40,10 @@ def version
 
     private
 
+    def views_paths
+      Rails.application.config.paths["db/views"].expanded
+    end
+
     def filename
       "#{@name}_v#{version}.sql"
     end

lib/scenic/railtie.rb

@@ -7,6 +7,8 @@ module Scenic
   # @see Scenic.load
   class Railtie < Rails::Railtie
     initializer "scenic.load" do
+      Rails.application.config.paths.add("db/views")
+
       ActiveSupport.on_load :active_record do
         Scenic.load
       end

spec/fixtures/db_views/searches_v01.sql

@@ -0,0 +1 @@
+SELECT text 'Hi' as greeting

spec/scenic/definition_spec.rb

@@ -2,22 +2,51 @@
 
 module Scenic
   describe Definition do
+    describe "find_definition" do
+      it "raises an error if file cant be found" do
+        definition = Definition.new("not_valid", 1)
+
+        expect {
+          definition.find_definition
+        }.to raise_error RuntimeError, /Unable to locate not_valid_v01.sql/
+      end
+
+      it "looks inside Rails.root db/views" do
+        definition = Definition.new("not_valid", 1)
+
+        expect {
+          definition.find_definition
+        }.to raise_error RuntimeError, /spec\/dummy\/db\/views/
+      end
+
+      it "looks inside configured additional paths" do
+        with_views_fixtures do
+          definition = Definition.new("empty_view", 1)
+
+          expect(definition.find_definition).to be
+        end
+      end
+    end
+
     describe "to_sql" do
       it "returns the content of a view definition" do
         sql_definition = "SELECT text 'Hi' as greeting"
-        allow(File).to receive(:read).and_return(sql_definition)
 
-        definition = Definition.new("searches", 1)
+        with_views_fixtures do
+          definition = Definition.new("searches", 1)
 
-        expect(definition.to_sql).to eq sql_definition
+          expect(definition.to_sql).to eq sql_definition
+        end
       end
 
       it "raises an error if the file is empty" do
-        allow(File).to receive(:read).and_return("")
+        definition = Definition.new("empty_view", 1)
 
-        expect do
-          Definition.new("searches", 1).to_sql
-        end.to raise_error RuntimeError
+        with_views_fixtures do
+          expect do
+            definition.to_sql
+          end.to raise_error RuntimeError
+        end
       end
     end
 
@@ -52,5 +81,13 @@ module Scenic
         expect(definition.version).to eq "15"
       end
     end
+
+    def with_views_fixtures
+      original = Rails.application.config.paths["db/views"].to_a
+      Rails.application.config.paths["db/views"] << File.expand_path("../../fixtures/db_views", __FILE__)
+      yield
+    ensure
+      Rails.application.config.paths["db/views"] = original
+    end
   end
 end