Example usage for org.apache.solr.common SolrInputDocument size

Introduction

In this page you can find the example usage for org.apache.solr.common SolrInputDocument size.

Prototype

@Override
    public int size() 

Source Link

Usage

From source file:com.talis.rdf.solr.DefaultDocumentBuilderTest.java

License:Apache License

@Test
public void getDocumentAddsDefaultFields() {
    SolrInputDocument doc = quadsToDoc.getDocument(DOCUMENT_KEY, new ArrayList<Quad>());
    assertNotNull(doc);//  www .  j  a  v a2s.  c om
    assertEquals(SUBJECT_URI, doc.getField(FieldNames.DOCUMENT_KEY).getValue());
    assertEquals(SUBJECT_URI, doc.getField(FieldNames.SUBJECT_URI).getValue());
    assertNotNull(doc.getField(FieldNames.INDEX_DATE).getValue());
    assertEquals(3, doc.size());
}

---

From source file:org.apache.sentry.tests.e2e.solr.AbstractSolrSentryTestBase.java

License:Apache License

/**
 * Function to validate the content of Solr response with that of input document.
 * @param solrInputDoc - Solr doc inserted into Solr
 * @param solrRespDocs - List of Solr doc obtained as response
 * (NOTE: This function ignores "_version_" field in validating Solr doc content)
 *//*from   w w w  . j  a v  a 2 s  . co m*/
public void validateSolrDocContent(SolrInputDocument solrInputDoc, SolrDocumentList solrRespDocs) {
    for (SolrDocument solrRespDoc : solrRespDocs) {
        String expFieldValue = (String) solrInputDoc.getFieldValue("id");
        String resFieldValue = (String) solrRespDoc.getFieldValue("id");
        if (expFieldValue.equals(resFieldValue)) {
            int expectedRespFieldCount = solrRespDoc.size();
            if (solrRespDoc.containsKey("_version_")) {
                expectedRespFieldCount = expectedRespFieldCount - 1;
            }
            int expectedOrigFieldCount = solrInputDoc.size();
            if (solrInputDoc.containsKey("_version_")) {
                expectedOrigFieldCount = expectedOrigFieldCount - 1;
            }
            assertEquals("Expected " + expectedOrigFieldCount + " fields. But, found " + expectedRespFieldCount
                    + " fields", expectedOrigFieldCount, expectedRespFieldCount);
            for (String field : solrInputDoc.getFieldNames()) {
                if (field.equals("_version_") == true) {
                    continue;
                }

                expFieldValue = (String) solrInputDoc.getFieldValue(field);
                resFieldValue = (String) solrRespDoc.getFieldValue(field);
                assertEquals("Expected value for field: " + field + " is " + expFieldValue + "; But, found "
                        + resFieldValue, expFieldValue, resFieldValue);
            }

            return;
        }
    }

    fail("Solr doc not found in Solr collection");
}

---

From source file:org.craftercms.search.service.impl.SolrDocumentBuilderImplTest.java

License:Open Source License

