Commit 2affbe92 authored by Noel Alonso's avatar Noel Alonso
Browse files

Arregla tests de documentReport y speciesReport

parent c95cb512

src/test/java/es/redmic/test/tasks/integration/report/document/ReportDocumentTest.java

+43 −3
Original line number Diff line number Diff line
@@ -23,23 +23,63 @@ package es.redmic.test.tasks.integration.report.document; 
import java.util.HashMap;

import java.util.Map;



import static org.mockito.ArgumentMatchers.any;

import static org.mockito.Mockito.when;



import org.junit.Before;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.beans.factory.annotation.Value;

import org.springframework.boot.test.mock.mockito.MockBean;

import org.springframework.kafka.annotation.KafkaListener;

import org.springframework.test.context.TestPropertySource;

import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;



import es.redmic.brokerlib.avro.common.MessageWrapper;

import es.redmic.es.administrative.repository.ActivityESRepository;

import es.redmic.es.administrative.repository.DocumentESRepository;

import es.redmic.models.es.administrative.model.Activity;

import es.redmic.models.es.administrative.model.Document;

import es.redmic.models.es.data.common.model.DataSearchWrapper;

import es.redmic.test.tasks.integration.report.common.ReportBaseTest;



@RunWith(SpringJUnit4ClassRunner.class)

@TestPropertySource(properties = { "schema.registry.port=18088" })

public class ReportDocumentTest extends ReportBaseTest {



	private static final Long DOCUMENT_ID = 1536L;



	private static final Long ACTIVITY_ID = 44L;



	@Value("${broker.topic.task.report.document.run}")

	String REPORT_DOCUMENT_RUN_TOPIC;



	@MockBean

	ActivityESRepository activityRepository;



	@Autowired

	DocumentESRepository repository;



	@Before

	public void config() {



		Document document = new Document();

		document.setId(DOCUMENT_ID);

		document.setTitle("Documento de prueba");

		document.setLanguage("es");

		repository.save(document);



		Activity activity = new Activity();

		activity.setId(ACTIVITY_ID);

		activity.setName("Prueba");



		DataSearchWrapper<Activity> result = new DataSearchWrapper<>();

		result.setSource(0, activity);



		when(activityRepository.findByDocuments(any(), any())).thenReturn(result);

	}



	@Override

	@Test

	public void runIngestData_ReturnStatusMsg_IfBrokerIsListen() throws Exception {
@@ -62,7 +102,7 @@ public class ReportDocumentTest extends ReportBaseTest { 
	protected Map<String, String> createRunTaskRequest() {



		Map<String, String> request = new HashMap<>();

		request.put("selectId", "1536");

		request.put("selectId", DOCUMENT_ID.toString());

		return request;

	}

}

src/test/java/es/redmic/test/tasks/integration/report/species/ReportSpeciesTest.java

+77 −3
Original line number Diff line number Diff line 
package es.redmic.test.tasks.integration.report.species;



import static org.mockito.ArgumentMatchers.any;

import static org.mockito.Mockito.when;



import java.util.ArrayList;



/*-

 * #%L

 * Tasks
@@ -21,25 +26,94 @@ package es.redmic.test.tasks.integration.report.species; 
 */



import java.util.HashMap;

import java.util.List;

import java.util.Map;



import org.junit.Before;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.springframework.beans.factory.annotation.Value;

import org.springframework.boot.test.mock.mockito.MockBean;

import org.springframework.kafka.annotation.KafkaListener;

import org.springframework.test.context.TestPropertySource;

import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;



import es.redmic.brokerlib.avro.common.MessageWrapper;

import es.redmic.es.administrative.taxonomy.service.SpeciesESService;

import es.redmic.es.administrative.taxonomy.service.TaxonESService;

import es.redmic.models.es.administrative.dto.ActivityDTO;

import es.redmic.models.es.administrative.dto.DocumentDTO;

import es.redmic.models.es.administrative.taxonomy.dto.SpeciesDTO;

import es.redmic.models.es.administrative.taxonomy.dto.TaxonDTO;

import es.redmic.models.es.administrative.taxonomy.model.Species;

import es.redmic.models.es.common.dto.JSONCollectionDTO;

import es.redmic.models.es.common.model.DomainES;

import es.redmic.models.es.data.common.model.DataHitWrapper;

import es.redmic.models.es.maintenance.taxonomy.dto.RankDTO;

import es.redmic.test.tasks.integration.report.common.ReportBaseTest;



@RunWith(SpringJUnit4ClassRunner.class)

@TestPropertySource(properties = { "schema.registry.port=18091" })

public class ReportSpeciesTest extends ReportBaseTest {



	private static final Long SPECIES_ID = 15L;



	@MockBean

	SpeciesESService service;



	@MockBean

	TaxonESService taxonService;



	@Value("${broker.topic.task.report.species.run}")

	String REPORT_SPECIES_RUN_TOPIC;



	@Before

	public void config() {



		SpeciesDTO speciesDTO = new SpeciesDTO();

		speciesDTO.setId(SPECIES_ID);

		speciesDTO.setPath("r.1." + SPECIES_ID);

		speciesDTO.setScientificName("scientificname");



		RankDTO rank = new RankDTO();

		rank.setId(10L);

		rank.setName("Especie");

		rank.setName_en("Species");

		speciesDTO.setRank(rank);



		when(service.get(any())).thenReturn(speciesDTO);



		TaxonDTO parent = new TaxonDTO();

		parent.setId(1L);

		RankDTO rankParent = new RankDTO();

		rankParent.setId(1L);

		parent.setRank(rankParent);



		JSONCollectionDTO ancestors = new JSONCollectionDTO();

		ancestors.addData(parent);

		when(taxonService.mget(any())).thenReturn(ancestors);



		JSONCollectionDTO activities = new JSONCollectionDTO();

		List<ActivityDTO> data = new ArrayList<>();

		ActivityDTO activity = new ActivityDTO();

		activity.setId(12L);

		activity.setName("Prueba");

		data.add(activity);

		activities.setData(data);



		when(service.getActivities(any(), any())).thenReturn(activities);



		JSONCollectionDTO documents = new JSONCollectionDTO();

		List<DocumentDTO> docList = new ArrayList<>();

		DocumentDTO document = new DocumentDTO();

		document.setId(123L);

		document.setTitle("Documento de prueba");

		document.setLanguage("es");

		docList.add(document);

		documents.setData(docList);

		when(service.getDocuments(any(), any())).thenReturn(documents);

	}



	@Override

	@Test

	public void runIngestData_ReturnStatusMsg_IfBrokerIsListen() throws Exception {
@@ -62,7 +136,7 @@ public class ReportSpeciesTest extends ReportBaseTest { 
	protected Map<String, String> createRunTaskRequest() {



		Map<String, String> request = new HashMap<>();

		request.put("selectId", "20144");

		request.put("selectId", SPECIES_ID.toString());

		return request;

	}

}