Added version check to ensure correct versions
diff --git a/stack/core/src/test/java/org/apache/usergrid/corepersistence/migration/EntityTypeMappingMigrationIT.java b/stack/core/src/test/java/org/apache/usergrid/corepersistence/migration/EntityTypeMappingMigrationIT.java
index d2c31ff..b218368 100644
--- a/stack/core/src/test/java/org/apache/usergrid/corepersistence/migration/EntityTypeMappingMigrationIT.java
+++ b/stack/core/src/test/java/org/apache/usergrid/corepersistence/migration/EntityTypeMappingMigrationIT.java
@@ -77,6 +77,7 @@
@Test
public void testIdMapping() throws Throwable {
+ assertEquals("version 1 expected", 1, entityTypeMappingMigration.getVersion());
final EntityManager newAppEm = app.getEntityManager();
@@ -112,6 +113,7 @@
+
AllEntitiesInSystemObservable.getAllEntitiesInSystem( managerCache )
.doOnNext( new Action1<AllEntitiesInSystemObservable.EntityData>() {
@Override
diff --git a/stack/core/src/test/java/org/apache/usergrid/corepersistence/migration/GraphShardVersionMigrationIT.java b/stack/core/src/test/java/org/apache/usergrid/corepersistence/migration/GraphShardVersionMigrationIT.java
index e16c60d..1e6f0d5 100644
--- a/stack/core/src/test/java/org/apache/usergrid/corepersistence/migration/GraphShardVersionMigrationIT.java
+++ b/stack/core/src/test/java/org/apache/usergrid/corepersistence/migration/GraphShardVersionMigrationIT.java
@@ -82,6 +82,7 @@
@Test
public void testIdMapping() throws Throwable {
+ assertEquals("version 2 expected", 2, graphShardVersionMigration.getVersion());
/**
* Drop our migration keyspaces to ensure we don't have a "new version in there"