test/lib/railroady/models_diagram_spec.rb in railroady-1.2.0 vs test/lib/railroady/models_diagram_spec.rb in railroady-1.3.0

@@ -7,10 +7,17 @@
       md = ModelsDiagram.new
       files = md.get_files("test/file_fixture/")
       files.size.must_equal 3
     end
 
+    it 'should include concerns if specified' do
+      options = OptionsStruct.new(:include_concerns => true)
+      ad = ModelsDiagram.new(options)
+      files = ad.get_files("test/file_fixture/")
+      files.size.must_equal 4
+    end
+
     it 'should exclude a specific file' do
       options = OptionsStruct.new(:exclude => ['test/file_fixture/app/models/dummy1.rb'])
       md = ModelsDiagram.new(options)
       files = md.get_files("test/file_fixture/")
       files.size.must_equal 2
@@ -43,7 +50,16 @@
       md = ModelsDiagram.new(options)
       files = md.get_files("test/file_fixture/")
       files.size.must_equal 1
     end
 
+    it "should include engine files" do
+      options = OptionsStruct.new(:engine_models => true)
+      md = ModelsDiagram.new(options)
+      engines = [OpenStruct.new(:root => "test/file_fixture/lib")]
+      md.stub(:engines, engines) do
+        md.get_files.must_include("test/file_fixture/lib/app/models/dummy1.rb")
+      end
+    end
+    
   end
 end