Skip to content

Commit

Permalink
HSEARCH-3589 Test accessing the indexed entities and their index mana…
Browse files Browse the repository at this point in the history 
…ger in the ORM mapper
  • Loading branch information
@yrodiere
yrodiere committed May 6, 2020
1 parent 6a019ab commit b82116a
Show file tree
Hide file tree
Showing 3 changed files with 160 additions and 16 deletions.
168 changes: 152 additions & 16 deletions ...rc/test/java/org/hibernate/search/integrationtest/mapper/orm/mapping/SearchMappingIT.java
View file
Expand Up @@ -6,26 +6,38 @@
*/
package org.hibernate.search.integrationtest.mapper.orm.mapping;

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

import java.util.Collection;
import java.util.List;
import javax.persistence.Entity;
import javax.persistence.Id;
import javax.persistence.ManyToOne;
import javax.persistence.OneToMany;

import org.hibernate.SessionFactory;
import org.hibernate.search.engine.backend.Backend;
import org.hibernate.search.engine.backend.index.IndexManager;
import org.hibernate.search.mapper.orm.Search;
import org.hibernate.search.mapper.orm.mapping.SearchIndexedEntity;
import org.hibernate.search.mapper.orm.mapping.SearchMapping;
import org.hibernate.search.mapper.pojo.mapping.definition.annotation.GenericField;
import org.hibernate.search.mapper.pojo.mapping.definition.annotation.Indexed;
import org.hibernate.search.util.common.SearchException;
import org.hibernate.search.util.impl.integrationtest.common.rule.BackendMock;
import org.hibernate.search.util.impl.integrationtest.common.stub.backend.index.impl.StubBackend;
import org.hibernate.search.util.impl.integrationtest.common.stub.backend.index.impl.StubIndexManager;
import org.hibernate.search.util.impl.integrationtest.mapper.orm.OrmSetupHelper;
import org.hibernate.search.util.impl.test.annotation.TestForIssue;

import org.hibernate.testing.TestForIssue;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;

import org.assertj.core.api.Assertions;
import org.assertj.core.api.InstanceOfAssertFactories;

