package io.quarkus.hibernate.search.elasticsearch.test.multiplepersistenceunits;

import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatThrownBy;

import javax.enterprise.context.control.ActivateRequestContext;
import javax.enterprise.inject.Instance;
import javax.enterprise.inject.spi.CDI;
import javax.inject.Inject;
import javax.persistence.EntityManager;
import javax.transaction.UserTransaction;

import org.hibernate.search.mapper.orm.Search;
import org.hibernate.search.mapper.orm.entity.SearchIndexedEntity;
import org.hibernate.search.mapper.orm.mapping.SearchMapping;
import org.hibernate.search.mapper.orm.session.SearchSession;
import org.hibernate.search.util.common.SearchException;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.RegisterExtension;

import io.quarkus.hibernate.orm.PersistenceUnit;
import io.quarkus.hibernate.search.elasticsearch.test.multiplepersistenceunits.defaultpu.DefaultPUEntity;
import io.quarkus.hibernate.search.elasticsearch.test.multiplepersistenceunits.pu1.PU1Entity;
import io.quarkus.hibernate.search.elasticsearch.test.multiplepersistenceunits.pu2.PU2Entity;
import io.quarkus.hibernate.search.elasticsearch.test.multiplepersistenceunits.pu3.PU3Entity;
import io.quarkus.hibernate.search.elasticsearch.test.util.TransactionUtils;
import io.quarkus.test.QuarkusUnitTest;

public class MultiplePersistenceUnitsCdiTest {

    @RegisterExtension
    static QuarkusUnitTest runner = new QuarkusUnitTest()
            .withApplicationRoot((jar) -> jar
                    .addClass(TransactionUtils.class)
                    .addPackage(DefaultPUEntity.class.getPackage())
                    .addPackage(PU1Entity.class.getPackage())
                    .addPackage(PU2Entity.class.getPackage())
                    .addPackage(PU3Entity.class.getPackage())
                    .addAsResource("application-multiple-persistence-units.properties", "application.properties"));

    @Inject
    SearchMapping defaultPUMapping;

    @Inject
    SearchSession defaultPUSession;

    @Inject
    @PersistenceUnit("pu1")
    SearchMapping pu1Mapping;

    @Inject
    @PersistenceUnit("pu1")
    SearchSession pu1Session;

    @Inject
    @PersistenceUnit("pu2")
    SearchMapping pu2Mapping;

    @Inject
    @PersistenceUnit("pu2")
    SearchSession pu2Session;

    @Inject
    @PersistenceUnit("pu3")
    EntityManager pu3EntityManager;

    @Inject
    UserTransaction transaction;

    @Test
    public void testDefaultMapping() {
        assertThat(defaultPUMapping.allIndexedEntities())
                .hasSize(1)
                .element(0)
                .returns(DefaultPUEntity.class, SearchIndexedEntity::javaClass);
    }

    @Test
    @ActivateRequestContext
    public void testDefaultSession() {
        DefaultPUEntity entity = new DefaultPUEntity("someText");
        inTransaction(() -> defaultPUSession.toEntityManager().persist(entity));
        inTransaction(() -> assertThat(defaultPUSession.search(DefaultPUEntity.class)
                .where(f -> f.matchAll())
                .fetchHits(20))
                        .hasSize(1)
                        .element(0)
                        .returns(entity.getId(), DefaultPUEntity::getId));
    }

    @Test
    public void testPU1Mapping() {
        assertThat(pu1Mapping.allIndexedEntities())
                .hasSize(1)
                .element(0)
                .returns(PU1Entity.class, SearchIndexedEntity::javaClass);
    }

    @Test
    @ActivateRequestContext
    public void testPU1Session() {
        PU1Entity entity = new PU1Entity("someText");
        inTransaction(() -> pu1Session.toEntityManager().persist(entity));
        inTransaction(() -> assertThat(pu1Session.search(PU1Entity.class)
                .where(f -> f.matchAll())
                .fetchHits(20))
                        .hasSize(1)
                        .element(0)
                        .returns(entity.getId(), PU1Entity::getId));
    }

    @Test
    public void testPU2Mapping() {
        assertThat(pu2Mapping.allIndexedEntities())
                .hasSize(1)
                .element(0)
                .returns(PU2Entity.class, SearchIndexedEntity::javaClass);
    }

    @Test
    @ActivateRequestContext
    public void testPU2Session() {
        PU2Entity entity = new PU2Entity("someText");
        inTransaction(() -> pu1Session.toEntityManager().persist(entity));
        inTransaction(() -> assertThat(pu1Session.search(PU2Entity.class)
                .where(f -> f.matchAll())
                .fetchHits(20))
                        .hasSize(1)
                        .element(0)
                        .returns(entity.getId(), PU2Entity::getId));
    }

    @Test
    public void testPU3Mapping() {
        // There are no indexed entities in PU3: Hibernate Search should be disabled

        Instance<SearchMapping> pu3MappingInstance = CDI.current().getBeanManager().createInstance()
                .select(SearchMapping.class, new PersistenceUnit.PersistenceUnitLiteral("pu3"));
        assertThat(pu3MappingInstance.isUnsatisfied()).isTrue();

        assertThatThrownBy(() -> Search.mapping(pu3EntityManager.getEntityManagerFactory()))
                .isInstanceOf(SearchException.class)
                .hasMessageContaining("Hibernate Search was not initialized");
    }

    @Test
    @ActivateRequestContext
    public void testPU3Session() {
        // There are no indexed entities in PU3: Hibernate Search should be disabled

        Instance<SearchSession> pu3SessionInstance = CDI.current().getBeanManager().createInstance()
                .select(SearchSession.class, new PersistenceUnit.PersistenceUnitLiteral("pu3"));
        assertThat(pu3SessionInstance.isUnsatisfied()).isTrue();

        inTransaction(() -> assertThatThrownBy(() -> Search.session(pu3EntityManager).search(PU3Entity.class))
                .isInstanceOf(SearchException.class)
                .hasMessageContaining("Hibernate Search was not initialized"));
    }

    private void inTransaction(Runnable runnable) {
        TransactionUtils.inTransaction(transaction, runnable);
    }
}

@@ -4411,33 +4411,33 @@
 saction(() -%3E pu
-1
+2
 Session.toEntity
@@ -4497,33 +4497,33 @@
 -%3E assertThat(pu
-1
+2
 Session.search(P

Fix incorrect test code