Compatibility fixes with sqlalchemy-migrate, this is for the better a…

…nyway Signed-off-by: Alfred Hall <ahall@ahall.org>
ahall · Apr 19, 2011 · 85b54b1 · 85b54b1
1 parent f212c6f
commit 85b54b1
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 7 deletions.
diff --git a/src/Sector.Tests/MigrateApiTest.cs b/src/Sector.Tests/MigrateApiTest.cs
@@ -110,20 +110,18 @@ public void GetDbVersion()
             // Now create migrate version with version 0, then it shall return true.
             using (ISession session = dbFactory.OpenSession())
             {
-                int migId = 0;
                 using (ITransaction transaction = session.BeginTransaction())
                 {
                     var migVer = new MigrateVersion(repository.RepositoryId, repository.RepositoryPath, 0);
                     session.Save(migVer);
                     transaction.Commit();
-                    migId = migVer.Id;
                 }
 
                 Assert.AreEqual(0, migrateApi.GetDbVersion(repository));
 
                 using (ITransaction transaction = session.BeginTransaction())
                 {
-                    var migVer = session.Get<MigrateVersion>(migId);
+                    var migVer = session.Get<MigrateVersion>(repository.RepositoryId);
                     migVer.Version = 10;
                     session.Update(migVer);
                     transaction.Commit();

diff --git a/src/Sector/Entities/MigrateVersion.cs b/src/Sector/Entities/MigrateVersion.cs
@@ -4,8 +4,6 @@ namespace Sector.Entities
 {
     public class MigrateVersion
     {
-        public virtual int Id { get; private set; }
-
         /// <summary>
         /// Gets or sets the repository identifier.
         /// </summary>

diff --git a/src/Sector/Mappings/MigrateVersionMap.cs b/src/Sector/Mappings/MigrateVersionMap.cs
@@ -8,9 +8,8 @@ public class MigrateVersionMap : ClassMap<MigrateVersion>
     {
         public MigrateVersionMap()
         {
-            Id(x => x.Id, "id");
+            Id(x => x.RepositoryId, "repository_id");
             Map(x => x.RepositoryPath).Column("repository_path");
-            Map(x => x.RepositoryId).Column("repository_id");
             Map(x => x.Version).Column("version");
 
             Table("migrate_version");