@TestForIssue( jiraKey = "HSEARCH-3640" )
public class SearchMappingIT {

private static final String BACKEND_1_NAME = "stubBackend1";
Expand All @@ -40,45 +52,139 @@ public class SearchMappingIT {
@Rule
public OrmSetupHelper ormSetupHelper = OrmSetupHelper.withBackendMocks( backendMock1, backendMock2 );

private SessionFactory sessionFactory;
private SearchMapping mapping;

@Before
public void before() {
backendMock1.expectAnySchema( Person.INDEX_NAME );
backendMock2.expectAnySchema( Pet.JPA_ENTITY_NAME );
sessionFactory = ormSetupHelper.start().setup( Person.class, Pet.class );
SessionFactory sessionFactory = ormSetupHelper.start().setup( Person.class, Pet.class, Toy.class );
backendMock1.verifyExpectationsMet();
backendMock2.verifyExpectationsMet();
mapping = Search.mapping( sessionFactory );
}

@Test
@TestForIssue(jiraKey = "HSEARCH-3589")
public void indexedEntity_byName() {
SearchIndexedEntity entity = mapping.indexedEntity( Person.JPA_ENTITY_NAME );
assertThat( entity )
.isNotNull()
.returns( Person.JPA_ENTITY_NAME, SearchIndexedEntity::jpaName )
.returns( Person.class, SearchIndexedEntity::javaClass );
checkIndexManager( Person.INDEX_NAME, entity.indexManager() );
}

@Test
@TestForIssue(jiraKey = "HSEARCH-3589")
public void indexedEntity_byName_notEntity() {
assertThatThrownBy( () -> mapping.indexedEntity( "invalid" ) )
.isInstanceOf( SearchException.class )
.hasMessageContainingAll(
"Unknown entity name: 'invalid'",
"Available entity names:",
Person.class.getName(),
Person.JPA_ENTITY_NAME,
Pet.class.getName(),
Pet.JPA_ENTITY_NAME,
Toy.class.getName(),
Toy.JPA_ENTITY_NAME
);
}

@Test
@TestForIssue(jiraKey = "HSEARCH-3589")
public void indexedEntity_byName_notIndexed() {
assertThatThrownBy( () -> mapping.indexedEntity( Toy.JPA_ENTITY_NAME ) )
.isInstanceOf( SearchException.class )
.hasMessageContaining(
"Entity '" + Toy.JPA_ENTITY_NAME + "' is not indexed"
);
}

@Test
@TestForIssue(jiraKey = "HSEARCH-3589")
public void indexedEntity_byJavaClass() {
SearchIndexedEntity entity = mapping.indexedEntity( Person.class );
assertThat( entity )
.isNotNull()
.returns( Person.JPA_ENTITY_NAME, SearchIndexedEntity::jpaName )
.returns( Person.class, SearchIndexedEntity::javaClass );
checkIndexManager( Person.INDEX_NAME, entity.indexManager() );
}

@Test
@TestForIssue(jiraKey = "HSEARCH-3589")
public void indexedEntity_byJavaClass_notEntity() {
assertThatThrownBy( () -> mapping.indexedEntity( String.class ) )
.isInstanceOf( SearchException.class )
.hasMessageContaining(
"Type '" + String.class.getName() + "' is not an entity type, or the entity is not indexed"
);
}

@Test
@TestForIssue(jiraKey = "HSEARCH-3589")
public void indexedEntity_byJavaClass_notIndexed() {
assertThatThrownBy( () -> mapping.indexedEntity( Toy.class ) )
.isInstanceOf( SearchException.class )
.hasMessageContaining(
"Type '" + Toy.class.getName() + "' is not an entity type, or the entity is not indexed"
);
}

@Test
@TestForIssue(jiraKey = "HSEARCH-3589")
public void allIndexedEntities() {
Collection<? extends SearchIndexedEntity> entities = mapping.allIndexedEntities();
Assertions.<SearchIndexedEntity>assertThat( entities )
.extracting( SearchIndexedEntity::jpaName )
.containsExactlyInAnyOrder(
Person.JPA_ENTITY_NAME,
Pet.JPA_ENTITY_NAME
);
}

@Test
@TestForIssue(jiraKey = "HSEARCH-3640")
public void getIndexManager_customIndexName() {
SearchMapping target = Search.mapping( sessionFactory );
IndexManager indexManager = target.getIndexManager( Person.INDEX_NAME );
Assertions.assertThat( indexManager ).isNotNull();
IndexManager indexManager = mapping.getIndexManager( Person.INDEX_NAME );
checkIndexManager( Person.INDEX_NAME, indexManager );
}

@Test
@TestForIssue(jiraKey = "HSEARCH-3640")
public void getIndexManager_defaultIndexName() {
SearchMapping target = Search.mapping( sessionFactory );
IndexManager indexManager = target.getIndexManager( Pet.JPA_ENTITY_NAME );
Assertions.assertThat( indexManager ).isNotNull();
IndexManager indexManager = mapping.getIndexManager( Pet.JPA_ENTITY_NAME );
checkIndexManager( Pet.JPA_ENTITY_NAME, indexManager );
}

@Test
@TestForIssue(jiraKey = "HSEARCH-3640")
public void getBackendByName() {
SearchMapping target = Search.mapping( sessionFactory );
Backend backend = mapping.getBackend( BACKEND_1_NAME );
checkBackend( BACKEND_1_NAME, backend );

Object backend = target.getBackend( BACKEND_1_NAME );
Assertions.assertThat( backend ).isNotNull();
backend = mapping.getBackend( BACKEND_2_NAME );
checkBackend( BACKEND_2_NAME, backend );
}

private void checkIndexManager(String expectedIndexName, IndexManager indexManager) {
assertThat( indexManager )
.asInstanceOf( InstanceOfAssertFactories.type( StubIndexManager.class ) )
.returns( expectedIndexName, StubIndexManager::getName );
}

backend = target.getBackend( BACKEND_2_NAME );
Assertions.assertThat( backend ).isNotNull();
private void checkBackend(String expectedBackendName, Backend indexManager) {
assertThat( indexManager )
.asInstanceOf( InstanceOfAssertFactories.type( StubBackend.class ) )
.returns( expectedBackendName, StubBackend::getName );
}

@Entity(name = "PersonEntity")
@Entity(name = Person.JPA_ENTITY_NAME)
@Indexed(index = Person.INDEX_NAME, backend = BACKEND_1_NAME)
private static class Person {
public static final String JPA_ENTITY_NAME = "PersonEntity";
public static final String INDEX_NAME = "Person";

@Id
Expand Down Expand Up @@ -106,6 +212,8 @@ private static class Pet {
private Integer id;
@GenericField
private String nickname;
@OneToMany(mappedBy = "owner")
private List<Toy> toys;

public Pet() {
}
Expand All @@ -116,5 +224,33 @@ public Integer getId() {
public String getNickname() {
return nickname;
}
public List<Toy> getToys() {
return toys;
}
}

@Entity(name = Toy.JPA_ENTITY_NAME)
private static class Toy {
public static final String JPA_ENTITY_NAME = "Toy";

@Id
private Integer id;
@GenericField
private String name;
@ManyToOne
private Pet owner;

public Toy() {
}

public Integer getId() {
return id;
}
public String getName() {
return name;
}
public Pet getOwner() {
return owner;
}
}
}
4 changes: 4 additions & 0 deletions ...ibernate/search/util/impl/integrationtest/common/stub/backend/index/impl/StubBackend.java
View file
Expand Up @@ -31,6 +31,10 @@ public String toString() {
return StubBackend.class.getSimpleName() + "[" + name + "]";
}

public String getName() {
return name;
}

@Override
public void start(BackendStartContext context) {
// Nothing to do
Expand Down
4 changes: 4 additions & 0 deletions ...ate/search/util/impl/integrationtest/common/stub/backend/index/impl/StubIndexManager.java
View file
Expand Up @@ -82,6 +82,10 @@ public String toString() {
return StubIndexManager.class.getSimpleName() + "[" + name + "]";
}

public String getName() {
return name;
}

@Override
public IndexSchemaManager getSchemaManager() {
checkStarted();
Expand Down

0 comments on commit b82116a

Please sign in to comment.