@Test
public void testBuildForXml() throws Exception {
    Charset encoding = Charset.defaultCharset();
    String xml = IOUtils.toString((new ClassPathResource("/docs/ipad.xml")).getInputStream(), encoding);

    SolrInputDocument doc = builder.build(SITE, IPAD_ID, xml, true);

    assertNotNull(doc);//from  ww w  .  ja v  a 2 s .co m
    assertEquals(15, doc.size());
    assertNull(doc.getFieldValue("code"));
    assertNotNull(doc.getFieldValue(DEFAULT_PUBLISHING_DATE_FIELD_NAME));
    assertNotNull(doc.getFieldValue(DEFAULT_PUBLISHING_DATE_ALT_FIELD_NAME));
    assertEquals(SITE, doc.getFieldValue(DEFAULT_SITE_FIELD_NAME));
    assertEquals(SITE + ":" + IPAD_ID, doc.getFieldValue(DEFAULT_ID_FIELD_NAME));
    assertEquals(SITE + ":" + IPAD_ID, doc.getFieldValue(DEFAULT_ROOT_ID_FIELD_NAME));
    assertEquals(IPAD_ID, doc.getFieldValue(DEFAULT_LOCAL_ID_FIELD_NAME));
    assertEquals("product", doc.getFieldValue(DEFAULT_CONTENT_TYPE_FIELD_NAME));
    assertEquals("iPad Air 64GB", doc.getFieldValue("name_s"));
    assertEquals("iPad Air 64GB", doc.getFieldValue("name_t"));
    assertEquals("Apple MH182LL/A iPad Air 9.7-Inch Retina Display 64GB, Wi-Fi (Gold)",
            doc.getFieldValue("description_html").toString().trim());
    assertEquals("2014-10-01T00:00:00.000Z", doc.getFieldValue("availableDate_dt"));
    assertEquals(Arrays.asList("Apple", "iPad", "Tablet"), doc.getFieldValues("tags.value_smv"));
    assertEquals(Arrays.asList("Silicon case with stand for iPad Air 64GB", "Lighting cable for iPad"),
            trimValues(doc.getFieldValues("accessories.item.description_html")));
    assertEquals(Arrays.asList("Case", "Lighting Cable"), doc.getFieldValues("accessories.item.name_smv"));
    assertEquals(Arrays.asList("Black", "Blue", "Red"),
            doc.getFieldValues("accessories.item.colors.color_smv"));

    // Asset sub-docs
    List<SolrInputDocument> subDocs = doc.getChildDocuments();

    assertNotNull(subDocs);
    assertEquals(2, subDocs.size());

    // Assert first doc
    SolrInputDocument subDoc1 = subDocs.get(0);

    assertEquals(16, subDoc1.size());
    assertNotNull(doc.getFieldValue(DEFAULT_PUBLISHING_DATE_FIELD_NAME));
    assertNotNull(doc.getFieldValue(DEFAULT_PUBLISHING_DATE_ALT_FIELD_NAME));
    assertEquals(SITE, subDoc1.getFieldValue(DEFAULT_SITE_FIELD_NAME));
    assertEquals(SITE + ":" + IPAD_ID_ACCESSORIES0, subDoc1.getFieldValue(DEFAULT_ID_FIELD_NAME));
    assertEquals(SITE + ":" + IPAD_ID, subDoc1.getFieldValue(DEFAULT_ROOT_ID_FIELD_NAME));
    assertEquals(IPAD_ID_ACCESSORIES0, subDoc1.getFieldValue(DEFAULT_LOCAL_ID_FIELD_NAME));
    assertEquals(SITE + ":" + IPAD_ID, subDoc1.getFieldValue(DEFAULT_PARENT_ID_FIELD_NAME));
    assertEquals("product_accessories", subDoc1.getFieldValue(DEFAULT_CONTENT_TYPE_FIELD_NAME));
    assertEquals("Case", subDoc1.getFieldValue("accessories.item.name_s"));
    assertEquals("Silicon case with stand for iPad Air 64GB",
            subDoc1.getFieldValue("accessories.item.description_html").toString().trim());
    assertEquals(Arrays.asList("Black", "Blue", "Red"),
            subDoc1.getFieldValues("accessories.item.colors.color_smv"));
    assertEquals("iPad Air 64GB", subDoc1.getFieldValue("name_s"));
    assertEquals("iPad Air 64GB", subDoc1.getFieldValue("name_t"));
    assertEquals("Apple MH182LL/A iPad Air 9.7-Inch Retina Display 64GB, Wi-Fi (Gold)",
            subDoc1.getFieldValue("description_html").toString().trim());
    assertEquals("2014-10-01T00:00:00.000Z", subDoc1.getFieldValue("availableDate_dt"));
    assertEquals(Arrays.asList("Apple", "iPad", "Tablet"), subDoc1.getFieldValues("tags.value_smv"));

    // Assert second doc
    SolrInputDocument subDoc2 = subDocs.get(1);

    assertEquals(15, subDoc2.size());
    assertNotNull(doc.getFieldValue(DEFAULT_PUBLISHING_DATE_FIELD_NAME));
    assertNotNull(doc.getFieldValue(DEFAULT_PUBLISHING_DATE_ALT_FIELD_NAME));
    assertEquals(SITE, subDoc2.getFieldValue(DEFAULT_SITE_FIELD_NAME));
    assertEquals(SITE + ":" + IPAD_ID_ACCESSORIES1, subDoc2.getFieldValue(DEFAULT_ID_FIELD_NAME));
    assertEquals(SITE + ":" + IPAD_ID, subDoc1.getFieldValue(DEFAULT_ROOT_ID_FIELD_NAME));
    assertEquals(IPAD_ID_ACCESSORIES1, subDoc2.getFieldValue(DEFAULT_LOCAL_ID_FIELD_NAME));
    assertEquals(SITE + ":" + IPAD_ID, subDoc2.getFieldValue(DEFAULT_PARENT_ID_FIELD_NAME));
    assertEquals("product_accessories", subDoc2.getFieldValue(DEFAULT_CONTENT_TYPE_FIELD_NAME));
    assertEquals("Lighting Cable", subDoc2.getFieldValue("accessories.item.name_s"));
    assertEquals("Lighting cable for iPad",
            subDoc2.getFieldValue("accessories.item.description_html").toString().trim());
    assertEquals("iPad Air 64GB", subDoc2.getFieldValue("name_s"));
    assertEquals("iPad Air 64GB", subDoc2.getFieldValue("name_t"));
    assertEquals("Apple MH182LL/A iPad Air 9.7-Inch Retina Display 64GB, Wi-Fi (Gold)",
            subDoc2.getFieldValue("description_html").toString().trim());
    assertEquals("2014-10-01T00:00:00.000Z", subDoc2.getFieldValue("availableDate_dt"));
    assertEquals(Arrays.asList("Apple", "iPad", "Tablet"), subDoc2.getFieldValues("tags.value_smv"));
}

