add path to "database" file

szabgab · Feb 21, 2013 · f1007c7 · f1007c7
1 parent 12de6db
commit f1007c7
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 2 deletions.
diff --git a/lib/WWW/AdServer/Database.pm b/lib/WWW/AdServer/Database.pm
@@ -3,4 +3,25 @@ use warnings;
 package WWW::AdServer::Database;
 use Moo;
 
+has dsn => (
+    is       => 'ro',
+    required => 1,
+);
+
+has type => (
+    is       => 'rw',
+);
+
+
+sub BUILD {
+    my ($self) = @_;
+    if (not $self->type) {
+        if ($self->dsn =~ /\.yml$/) {
+            $self->type('YAML');
+            require WWW::AdServer::Database::YAML;
+        }
+    }
+    return;
+}
+
 1;
diff --git a/t/01-adserver.t b/t/01-adserver.t
@@ -3,13 +3,15 @@ use warnings;
 
 use Test::More;
 
-plan tests => 2;
+plan tests => 4;
 
 use WWW::AdServer;
 
 my $ads = WWW::AdServer->new;
 isa_ok($ads, 'WWW::AdServer');
 
-my $db = WWW::AdServer::Database->new;
+my $db = WWW::AdServer::Database->new(dsn => 't/files/ads.yml');
 isa_ok($db, 'WWW::AdServer::Database');
+is $db->dsn, 't/files/ads.yml', 'dsn ok';
+is $db->type, 'YAML', 'type ok';