Example usage for org.apache.solr.client.solrj.response UpdateResponse UpdateResponse

Introduction

In this page you can find the example usage for org.apache.solr.client.solrj.response UpdateResponse UpdateResponse.

Prototype

UpdateResponse

Source Link

Usage

From source file:at.pagu.soldockr.core.SolrTemplateTest.java

License:Apache License

@Test
public void testAddBean() throws IOException, SolrServerException {
    Mockito.when(solrServerMock.add(Mockito.any(SolrInputDocument.class))).thenReturn(new UpdateResponse());
    UpdateResponse updateResponse = solrTemplate.executeAddBean(SIMPLE_OBJECT);
    Assert.assertNotNull(updateResponse);

    ArgumentCaptor<SolrInputDocument> captor = ArgumentCaptor.forClass(SolrInputDocument.class);
    Mockito.verify(solrServerMock, Mockito.times(1)).add(captor.capture());

    Assert.assertEquals(SIMPLE_OBJECT.getId(), captor.getValue().getFieldValue("id"));
    Assert.assertEquals(SIMPLE_OBJECT.getValue(), captor.getValue().getFieldValue("value"));
}

From source file:at.pagu.soldockr.core.SolrTemplateTest.java

License:Apache License

@SuppressWarnings("unchecked")
@Test//  w  w w.j a  v a 2  s  . c  o  m
public void testAddBeans() throws IOException, SolrServerException {
    Mockito.when(solrServerMock.add(Mockito.anyCollectionOf(SolrInputDocument.class)))
            .thenReturn(new UpdateResponse());
    List<SimpleJavaObject> collection = Arrays.asList(new SimpleJavaObject("1", 1l),
            new SimpleJavaObject("2", 2l), new SimpleJavaObject("3", 3l));
    UpdateResponse updateResponse = solrTemplate.executeAddBeans(collection);
    Assert.assertNotNull(updateResponse);

    @SuppressWarnings("rawtypes")
    ArgumentCaptor<List> captor = ArgumentCaptor.forClass(List.class);
    Mockito.verify(solrServerMock, Mockito.times(1)).add(captor.capture());

    Assert.assertEquals(3, captor.getValue().size());
}

From source file:at.pagu.soldockr.core.SolrTemplateTest.java

License:Apache License

@Test
public void testAddDocument() throws IOException, SolrServerException {
    Mockito.when(solrServerMock.add(Mockito.any(SolrInputDocument.class))).thenReturn(new UpdateResponse());
    UpdateResponse updateResponse = solrTemplate.executeAddDocument(SIMPLE_DOCUMENT);
    Assert.assertNotNull(updateResponse);
    Mockito.verify(solrServerMock, Mockito.times(1)).add(Mockito.eq(SIMPLE_DOCUMENT));
}

From source file:at.pagu.soldockr.core.SolrTemplateTest.java

License:Apache License

@Test
public void testAddDocuments() throws IOException, SolrServerException {
    Mockito.when(solrServerMock.add(Mockito.anyCollectionOf(SolrInputDocument.class)))
            .thenReturn(new UpdateResponse());
    List<SolrInputDocument> collection = Arrays.asList(SIMPLE_DOCUMENT);
    UpdateResponse updateResponse = solrTemplate.executeAddDocuments(collection);
    Assert.assertNotNull(updateResponse);
    Mockito.verify(solrServerMock, Mockito.times(1)).add(Mockito.eq(collection));
}

From source file:at.pagu.soldockr.core.SolrTemplateTest.java

License:Apache License

@Test
public void testDeleteById() throws IOException, SolrServerException {
    Mockito.when(solrServerMock.deleteById(Mockito.anyString())).thenReturn(new UpdateResponse());
    UpdateResponse updateResponse = solrTemplate.executeDeleteById("1");
    Assert.assertNotNull(updateResponse);
    Mockito.verify(solrServerMock, Mockito.times(1)).deleteById(Mockito.eq("1"));
}

From source file:at.pagu.soldockr.core.SolrTemplateTest.java

License:Apache License

@Test
public void testDeleteByIdWithCollection() throws IOException, SolrServerException {
    Mockito.when(solrServerMock.deleteById(Mockito.anyListOf(String.class))).thenReturn(new UpdateResponse());
    List<String> idsToDelete = Arrays.asList("1", "2");
    UpdateResponse updateResponse = solrTemplate.executeDeleteById(idsToDelete);
    Assert.assertNotNull(updateResponse);

    @SuppressWarnings("unchecked")
    ArgumentCaptor<List<String>> captor = (ArgumentCaptor<List<String>>) (Object) ArgumentCaptor
            .forClass(List.class);

    Mockito.verify(solrServerMock, Mockito.times(1)).deleteById(captor.capture());

    Assert.assertEquals(idsToDelete.size(), captor.getValue().size());
    for (String s : idsToDelete) {
        Assert.assertTrue(captor.getValue().contains(s));
    }//w  w  w . ja  va2  s  .  co  m
}

From source file:at.pagu.soldockr.core.SolrTemplateTest.java

License:Apache License

@Test
public void testCommit() throws SolrServerException, IOException {
    Mockito.when(solrServerMock.commit()).thenReturn(new UpdateResponse());
    solrTemplate.executeCommit();/*  www  .  jav a 2s  .c om*/
    Mockito.verify(solrServerMock, Mockito.times(1)).commit();
}

From source file:at.pagu.soldockr.core.SolrTemplateTest.java

License:Apache License

@Test
public void testRollback() throws SolrServerException, IOException {
    Mockito.when(solrServerMock.rollback()).thenReturn(new UpdateResponse());
    solrTemplate.executeRollback();/*from  w  ww .  ja v a 2  s . c o  m*/
    Mockito.verify(solrServerMock, Mockito.times(1)).rollback();
}

From source file:com.cloudera.cdk.morphline.solr.CollectingDocumentLoader.java

License:Apache License

@Override
public UpdateResponse rollbackTransaction() {
    LOGGER.trace("rollback");
    return new UpdateResponse();
}

From source file:com.cloudera.cdk.morphline.solr.SolrServerDocumentLoader.java

License:Apache License

@Override
public UpdateResponse rollbackTransaction() throws SolrServerException, IOException {
    LOGGER.trace("rollback");
    if (!(server instanceof CloudSolrServer)) {
        return server.rollback();
    } else {/*from   w w w.ja  va  2 s  . c  om*/
        return new UpdateResponse();
    }
}