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

@@ -7,9 +7,16 @@
       ad = AasmDiagram.new
       files = ad.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 = AasmDiagram.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'])
       ad = AasmDiagram.new(options)
       files = ad.get_files("test/file_fixture/")
       files.size.must_equal 2