---

From source file:org.craftercms.search.service.impl.SolrDocumentBuilderImplTest.java

License:Open Source License

@Test
public void testBuildForFields() throws Exception {
    MultiValueMap<String, String> fields = new LinkedMultiValueMap<>(3);
    fields.add("name", "Samsung Galaxy Tab 4");
    fields.add("description_html", "<p>Samsung Galaxy Tab 4 (7-Inch, White)</p>");
    fields.add("availableDate_dt", "2014-04-01T11:00:00.000Z");
    fields.add("tags.value_smv", "Samsung");
    fields.add("tags.value_smv", "Galaxy");
    fields.add("tags.value_smv", "Tablet");

    SolrInputDocument doc = builder.build(SITE, TAB_ID, fields);

    assertNotNull(doc);//from   www.  j  a va 2s. c om
    assertEquals(10, doc.size());
    assertNotNull(doc.getFieldValue(DEFAULT_PUBLISHING_DATE_FIELD_NAME));
    assertNotNull(doc.getFieldValue(DEFAULT_PUBLISHING_DATE_ALT_FIELD_NAME));
    assertEquals(SITE, doc.getFieldValue(DEFAULT_SITE_FIELD_NAME));
    assertEquals(SITE + ":" + TAB_ID, doc.getFieldValue(DEFAULT_ID_FIELD_NAME));
    assertEquals(SITE + ":" + TAB_ID, doc.getFieldValue(DEFAULT_ROOT_ID_FIELD_NAME));
    assertEquals(TAB_ID, doc.getFieldValue(DEFAULT_LOCAL_ID_FIELD_NAME));
    assertEquals("Samsung Galaxy Tab 4", doc.getFieldValue("name"));
    assertEquals("Samsung Galaxy Tab 4 (7-Inch, White)",
            doc.getFieldValue("description_html").toString().trim());
    assertEquals("2014-04-01T11:00:00.000Z", doc.getFieldValue("availableDate_dt"));
    assertEquals(Arrays.asList("Samsung", "Galaxy", "Tablet"), doc.getFieldValues("tags.value_smv"));
}

