Allow for testing snippets on .pod files

Not just `.pm` files.

Author: zmughal

xt/author/pod-snippets.t

@@ -4,6 +4,7 @@ use strict;
 use warnings;
 use Test::More;
 use Test::Pod::Snippets;
+use With::Roles;
 
 my @modules = qw/
 	AI::TensorFlow::Libtensorflow
@@ -16,6 +17,30 @@ my @modules = qw/
 
 plan tests => 0 + @modules;
 
+package # hide from PAUSE
+	Test::Pod::Snippets::Role::PodLocatable {
+
+	use Moose::Role;
+	use Pod::Simple::Search;
+
+	around _parse => sub {
+		my $orig = shift;
+		my ($self, $type, $input) = @_;
+
+		my $output = eval { $orig->(@_); };
+		my $error = $@;
+		if( $error =~ /not found in \@INC/ && $type eq 'module' ) {
+			my $pod_file = Pod::Simple::Search->new->find($input);
+			if( -f $pod_file ) {
+				return $orig->($self, 'file', $pod_file )
+			} else {
+				die "$error\nUnable to find POD file for $input\n";
+			}
+		}
+
+		return $output;
+	};
+}
 
 package # hide from PAUSE
 	My::Test::Pod::Snippets::Parser {
@@ -38,7 +63,7 @@ package # hide from PAUSE
 
 for (@modules) {
 	my $parser = My::Test::Pod::Snippets::Parser->new;
-	my $tps = Test::Pod::Snippets->new(
+	my $tps = Test::Pod::Snippets->with::roles('+PodLocatable')->new(
 		parser => $parser,
 	);
 	$parser->{tps} = $tps;