1 files changed, 98 insertions, 0 deletions
diff --git a/actionpack/test/journey/nfa/simulator_test.rb b/actionpack/test/journey/nfa/simulator_test.rb
new file mode 100644
index 0000000000..673a491fe5
--- /dev/null
+++ b/actionpack/test/journey/nfa/simulator_test.rb
@@ -0,0 +1,98 @@
+require 'abstract_unit'
+
+module ActionDispatch
+  module Journey
+    module NFA
+      class TestSimulator < ActiveSupport::TestCase
+        def test_simulate_simple
+          sim = simulator_for ['/foo']
+          assert_match sim, '/foo'
+        end
+
+        def test_simulate_simple_no_match
+          sim = simulator_for ['/foo']
+          assert_no_match sim, 'foo'
+        end
+
+        def test_simulate_simple_no_match_too_long
+          sim = simulator_for ['/foo']
+          assert_no_match sim, '/foo/bar'
+        end
+
+        def test_simulate_simple_no_match_wrong_string
+          sim = simulator_for ['/foo']
+          assert_no_match sim, '/bar'
+        end
+
+        def test_simulate_regex
+          sim = simulator_for ['/:foo/bar']
+          assert_match sim, '/bar/bar'
+          assert_match sim, '/foo/bar'
+        end
+
+        def test_simulate_or
+          sim = simulator_for ['/foo', '/bar']
+          assert_match sim, '/bar'
+          assert_match sim, '/foo'
+          assert_no_match sim, '/baz'
+        end
+
+        def test_simulate_optional
+          sim = simulator_for ['/foo(/bar)']
+          assert_match sim, '/foo'
+          assert_match sim, '/foo/bar'
+          assert_no_match sim, '/foo/'
+        end
+
+        def test_matchdata_has_memos
+          paths   = %w{ /foo /bar }
+          parser  = Journey::Parser.new
+          asts    = paths.map { |x|
+            ast = parser.parse x
+            ast.each { |n| n.memo = ast}
+            ast
+          }
+
+          expected = asts.first
+
+          builder = Builder.new Nodes::Or.new asts
+
+          sim = Simulator.new builder.transition_table
+
+          md = sim.match '/foo'
+          assert_equal [expected], md.memos
+        end
+
+        def test_matchdata_memos_on_merge
+          parser = Journey::Parser.new
+          routes = [
+            '/articles(.:format)',
+            '/articles/new(.:format)',
+            '/articles/:id/edit(.:format)',
+            '/articles/:id(.:format)',
+          ].map { |path|
+            ast = parser.parse path
+            ast.each { |n| n.memo = ast }
+            ast
+          }
+
+          asts = routes.dup
+
+          ast = Nodes::Or.new routes
+
+          nfa   = Journey::NFA::Builder.new ast
+          sim = Simulator.new nfa.transition_table
+          md = sim.match '/articles'
+          assert_equal [asts.first], md.memos
+        end
+
+        def simulator_for paths
+          parser  = Journey::Parser.new
+          asts    = paths.map { |x| parser.parse x }
+          builder = Builder.new Nodes::Or.new asts
+          Simulator.new builder.transition_table
+        end
+      end
+    end
+  end
+end