X-Git-Url: https://code.wpia.club/?a=blobdiff_plain;f=tests%2Forg%2Fcacert%2Fgigi%2Futil%2FTestPasswordMigration.java;fp=tests%2Forg%2Fcacert%2Fgigi%2Futil%2FTestPasswordMigration.java;h=7ff4ea6dd76176bdc9e577c8610e4eb4c655e557;hb=a0232b6e40e7e09767f0444d24e18bf12dafc362;hp=2992e05c9bf0eb9b70e4cb60194ce5ccf89c0a1f;hpb=851b2db2211e0f7770065dc4558cc0de74a39df4;p=gigi.git

diff --git a/tests/org/cacert/gigi/util/TestPasswordMigration.java b/tests/org/cacert/gigi/util/TestPasswordMigration.java
index 2992e05c..7ff4ea6d 100644
--- a/tests/org/cacert/gigi/util/TestPasswordMigration.java
+++ b/tests/org/cacert/gigi/util/TestPasswordMigration.java
@@ -5,7 +5,6 @@ import static org.junit.Assert.*;
 
 import java.io.IOException;
 
-import org.cacert.gigi.database.DatabaseConnection;
 import org.cacert.gigi.database.GigiPreparedStatement;
 import org.cacert.gigi.database.GigiResultSet;
 import org.cacert.gigi.testUtils.ManagedTest;
@@ -20,18 +19,20 @@ public class TestPasswordMigration extends ManagedTest {
 
     @Test
     public void testPasswordMigration() throws IOException {
-        GigiPreparedStatement stmt = DatabaseConnection.getInstance().prepare("UPDATE users SET `password`=? WHERE id=?");
-        stmt.setString(1, PasswordHash.sha1("a"));
-        stmt.setInt(2, ru.getUser().getId());
-        stmt.execute();
+        try (GigiPreparedStatement stmt = new GigiPreparedStatement("UPDATE users SET `password`=? WHERE id=?")) {
+            stmt.setString(1, PasswordHash.sha1("a"));
+            stmt.setInt(2, ru.getUser().getId());
+            stmt.execute();
+        }
         String cookie = login(ru.getUser().getEmail(), "a");
         assertTrue(isLoggedin(cookie));
 
-        stmt = DatabaseConnection.getInstance().prepare("SELECT `password` FROM users WHERE id=?");
-        stmt.setInt(1, ru.getUser().getId());
-        GigiResultSet res = stmt.executeQuery();
-        assertTrue(res.next());
-        String newHash = res.getString(1);
-        assertThat(newHash, containsString("$"));
+        try (GigiPreparedStatement stmt = new GigiPreparedStatement("SELECT `password` FROM users WHERE id=?")) {
+            stmt.setInt(1, ru.getUser().getId());
+            GigiResultSet res = stmt.executeQuery();
+            assertTrue(res.next());
+            String newHash = res.getString(1);
+            assertThat(newHash, containsString("$"));
+        }
     }
 }