Tests for user agent injection

miquelruiz · Jan 16, 2013 · 3b730b6 · 3b730b6
1 parent 892c39f
commit 3b730b6
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/t/01-opsview-rest.t b/t/01-opsview-rest.t
@@ -6,6 +6,8 @@ use FindBin;
 use lib "$FindBin::Bin/lib";
 use Opsview::REST::TestUtils;
 
+use HTTP::Tiny;
+
 use Test::More tests => 14;
 use Test::Exception;
 
@@ -34,7 +36,11 @@ SKIP: {
     is($@->reason, 'Unauthorized', '"Unauthorized" reason in exception');
     ok(defined $@->message, 'Message defined in exception');
 
-    my $ops = get_opsview();
+    my $ops = get_opsview(undef, undef, undef, ua => HTTP::Tiny->new);
+    like($ops->ua->agent, qr/HTTP-Tiny/, 'Force user agent');
+
+    $ops = get_opsview();
+    like($ops->ua->agent, qr/Opsview-REST/, 'Default user agent');
 
     isa_ok($ops, 'Opsview::REST', "Object created");
     ok(defined $ops->headers->{'X-Opsview-Token'}, "Logged in");

diff --git a/t/lib/Opsview/REST/TestUtils.pm b/t/lib/Opsview/REST/TestUtils.pm
@@ -16,11 +16,14 @@ use Test::Exception;
 use Test::Deep;
 
 sub get_opsview {
-    my ($url, $user, $pass) = (qw( http://localhost/rest admin initial ));
+    my ($url, $user, $pass, %opts) = 
+        (qw( http://localhost/rest admin initial ));
+
     return Opsview::REST->new(
         base_url => $ENV{OPSVIEW_REST_URL}  || $url,
         user     => $ENV{OPSVIEW_REST_USER} || $user,
         pass     => $ENV{OPSVIEW_REST_PASS} || $pass,
+        %opts
     );
 }