Use the expected state in column tests

SpineEventEngine · Sep 3, 2020 · 9c2667a · 9c2667a
1 parent bd252e2
commit 9c2667a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/server/src/test/java/io/spine/server/procman/ProcessManagerRepositoryTest.java b/server/src/test/java/io/spine/server/procman/ProcessManagerRepositoryTest.java
@@ -754,8 +754,12 @@ void updateColumns() {
 
  // Check the column value is propagated to the entity state.
  TestProcessManager entityWithColumns = afterMigration.next();
- assertThat(entityWithColumns.state()
- .getIdString()).isEqualTo(id.toString());
+ Project expectedState = pm
+ .state()
+ .toBuilder()
+ .setIdString(pm.getIdString())
+ .build();
+ assertThat(entityWithColumns.state()).isEqualTo(expectedState);
  }
 
  @Test

diff --git a/server/src/test/java/io/spine/server/projection/ProjectionRepositoryTest.java b/server/src/test/java/io/spine/server/projection/ProjectionRepositoryTest.java
@@ -657,8 +657,12 @@ void updateColumns() {
 
  // Check the column value is propagated to the entity state.
  TestProjection entityWithColumns = afterMigration.next();
- assertThat(entityWithColumns.state()
- .getIdString()).isEqualTo(id.toString());
+ Project expectedState = projection
+ .state()
+ .toBuilder()
+ .setIdString(projection.getIdString())
+ .build();
+ assertThat(entityWithColumns.state()).isEqualTo(expectedState);
  }
 
  @Test