tests/club/wpia/gigi/util/TestPasswordMigration.java

   1 package club.wpia.gigi.util;
   2
   3 import static org.hamcrest.CoreMatchers.*;
   4 import static org.junit.Assert.*;
   5
   6 import java.io.IOException;
   7
   8 import org.junit.Rule;
   9 import org.junit.Test;
  10
  11 import club.wpia.gigi.database.GigiPreparedStatement;
  12 import club.wpia.gigi.database.GigiResultSet;
  13 import club.wpia.gigi.testUtils.ManagedTest;
  14 import club.wpia.gigi.testUtils.RegisteredUser;
  15 import club.wpia.gigi.util.PasswordHash;
  16
  17 public class TestPasswordMigration extends ManagedTest {
  18
  19     @Rule
  20     public RegisteredUser ru = new RegisteredUser();
  21
  22     @Test
  23     public void testPasswordMigration() throws IOException {
  24         try (GigiPreparedStatement stmt = new GigiPreparedStatement("UPDATE users SET `password`=? WHERE id=?")) {
  25             stmt.setString(1, PasswordHash.sha1("a"));
  26             stmt.setInt(2, ru.getUser().getId());
  27             stmt.execute();
  28         }
  29         String cookie = login(ru.getUser().getEmail(), "a");
  30         assertTrue(isLoggedin(cookie));
  31
  32         try (GigiPreparedStatement stmt = new GigiPreparedStatement("SELECT `password` FROM users WHERE id=?")) {
  33             stmt.setInt(1, ru.getUser().getId());
  34             GigiResultSet res = stmt.executeQuery();
  35             assertTrue(res.next());
  36             String newHash = res.getString(1);
  37             assertThat(newHash, containsString("$"));
  38         }
  39     }
  40 }