Тестирование Trainer Advisor: сетап теста, подготовка хранилищ

TRUNCATE
    users, clients, therapists, exercises, exercise_steps, therapeutic_tasks, journal_entries, files, client_files, programs, program_exercises, appointments, appointment_types
    RESTART IDENTITY;

INSERT INTO users (email, password_hash, roles, created_at, version)
VALUES ('therapist@qyoga.pro', 'password', '{"ROLE_THERAPIST"}', now(), 1);

INSERT INTO therapists(id, first_name, last_name, created_at, version)
VALUES (1, 'Елена (тест)', 'Маркова', now(), 1);

INSERT INTO users (email, password_hash, roles, created_at, version)
VALUES ('admin@ta.pro', 'password', '{"ROLE_ADMIN"}', now(), 1);

INSERT INTO therapists(id, first_name, last_name, created_at, version)
VALUES (2, 'Админ (тест)', 'Адамов', now(), 1);

open class QYogaAppBaseTest {

    private val dataSource: DataSource = context.getBean(DataSource::class.java)

    @BeforeEach
    fun setupTestData() {
        dataSource.setupDb()
    }

}

class DbInitializer(
    private val dataSource: DataSource
) {

    fun executeScripts(vararg scripts: String) {
        dataSource.connection.use {
            scripts.forEach { script ->
                ScriptUtils.executeSqlScript(it, ClassPathResource(script))
            }
        }
    }

}

fun DataSource.setupDb() {
    DbInitializer(this).executeScripts("/db/shared-fixture.sql")
}

const val THE_THERAPIST_ID = 1L

const val THE_THERAPIST_LOGIN = "therapist@qyoga.pro"
const val THE_THERAPIST_PASSWORD = "password"

const val THE_THERAPIST_FIRST_NAME = "Елена (тест)"

val THE_THERAPIST_REF = AggregateReference.to<Therapist, Long>(THE_THERAPIST_ID)

val lowerCaseCyrillicLetters = ('а'..'я').toList()
val upperCaseCyrillicLetters = ('А'..'Я').toList()
val cyrillicLetters = lowerCaseCyrillicLetters + upperCaseCyrillicLetters

val lowerCaseLatinLetters = ('a'..'z').toList()

fun randomWord(letters: List<Char>, minLength: Int = 1, maxLength: Int = 12) =
    buildString {
        val length = Random.nextInt(minLength, maxLength)
        repeat(length) {
            append(letters[Random.nextInt(letters.size)])
        }
        check(this.length in minLength..maxLength)
    }

fun randomCyrillicWord(minLength: Int = 1, maxLength: Int = 12) =
    randomWord(cyrillicLetters, minLength, maxLength)

object ClientsObjectMother {

  fun createClientCardDto(
      firstName: String = faker.name().firstName(),
      lastName: String = faker.name().lastName(),
      middleName: String? = faker.name().nameWithMiddle().split(" ")[1],
      birthDate: LocalDate = randomBirthDate(),
      phone: String = randomPhoneNumber(),
      email: String? = randomEmail(),
      address: String? = randomCyrillicWord(),
      complains: String = randomCyrillicWord(),
      anamnesis: String? = randomSentence(),
      distributionSource: DistributionSource = randomDistributionSource(),
  ): ClientCardDto = createClientCardDtoMinimal(
      firstName,
      lastName,
      middleName,
      birthDate,
      phone,
      email,
      address,
      complains,
      anamnesis,
      distributionSource,
  )

  fun createClientCardDtoMinimal(
      firstName: String = randomCyrillicWord(),
      lastName: String = randomCyrillicWord(),
      middleName: String? = null,
      birthDate: LocalDate? = null,
      phone: String = randomPhoneNumber(),
      email: String? = null,
      address: String? = null,
      complains: String? = null,
      anamnesis: String? = null,
      distributionSource: DistributionSource? = null
  ): ClientCardDto = ClientCardDto(
      firstName,
      lastName,
      middleName,
      birthDate,
      phone,
      email,
      address,
      complains,
      anamnesis,
      distributionSource?.type,
      distributionSource?.comment,
  )

}

@Test
fun `Null distribution source comment should be persisted as null`() {
    // Given
    val therapist = TherapistClient.loginAsTheTherapist()
    val newClientRequest = ClientsObjectMother.createClientCardDto(
        distributionSource = DistributionSource(DistributionSourceType.OTHER, null)
    )

    // When
    therapist.clients.createClient(newClientRequest)

    // Then
    val clients = backgrounds.clients.getAllClients()
    clients.content.forAny { it shouldMatch newClientRequest }
}

@Test
fun `Edit page for just created client should be rendered correctly and contain empty log`() {
    // Given
    val therapist = TherapistClient.loginAsTheTherapist()
    val newClientRequest = createClientCardDto()
    val client = backgrounds.clients.createClients(listOf(newClientRequest), THE_THERAPIST_ID).first()

    // When
    val document = therapist.clients.getClientEditPage(client.id)

    // Then
    document shouldBe EmptyClientJournalPage
}