support constraints as arrays

vti · Feb 21, 2014 · 0da9c83 · 0da9c83
1 parent 1887fc2
commit 0da9c83
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/lib/Routes/Tiny/Pattern.pm b/lib/Routes/Tiny/Pattern.pm
@@ -229,6 +229,9 @@ sub _prepare_pattern {
 
             if (exists $self->{constraints}->{$name}) {
                 $constraint = $self->{constraints}->{$name};
+                if (ref $constraint eq 'ARRAY') {
+                    $constraint = '?:' . join('|', @$constraint);
+                }
                 $re .= "($constraint)";
             }
             else {

diff --git a/t/match-with-constraints.t b/t/match-with-constraints.t
@@ -1,7 +1,7 @@
 use strict;
 use warnings;
 
-use Test::More tests => 5;
+use Test::More;
 
 use Routes::Tiny;
 
@@ -25,3 +25,18 @@ ok($@ =~ qr/Required param 'id' was not passed when building a path/);
 
 eval { $r->build_path('article', id => 'abc'); };
 ok($@ =~ qr/Param 'id' fails a constraint/);
+
+subtest 'contraint as array' => sub {
+    my $r = Routes::Tiny->new;
+
+    $r->add_route(
+        '/articles/:id',
+        name        => 'article',
+        constraints => {id => [qw/1 2 3/]}
+    );
+
+    ok $r->match('/articles/1');
+    ok!$r->match('/articles/a');
+};
+
+done_testing;