use VersionStorage in the rest of our stuff

[dbsrgits/DBIx-Class-DeploymentHandler.git] / lib / DBIx / Class / DeploymentHandler / HandlesVersioning.pm

diff --git a/lib/DBIx/Class/DeploymentHandler/HandlesVersioning.pm b/lib/DBIx/Class/DeploymentHandler/HandlesVersioning.pm
index b1ebb78..f203066 100644 (file)
--- a/lib/DBIx/Class/DeploymentHandler/HandlesVersioning.pm
+++ b/lib/DBIx/Class/DeploymentHandler/HandlesVersioning.pm
@@ -10,18 +10,13 @@ has schema => (
   handles => [qw( schema_version )],
 );
 
-has version_rs => (
-  isa        => 'DBIx::Class::ResultSet',
+has version_storage => (
+  does       => 'DBIx::Class::DeploymentHandler::HandlesVersionStorage',
   is         => 'ro',
-  lazy_build => 1,
-  handles    => [qw( is_installed db_version )],
+  required   => 1,
+  handles    => 'DBIx::Class::DeploymentHandler::HandlesVersionStorage',
 );
 
-sub _build_version_rs {
-   $_[0]->schema->set_us_up_the_bomb;
-   $_[0]->schema->resultset('__VERSION')
-}
-
 has to_version => (
   is         => 'ro',
   lazy_build => 1,