test stash snapshot for not found page

mojolicious · Feb 11, 2014 · e8408b4 · e8408b4
1 parent c9392e1
commit e8408b4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/lib/Mojolicious/Controller.pm b/lib/Mojolicious/Controller.pm
@@ -420,8 +420,7 @@ sub _development {
     status   => $page eq 'exception' ? 500 : 404
   };
   $options->{exception} = $e if $page eq 'exception';
-  my $inline
-    = $renderer->_bundled($mode eq 'development' ? 'development' : $page);
+  my $inline = $renderer->_bundled($mode eq 'development' ? $mode : $page);
   return $self if _fallbacks($self, $options, $page, $inline);
   _fallbacks($self, {%$options, format => 'html'}, $page, $inline);
   return $self;

diff --git a/t/mojolicious/exception_lite_app.t b/t/mojolicious/exception_lite_app.t
@@ -24,8 +24,9 @@ helper dead_helper => sub { die "dead helper!\n" };
 # Custom rendering for missing "txt" template
 hook before_render => sub {
   my ($self, $args) = @_;
-  $args->{text} = 'Missing template.'
-    if ($args->{template} // '') eq 'not_found' && $args->{format} eq 'txt';
+  return unless ($args->{template} // '') eq 'not_found';
+  my $format = $self->stash('snapshot')->{format};
+  $args->{text} = "Missing $format template." if $args->{format} eq 'txt';
 };
 
 # Custom exception rendering for "txt"
@@ -215,7 +216,7 @@ $t->get_ok('/missing_template.json')->status_is(404)
 # Missing template with custom rendering
 $t->get_ok('/missing_template.txt')->status_is(404)
   ->content_type_is('text/plain;charset=UTF-8')
-  ->content_is('Missing template.');
+  ->content_is('Missing txt template.');
 
 # Missing template (failed rendering)
 $t->get_ok('/missing_template/too')->status_is(404)