package org.kuali.ole.ingest.action;
   
   import org.apache.log4j.Logger;
   import org.joda.time.DateTime;
   import org.junit.Before;
   import org.junit.Test;
   import org.junit.runner.RunWith;
   import org.kuali.ole.BibliographicRecordHandler;
   import org.kuali.ole.DataCarrierService;
  import org.kuali.ole.OLEConstants;
  import org.kuali.ole.docstore.model.xmlpojo.ingest.Response;
  import org.kuali.ole.docstore.model.xmlpojo.work.bib.marc.DataField;
  import org.kuali.ole.docstore.model.xstream.ingest.ResponseHandler;
  import org.kuali.ole.editor.service.DiscoveryHelperService;
  import org.kuali.ole.editor.service.DocstoreHelperService;
  import org.kuali.ole.ingest.FileUtil;
  import org.kuali.ole.ingest.pojo.OverlayLookupAction;
  import org.kuali.ole.ingest.pojo.OverlayOption;
  import org.kuali.ole.ingest.pojo.ProfileAttributeBo;
  import org.kuali.ole.pojo.bib.BibliographicRecord;
  import org.kuali.ole.pojo.bib.Collection;
  import org.kuali.ole.service.OverlayHelperService;
  import org.kuali.ole.service.impl.OverlayHelperServiceImpl;
  import org.kuali.rice.krad.service.BusinessObjectService;
  import org.kuali.rice.krad.service.KRADServiceLocator;
  import org.kuali.rice.krms.api.engine.ExecutionEnvironment;
  import org.kuali.rice.krms.api.engine.ExecutionOptions;
  import org.kuali.rice.krms.api.engine.SelectionCriteria;
  import org.kuali.rice.krms.api.engine.Term;
  import org.kuali.rice.krms.api.repository.agenda.AgendaDefinition;
  import org.kuali.rice.krms.framework.engine.BasicExecutionEnvironment;
  import org.kuali.rice.krms.framework.engine.TermResolutionEngineImpl;
  import org.mockito.Mock;
  import org.mockito.Mockito;
  import org.mockito.MockitoAnnotations;
  import org.springframework.test.context.ContextConfiguration;
  import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
  import org.springframework.test.context.transaction.TransactionConfiguration;
  import org.springframework.transaction.annotation.Transactional;
  import org.springframework.util.Assert;
  
  import java.io.File;
  import java.net.URL;
  import java.util.Arrays;
  import java.util.Collections;
  import java.util.HashMap;
  import java.util.List;
  
  import static org.junit.Assert.assertEquals;
  
  /**
   * Created by IntelliJ IDEA.
   * User: premkb
   * Date: 12/11/12
   * Time: 5:36 PM
   * To change this template use File | Settings | File Templates.
   */
  
  @ContextConfiguration(locations = {"classpath:/SpringBeans.xml"})
  @RunWith(value = SpringJUnit4ClassRunner.class)
  @TransactionConfiguration(defaultRollback = true)
  
  public class UpdateBibExcludingGPFieldAction_UT {
  
      private static final Logger LOG = Logger.getLogger(UpdateBibExcludingGPFieldAction_UT.class);
      private DocstoreHelperService docstoreHelperService;
      private OverlayHelperService overlayHelperService;
      private BibliographicRecordHandler bibliographicRecordHandler;
      private BusinessObjectService businessObjectService;
      @Mock
      private ExecutionEnvironment mockExecutionEnvironment;
  
      @Mock
      private DataCarrierService mockDataCarrierService;
  
      @Before
      public void setUp() throws Exception {
          MockitoAnnotations.initMocks(this);
          docstoreHelperService = new DocstoreHelperService();
          overlayHelperService = new OverlayHelperServiceImpl();
          bibliographicRecordHandler = new BibliographicRecordHandler();
          businessObjectService = KRADServiceLocator.getBusinessObjectService();
      }
  
      @Test
      @Transactional
      public void testUpdateBibExcludingGPF()throws Exception{
          System.setProperty("app.environment", "local");
          UpdateBibExcludingGPFieldAction updateBibExcludingGPFieldAction = new MockUpdateBibExcludingGPFieldAction();
          String bibUUID = creatBibliographicRecord();
          DiscoveryHelperService discoveryHelperService = new DiscoveryHelperService();
          BibliographicRecord bibliographicRecord = docstoreHelperService.getBibliographicRecord(bibUUID);
          BibliographicRecord newBibliographicRecord= getUpdatedBibliographicRecord();
          List<ProfileAttributeBo> profileAttributeBos = getProfileAttributeBos();
          List<OverlayOption> overlayOptionList = getOverlayOption();
          List<OverlayLookupAction> overlayLookupActionList = getOverlayLookupAction();
          List list = discoveryHelperService.getResponseFromSOLR("020a", getIsbn(bibliographicRecord));
          Mockito.when(mockDataCarrierService.getData(OLEConstants.REQUEST_BIB_RECORD)).thenReturn(newBibliographicRecord);
          Mockito.when(mockDataCarrierService.getData(OLEConstants.PROFILE_ATTRIBUTE_LIST)).thenReturn(profileAttributeBos);
         Mockito.when(mockDataCarrierService.getData(OLEConstants.OVERLAY_OPTION_LIST)).thenReturn(overlayOptionList);
         Mockito.when(mockDataCarrierService.getData(OLEConstants.OVERLAY_LOOKUP_ACTION_LIST)).thenReturn(overlayLookupActionList);
         Mockito.when(mockDataCarrierService.getData(OLEConstants.BIB_INFO_LIST_FROM_SOLR_RESPONSE)).thenReturn(list);
         updateBibExcludingGPFieldAction.execute(getMockExecutionEnvironment());
         Assert.isTrue((Boolean) mockExecutionEnvironment.getEngineResults().getAttribute(OLEConstants.UPDATE_BIB_EXCLUDING_GPF));
     }
 
     public ExecutionEnvironment getMockExecutionEnvironment() {
         SelectionCriteria sc1 =
                 SelectionCriteria.createCriteria(new DateTime(), Collections.<String, String>emptyMap(), Collections.singletonMap(
                         AgendaDefinition.Constants.EVENT, "foo"));
         mockExecutionEnvironment = new BasicExecutionEnvironment(sc1, Collections.<Term, Object>emptyMap(), new ExecutionOptions(), new TermResolutionEngineImpl());
         return mockExecutionEnvironment;
     }
     private class MockUpdateBibExcludingGPFieldAction extends UpdateBibExcludingGPFieldAction {
         @Override
         protected DataCarrierService getDataCarrierService() {
             return mockDataCarrierService;    //To change body of overridden methods use File | Settings | File Templates.
         }
     }
 
     private List<ProfileAttributeBo> getProfileAttributeBos() {
         ProfileAttributeBo profileAttributeBo1 = new ProfileAttributeBo();
         profileAttributeBo1.setAgendaName(OLEConstants.PROFILE_AGENDA_NM);
         profileAttributeBo1.setId(1);
         profileAttributeBo1.setAttributeName(OLEConstants.PROFILE_ATTRIBUTE_NM);
         profileAttributeBo1.setAttributeValue(OLEConstants.PROFILE_ATTRIBUTE_VALUE);
         List<ProfileAttributeBo> mockProfileAttributes = Arrays.asList(profileAttributeBo1);
         return mockProfileAttributes;
     }
 
     public List<OverlayOption> getOverlayOption() {
         HashMap map = new HashMap();
         map.put("agenda_name", OLEConstants.PROFILE_AGENDA_NM);
         List<OverlayOption> matching = (List<OverlayOption>) businessObjectService.findMatching(OverlayOption.class, map);
         return matching;
     }
 
     public List<OverlayLookupAction> getOverlayLookupAction() {
         HashMap map = new HashMap();
         map.put("agenda_name", OLEConstants.PROFILE_AGENDA_NM);
         List<OverlayLookupAction> matching = (List<OverlayLookupAction>) businessObjectService.findMatching(OverlayLookupAction.class, map);
         return matching;
     }
 
     private String getIsbn(BibliographicRecord bibliographicRecord){
         String isbn=null;
         List<DataField> marcDataFields = bibliographicRecord.getDatafields() ;
         for(DataField marcDataField :marcDataFields){
             if(marcDataField.getTag().equals("020")){
                 isbn = marcDataField.getSubFields().get(0).getValue();
                 break;
             }
         }
         return isbn;
     }
 
     public String creatBibliographicRecord()throws Exception{
         URL resource = getClass().getResource("sample-marc.xml");
         File file = new File(resource.toURI());
         String xmlContent = new FileUtil().readFile(file);
         String responseFromDocstore = docstoreHelperService.persistToDocstoreForEditor(xmlContent, null, "marc");
         ResponseHandler responseHandler = new ResponseHandler();
         Response response = responseHandler.toObject(responseFromDocstore);
         String bibUUID = overlayHelperService.getUUID(response,OLEConstants.BIB_DOC_TYPE);
         if(LOG.isInfoEnabled()){
             LOG.info("bibUUID------------->"+bibUUID);
         }
         return bibUUID;
     }
 
     public BibliographicRecord getUpdatedBibliographicRecord()throws Exception{
         System.setProperty("app.environment", "local");
         URL resource = getClass().getResource("sample-updated-marc.xml");
         File file = new File(resource.toURI());
         String xmlContent = new FileUtil().readFile(file);
         Collection bibliographicRecords = bibliographicRecordHandler.fromXML(xmlContent);
         List<BibliographicRecord> records = bibliographicRecords.getRecords();
         BibliographicRecord bibliographicRecord = records.get(0);
         return bibliographicRecord;
     }
 }