---

From source file:org.phenotips.vocabulary.internal.solr.OncoTreeTest.java

License:Open Source License

@Test
public void loadReturnsCorrectDataWhenReadingFromSourceThatDoesNotDefineIntermediates() throws IOException {
    final InputStream inputStream = getInputStream("src/test/resources/test2.txt");
    doReturn(inputStream).when(this.oncoTree).getInputStream(any(URL.class));
    final Collection<SolrInputDocument> data = this.oncoTree.load(this.url);
    Assert.assertEquals(4, data.size());
    for (final SolrInputDocument datum : data) {
        switch ((String) datum.getFieldValue(ID)) {
        case PREFIX + DCS_LABEL:
            Assert.assertEquals(3, datum.size());
            Assert.assertEquals(SOFT_TISSUE_LABEL, datum.getFieldValue(TISSUE));
            Assert.assertEquals(DCS_NAME_LABEL, datum.getFieldValue(NAME));
            break;
        case PREFIX + FDCS_LABEL:
            Assert.assertEquals(10, datum.size());
            Assert.assertEquals(2, datum.getFieldValues(TISSUE).size());
            Assert.assertTrue(datum.getFieldValues(TISSUE).contains(SOFT_TISSUE_LABEL));
            Assert.assertTrue(datum.getFieldValues(TISSUE).contains(MYELOID_LABEL));
            Assert.assertEquals(2, datum.getFieldValues(TERM_CATEGORY).size());
            Assert.assertTrue(datum.getFieldValues(TERM_CATEGORY).contains(PREFIX + HDCN_LABEL));
            Assert.assertTrue(datum.getFieldValues(TERM_CATEGORY).contains(PREFIX + DCS_LABEL));
            Assert.assertEquals(2, datum.getFieldValues(IS_A).size());
            Assert.assertTrue(datum.getFieldValues(IS_A).contains(PREFIX + HDCN_LABEL));
            Assert.assertTrue(datum.getFieldValues(IS_A).contains(PREFIX + DCS_LABEL));
            Assert.assertEquals(FDCS_NAME_LABEL, datum.getFieldValue(NAME));
            Assert.assertEquals(1, datum.getFieldValues(TERM_GROUP).size());
            Assert.assertTrue(datum.getFieldValues(TERM_GROUP).contains("Soft Tissue Sarcoma"));
            Assert.assertFalse(datum.getFieldValues(COLOUR).isEmpty());
            Assert.assertEquals(1, datum.getFieldValues(SYNONYM).size());
            Assert.assertTrue(datum.getFieldValues(SYNONYM).contains(FDCS_ALT_NAME_LABEL));
            Assert.assertEquals("C9281", datum.getFieldValue(NCI_ID));
            Assert.assertEquals("C1260325", datum.getFieldValue(UMLS_ID));
            break;
        case PREFIX + HDCN_LABEL:
            Assert.assertEquals(3, datum.size());
            Assert.assertEquals(1, datum.getFieldValues(TISSUE).size());
            Assert.assertTrue(datum.getFieldValues(TISSUE).contains(MYELOID_LABEL));
            Assert.assertEquals(HDCN_NAME_LABEL, datum.getFieldValue(NAME));
            break;
        case HEADER_INFO_LABEL:
            Assert.assertEquals(2, datum.size());
            Assert.assertNotNull(datum.getFieldValue(VERSION_FIELD_NAME));
            break;
        default://  ww w.ja  va2 s . c  o  m
            Assert.fail();
        }
    }
}

---

From source file:org.phenotips.vocabulary.internal.solr.OncoTreeTest.java

License:Open Source License

