Use declared connection variable

fbiville · May 26, 2016 · 98875d0 · 98875d0
1 parent f693ed3
commit 98875d0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/liquigraph-core/src/test/java/org/liquigraph/core/io/ChangelogGraphReaderTestSuite.java b/liquigraph-core/src/test/java/org/liquigraph/core/io/ChangelogGraphReaderTestSuite.java
@@ -48,7 +48,7 @@ public void reads_changelog_from_graph_database() throws SQLException {
                 connection
             );
 
-            Collection<Changeset> changesets = reader.read(graphDatabase().connection());
+            Collection<Changeset> changesets = reader.read(connection);
 
             assertThat(changesets).hasSize(1);
             Changeset changeset = changesets.iterator().next();
@@ -80,7 +80,7 @@ public void reads_changeset_with_multiple_queries() throws SQLException {
                 connection
             );
 
-            Collection<Changeset> changesets = reader.read(graphDatabase().connection());
+            Collection<Changeset> changesets = reader.read(connection);
 
             assertThat(changesets).hasSize(1);
             Changeset changeset = changesets.iterator().next();
@@ -127,7 +127,7 @@ public void migrates_pre_1_0_rc3_changelog_before_reading() throws SQLException
                     connection
             );
 
-            Collection<Changeset> changesets = reader.read(graphDatabase().connection());
+            Collection<Changeset> changesets = reader.read(connection);
 
             assertThat(changesets).hasSize(3);
             for (Changeset changeset : changesets) {