123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596 |
- /*
- * SonarQube
- * Copyright (C) 2009-2023 SonarSource SA
- * mailto:info AT sonarsource DOT com
- *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License as published by the Free Software Foundation; either
- * version 3 of the License, or (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public License
- * along with this program; if not, write to the Free Software Foundation,
- * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
- */
- package org.sonar.server.platform.db.migration.version.v100;
-
- import java.sql.SQLException;
- import java.util.HashMap;
- import java.util.Map;
- import java.util.stream.Collectors;
- import org.junit.Rule;
- import org.junit.Test;
- import org.sonar.core.util.UuidFactory;
- import org.sonar.core.util.UuidFactoryFast;
- import org.sonar.db.CoreDbTester;
- import org.sonar.server.platform.db.migration.step.DataChange;
-
- import static org.apache.commons.lang.RandomStringUtils.randomAlphabetic;
- import static org.apache.commons.lang.RandomStringUtils.randomNumeric;
- import static org.assertj.core.api.Assertions.assertThat;
-
- public class UpdateUserLocalValueInUsersTest {
-
- private final UuidFactory uuidFactory = UuidFactoryFast.getInstance();
-
- @Rule
- public CoreDbTester db = CoreDbTester.createForSchema(UpdateUserLocalValueInUsersTest.class, "schema.sql");
-
- private final DataChange underTest = new UpdateUserLocalValueInUsers(db.database());
-
- @Test
- public void migration_updates_user_local_if_null() throws SQLException {
- String userUuid1 = insertUser(true);
- String userUuid2 = insertUser(false);
- String userUuid3 = insertUser(null);
-
- underTest.execute();
-
- assertUserLocalIsUpdatedCorrectly(userUuid1, true);
- assertUserLocalIsUpdatedCorrectly(userUuid2, false);
- assertUserLocalIsUpdatedCorrectly(userUuid3, true);
- }
-
- @Test
- public void migration_should_be_reentrant() throws SQLException {
- String userUuid1 = insertUser(true);
- String userUuid2 = insertUser(false);
- String userUuid3 = insertUser(null);
-
- underTest.execute();
- // re-entrant
- underTest.execute();
-
- assertUserLocalIsUpdatedCorrectly(userUuid1, true);
- assertUserLocalIsUpdatedCorrectly(userUuid2, false);
- assertUserLocalIsUpdatedCorrectly(userUuid3, true);
- }
-
- private void assertUserLocalIsUpdatedCorrectly(String userUuid, boolean expected) {
- String selectSql = String.format("select USER_LOCAL from users where uuid='%s'", userUuid);
- assertThat(db.select(selectSql).stream()
- .map(row -> row.get("USER_LOCAL"))
- .toList())
- .containsExactlyInAnyOrder(expected);
- }
-
- private String insertUser(Boolean userLocal) {
- Map<String, Object> map = new HashMap<>();
- String uuid = uuidFactory.create();
- map.put("UUID", uuid);
- map.put("LOGIN", randomAlphabetic(20));
- map.put("EXTERNAL_LOGIN", randomAlphabetic(20));
- map.put("EXTERNAL_IDENTITY_PROVIDER", "sonarqube");
- map.put("EXTERNAL_ID", randomNumeric(5));
- map.put("CREATED_AT", System.currentTimeMillis());
- map.put("USER_LOCAL", userLocal);
- map.put("RESET_PASSWORD", false);
- db.executeInsert("users", map);
- return uuid;
- }
- }
|