@Test
public void loadReturnsCorrectDataWhenReadingFromSourceWithNonEnglishChars() {
    final Collection<SolrInputDocument> data = this.oncoTree.load(this.url);
    Assert.assertEquals(8, data.size());

    for (final SolrInputDocument datum : data) {
        switch ((String) datum.getFieldValue(ID)) {
        case PREFIX + ACA_LABEL:
            Assert.assertEquals(7, datum.size());
            Assert.assertEquals(ADRENAL_GLAND_LABEL, datum.getFieldValue(TISSUE));
            Assert.assertEquals(ACA_NAME_LABEL, datum.getFieldValue(NAME));
            Assert.assertEquals(ADRENOCORTICAL_CARCINOMA_LABEL, datum.getFieldValue(TERM_GROUP));
            Assert.assertEquals(PURPLE_LABEL, datum.getFieldValue(COLOUR));
            Assert.assertEquals("C9003", datum.getFieldValue(NCI_ID));
            Assert.assertEquals("C0206667", datum.getFieldValue(UMLS_ID));
            break;
        case PREFIX + EGC_LABEL:
            Assert.assertEquals(7, datum.size());
            Assert.assertEquals(ESOPHAGUS_STOMACH_LABEL, datum.getFieldValue(TISSUE));
            Assert.assertEquals(EGC_NAME_LABEL, datum.getFieldValue(NAME));
            Assert.assertEquals(ESOPHAGOGASTRIC_CANCER_LABEL, datum.getFieldValue(TERM_GROUP));
            Assert.assertEquals(BLUE_LABEL, datum.getFieldValue(COLOUR));
            Assert.assertEquals("C9296", datum.getFieldValue(NCI_ID));
            Assert.assertEquals("C1332166", datum.getFieldValue(UMLS_ID));
            break;
        case PREFIX + STAD_LABEL:
            final Set<String> stadTermCategory = new HashSet<>();
            stadTermCategory.add(PREFIX + EGC_LABEL);
            Assert.assertEquals(9, datum.size());
            Assert.assertEquals(ESOPHAGUS_STOMACH_LABEL, datum.getFieldValue(TISSUE));
            Assert.assertEquals(STAD_NAME_LABEL, datum.getFieldValue(NAME));
            Assert.assertEquals(ESOPHAGOGASTRIC_CANCER_LABEL, datum.getFieldValue(TERM_GROUP));
            Assert.assertEquals(BLUE_LABEL, datum.getFieldValue(COLOUR));
            Assert.assertEquals("C4004", datum.getFieldValue(NCI_ID));
            Assert.assertEquals("C0278701", datum.getFieldValue(UMLS_ID));
            Assert.assertEquals(PREFIX + EGC_LABEL, datum.getFieldValue(IS_A));
            Assert.assertEquals(stadTermCategory, datum.getFieldValues(TERM_CATEGORY));
            break;
        case PREFIX + DSTAD_LABEL:
            final Set<String> dstadTermCategory = new HashSet<>();
            dstadTermCategory.add(PREFIX + EGC_LABEL);
            dstadTermCategory.add(PREFIX + STAD_LABEL);
            Assert.assertEquals(9, datum.size());
            Assert.assertEquals(ESOPHAGUS_STOMACH_LABEL, datum.getFieldValue(TISSUE));
            Assert.assertEquals(DSTAD_NAME_LABEL, datum.getFieldValue(NAME));
            Assert.assertEquals(ESOPHAGOGASTRIC_CANCER_LABEL, datum.getFieldValue(TERM_GROUP));
            Assert.assertEquals(BLUE_LABEL, datum.getFieldValue(COLOUR));
            Assert.assertEquals("C9159", datum.getFieldValue(NCI_ID));
            Assert.assertEquals("C0279635", datum.getFieldValue(UMLS_ID));
            Assert.assertEquals(PREFIX + STAD_LABEL, datum.getFieldValue(IS_A));
            Assert.assertEquals(dstadTermCategory, datum.getFieldValues(TERM_CATEGORY));
            break;
        case PREFIX + TET_LABEL:
            Assert.assertEquals(7, datum.size());
            Assert.assertEquals(THYMUS_LABEL, datum.getFieldValue(TISSUE));
            Assert.assertEquals(TET_NAME_LABEL, datum.getFieldValue(NAME));
            Assert.assertEquals(THYMIC_TUMOR_LABEL, datum.getFieldValue(TERM_GROUP));
            Assert.assertEquals(PURPLE_LABEL, datum.getFieldValue(COLOUR));
            Assert.assertEquals("C6450", datum.getFieldValue(NCI_ID));
            Assert.assertEquals("C1266101", datum.getFieldValue(UMLS_ID));
            break;
        case PREFIX + THYC_LABEL:
            final Set<String> tetTermCategory = new HashSet<>();
            tetTermCategory.add(PREFIX + TET_LABEL);
            Assert.assertEquals(9, datum.size());
            Assert.assertEquals(THYMUS_LABEL, datum.getFieldValue(TISSUE));
            Assert.assertEquals(THYC_NAME_LABEL, datum.getFieldValue(NAME));
            Assert.assertEquals(THYMIC_TUMOR_LABEL, datum.getFieldValue(TERM_GROUP));
            Assert.assertEquals(PURPLE_LABEL, datum.getFieldValue(COLOUR));
            Assert.assertEquals("C7569", datum.getFieldValue(NCI_ID));
            Assert.assertEquals("C0205969", datum.getFieldValue(UMLS_ID));
            Assert.assertEquals(PREFIX + TET_LABEL, datum.getFieldValue(IS_A));
            Assert.assertEquals(tetTermCategory, datum.getFieldValues(TERM_CATEGORY));
            break;
        case PREFIX + TNET_LABEL:
            Assert.assertEquals(6, datum.size());
            Assert.assertEquals(THYMUS_LABEL, datum.getFieldValue(TISSUE));
            Assert.assertEquals(TNET_NAME_LABEL, datum.getFieldValue(NAME));
            Assert.assertEquals(THYMIC_TUMOR_LABEL, datum.getFieldValue(TERM_GROUP));
            Assert.assertEquals(PURPLE_LABEL, datum.getFieldValue(COLOUR));
            Assert.assertEquals("CL511204", datum.getFieldValue(UMLS_ID));
            break;
        case HEADER_INFO_LABEL:
            Assert.assertEquals(2, datum.size());
            Assert.assertNotNull(datum.getFieldValue(VERSION_FIELD_NAME));
            break;
        default:/*  w  ww  .jav a2s  . c  o m*/
            Assert.fail();
        }
    }
}

