From: Mugen Kenichi <mugen.kenichi@uninets.eu>
Date: Wed, 28 Sep 2011 13:17:31 +0000 (+0200)
Subject: better mongo test
X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=commitdiff_plain;h=d96c63f8fdb213f75b50ff0e9e0cdc7c103716a4;p=p5sagit%2FEmail-Archive.git

better mongo test
---

diff --git a/t/basic.t b/t/basic.t
index 36ee684..bc51d39 100644
--- a/t/basic.t
+++ b/t/basic.t
@@ -28,5 +28,6 @@ cmp_ok($found->header('subject'), 'eq', "Message in a bottle",
 
 done_testing;
 
+# cleanup
 unlink 't/test.db';
 
diff --git a/t/mongodb.t b/t/mongodb.t
index 6a81957..bd7c48b 100644
--- a/t/mongodb.t
+++ b/t/mongodb.t
@@ -4,47 +4,79 @@ use warnings;
 use Email::MIME;
 
 use Test::More;
+use Try::Tiny;
 
 use Email::Archive;
 use Email::Archive::Storage::MongoDB;
 
-my $email = Email::MIME->create(
-  header => [
-    From    => 'foo@example.com',
-    To      => 'drain@example.com',
-    Subject => 'Message in a bottle',
-    'Message-ID' => 'helloworld',
-  ],
-  body => 'hello there!'
-);
+my $test_host        = 'localhost';
+my $test_port        = 27017;
+my $mongo_database   = 'EmailArchiveTestMongoDB';
+my $mongo_collection = 'emails';
 
-my $storage = Email::Archive::Storage::MongoDB->new(
-  host       => 'localhost',
-  port       => 27017,
-  database   => 'EmailArchiveTestMongoDB',
-  collection => 'emails',
-);
+sub check_for_mongo {
+    my ($host, $port) = @_;
+    try {
+        require Net::Telnet;
+        my $telnet = Net::Telnet->new(
+            Timeout => 10,
+        );
 
-my $e = Email::Archive->new(
-  storage => $storage,
-);
+        my $ok = $telnet->open(
+            Host    => $host,
+            Port    => $port,
+        );
 
-$e->connect;
-$e->store($email);
+        return $ok;
+    }
+    catch {
+        return 0;
+    }
+}
 
-my $found = $e->retrieve('helloworld');
+SKIP: {
+    skip 'no MongoDB running or Net::Telnet not installed', 1
+        unless check_for_mongo($test_host, $test_port);
 
-$found = $e->retrieve('helloworld');
-cmp_ok($found->header('subject'), 'eq', "Message in a bottle",
-  "can find stored message by ID");
+    my $email = Email::MIME->create(
+      header => [
+        From    => 'foo@example.com',
+        To      => 'drain@example.com',
+        Subject => 'Message in a bottle',
+        'Message-ID' => 'helloworld',
+      ],
+      body => 'hello there!'
+    );
 
+    my $storage = Email::Archive::Storage::MongoDB->new(
+      host       => $test_host,
+      port       => $test_port,
+      database   => $mongo_database,
+      collection => $mongo_collection,
+    );
 
-done_testing;
+    my $e = Email::Archive->new(
+      storage => $storage,
+    );
+
+    $e->connect;
+    $e->store($email);
+
+    my $found = $e->retrieve('helloworld');
+
+    $found = $e->retrieve('helloworld');
+    cmp_ok($found->header('subject'), 'eq', "Message in a bottle",
+      "can find stored message by ID");
 
-my $conn = MongoDB::Connection->new(
-    host => 'localhost',
-    port => 27017
-);
+    # cleanup
+    my $conn = MongoDB::Connection->new(
+        host => $test_host,
+        port => $test_port,
+    );
 
-$conn->EmailArchiveTestMongoDB->drop;
+    $conn->$mongo_database->drop;
+
+}
+
+done_testing;