package ...

import org.testng.annotations.BeforeClass;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

public class Example {
    private User user;


    @BeforeClass(description = "Подготовка данных для тестов")
    public void prepareTestData() {
        dbSteps().cleanUsers();
        user = prepareUser("main-region-user-name", "test", "region-main");
    }

    @BeforeMethod(description = "Подготовка между тестами")
    public void prepareTests() {
        dbSteps().cleanUserJoins(user);
    }

    @Test(description = "Объединение пользователя в главный регион")
    public void joinUserToMainRegion() {
        var userForJoin = Allure.precondition(() -> prepareUser("region-1-user-name", "test", "region-1"));
        apiSteps().joinUsers(user, userForJoin).check(
                response -> response.shouldOk()
                        .userIdShould(equalTo(user.id()))
                        .joinedUserIdsShould(hasSize(1))
                        .joinedUserIdsShould(everyItem(equalTo(userForJoin.id())))
        );
    }

    @Test(description = "Объединение нескольких пользователей из разных регионов в главный регион")
    public void joinUsersFromDifferentRegionsToMainRegion() {
        var firstUserForJoin = Allure.precondition(() -> prepareUser("region-1-user-name", "test", "region-1"));
        var secondUserForJoin = Allure.precondition(() -> prepareUser("region-2-user-name", "test", "region-2"));
        apiSteps().joinUsers(user, firstUserForJoin, secondUserForJoin).check(
                response -> response.shouldOk()
                        .userIdShould(equalTo(user.id()))
                        .joinedUserIdsShould(hasSize(2))
                        .joinedUserIdsShould(equalTo(List.of(firstUserForJoin.id(), secondUserForJoin.id())))
        );
    }


    @Precondition
    public User userInBlockedRegion(String name, String lastName, String regionId) {
        dbStesp().setRegionBlocked(regionId);
        return prepareUser(name, lastName, regionId);
    }


    @Test(description = "Объединение пользователя в главный регион из заблокированного")
    public void joinUserToMainRegionFromBlocked() {
        var regionId = "region-35";
        var userForJoin = userInBlockedRegion("region-blocked-user-name", "test", regionId);
        apiSteps().joinUsers(user, userForJoin).check(
                response -> response.shouldError(new ApiError(500, "Region '%s' blocked".formatted(regionId)))
        );
    }
}