---

From source file:org.roda.core.index.SolrUtilsTest.java

@Test
public void testGetDescriptiveMetadataFields() throws RODAException {
    final DefaultStoragePath strangeMetadataPath = DefaultStoragePath.parse(
            CorporaConstants.SOURCE_DESC_METADATA_CONTAINER, CorporaConstants.STRANGE_DESC_METADATA_FILE);
    Binary strangeMetadata = corporaService.getBinary(strangeMetadataPath);

    SolrInputDocument descriptiveMetataFields = SolrUtils.getDescriptiveMetadataFields(strangeMetadata, null,
            null);/*from   w  ww  . java  2  s  .com*/

    assertNotNull(descriptiveMetataFields);
    assertEquals(5, descriptiveMetataFields.size());
    SolrInputField field1 = descriptiveMetataFields
            .getField(RodaConstants.INDEX_OTHER_DESCRIPTIVE_DATA_PREFIX + ".note.to_txt");
    assertNotNull(field1);
    assertEquals(RodaConstants.INDEX_OTHER_DESCRIPTIVE_DATA_PREFIX + ".note.to_txt", field1.getName());
    assertEquals("Tove", field1.getValue());
}

---

From source file:uk.bl.wa.solr.QueueingHttpSolrServer.java

License:Open Source License

private boolean checkMemory(SolrInputDocument doc) {
    return ((doc.size() * MEM_FACTOR) > runtime.freeMemory());
}

---