/** * Copyright or © or Copr. Ministère Français chargé de la Culture * et de la Communication (2013) * <p/> * contact.gincoculture_at_gouv.fr * <p/> * This software is a computer program whose purpose is to provide a thesaurus * management solution. * <p/> * This software is governed by the CeCILL license under French law and * abiding by the rules of distribution of free software. You can use, * modify and/ or redistribute the software under the terms of the CeCILL * license as circulated by CEA, CNRS and INRIA at the following URL * "http://www.cecill.info". * <p/> * As a counterpart to the access to the source code and rights to copy, * modify and redistribute granted by the license, users are provided only * with a limited warranty and the software's author, the holder of the * economic rights, and the successive licensors have only limited liability. * <p/> * In this respect, the user's attention is drawn to the risks associated * with loading, using, modifying and/or developing or reproducing the * software by the user in light of its specific status of free software, * that may mean that it is complicated to manipulate, and that also * therefore means that it is reserved for developers and experienced * professionals having in-depth computer knowledge. Users are therefore * encouraged to load and test the software's suitability as regards their * requirements in conditions enabling the security of their systemsand/or * data to be ensured and, more generally, to use and operate it in the * same conditions as regards security. * <p/> * The fact that you are presently reading this means that you have had * knowledge of the CeCILL license and that you accept its terms. */ package fr.mcc.ginco.tests.solr; import java.io.IOException; import java.util.ArrayList; import java.util.List; import junit.framework.Assert; import org.apache.solr.client.solrj.SolrServerException; import org.apache.solr.common.SolrInputDocument; import org.junit.Before; import org.junit.Test; import org.mockito.InjectMocks; import org.mockito.Mock; import org.mockito.Mockito; import org.mockito.MockitoAnnotations; import fr.mcc.ginco.beans.Language; import fr.mcc.ginco.beans.Note; import fr.mcc.ginco.beans.Thesaurus; import fr.mcc.ginco.beans.ThesaurusConcept; import fr.mcc.ginco.beans.ThesaurusTerm; import fr.mcc.ginco.services.INoteService; import fr.mcc.ginco.services.IThesaurusConceptService; import fr.mcc.ginco.solr.ConceptSolrConverter; import fr.mcc.ginco.solr.ExtEntityType; import fr.mcc.ginco.solr.SolrField; import fr.mcc.ginco.utils.DateUtil; public class ConceptSolrConverterTest { @InjectMocks private ConceptSolrConverter conceptSolrConverter; @Mock(name = "noteService") private INoteService noteService; @Mock(name = "thesaurusConceptService") private IThesaurusConceptService thesaurusConceptService; @Before public void init() { MockitoAnnotations.initMocks(this); } @Test public void testConvertSolrConcept() throws SolrServerException, IOException { Thesaurus fakeThesaurus = new Thesaurus(); fakeThesaurus.setIdentifier("http://th1"); fakeThesaurus.setTitle("Fake thesaurus"); Language langFR = new Language(); langFR.setId("fr-FR"); Language langEN = new Language(); langFR.setId("en-US"); ThesaurusConcept fakeThesaurusConcept = new ThesaurusConcept(); fakeThesaurusConcept.setIdentifier("http://c1"); fakeThesaurusConcept.setThesaurus(fakeThesaurus); fakeThesaurusConcept.setCreated(DateUtil.nowDate()); fakeThesaurusConcept.setModified(DateUtil.nowDate()); fakeThesaurusConcept.setStatus(1); ThesaurusTerm fakeFRPrefTerm = new ThesaurusTerm(); fakeFRPrefTerm.setLanguage(langFR); ThesaurusTerm fakeENPrefTerm = new ThesaurusTerm(); fakeENPrefTerm.setLanguage(langEN); List<ThesaurusTerm> fakePrefTermList = new ArrayList<ThesaurusTerm>(); fakePrefTermList.add(fakeFRPrefTerm); fakePrefTermList.add(fakeENPrefTerm); List<Note> mockListNote = new ArrayList<Note>(); Note fakeNote1 = new Note(); fakeNote1.setLexicalValue("Fake note 1"); Note fakeNote2 = new Note(); fakeNote2.setLexicalValue("Fake note 2"); mockListNote.add(fakeNote1); mockListNote.add(fakeNote2); Mockito.when( thesaurusConceptService.getConceptPreferredTerm(fakeThesaurusConcept.getIdentifier())).thenReturn(fakeFRPrefTerm); Mockito.when( thesaurusConceptService.getConceptPreferredTerms(fakeThesaurusConcept.getIdentifier())).thenReturn(fakePrefTermList); Mockito.when( noteService.getConceptNotePaginatedList(Mockito.anyString(), Mockito.anyInt(), Mockito.anyInt())).thenReturn(mockListNote); SolrInputDocument doc = conceptSolrConverter.convertSolrConcept(fakeThesaurusConcept); Assert.assertEquals(fakeThesaurus.getIdentifier(), doc .getFieldValue(SolrField.THESAURUSID)); Assert.assertEquals(fakeThesaurus.getTitle(), doc .getFieldValue(SolrField.THESAURUSTITLE)); Assert.assertEquals(fakeThesaurusConcept.getIdentifier(), doc .getFieldValue(SolrField.IDENTIFIER)); Assert.assertEquals(fakeFRPrefTerm.getLexicalValue(), doc .getFieldValue(SolrField.LEXICALVALUE)); Assert.assertEquals(ThesaurusConcept.class.getSimpleName(), doc .getFieldValue(SolrField.TYPE)); Assert.assertEquals(ExtEntityType.CONCEPT, doc .getFieldValue(SolrField.EXT_TYPE)); Assert.assertEquals(fakeThesaurusConcept.getModified(), doc .getFieldValue(SolrField.MODIFIED)); Assert.assertEquals(fakeThesaurusConcept.getCreated(), doc .getFieldValue(SolrField.CREATED)); Assert.assertEquals(fakeThesaurusConcept.getStatus(), doc .getFieldValue(SolrField.STATUS)); Assert.assertTrue(doc .getFieldValues(SolrField.NOTES).contains(fakeNote1.getLexicalValue())); Assert.assertTrue(doc .getFieldValues(SolrField.NOTES).contains(fakeNote2.getLexicalValue())); Assert.assertTrue(doc .getFieldValues(SolrField.LANGUAGE).contains(fakeFRPrefTerm.getLanguage().getId())); Assert.assertTrue(doc .getFieldValues(SolrField.LANGUAGE).contains(fakeENPrefTerm.getLanguage().getId())); } }