Github user dsmiley commented on a diff in the pull request: https://github.com/apache/lucene-solr/pull/455#discussion_r224131571 --- Diff: solr/core/src/test/org/apache/solr/update/processor/AtomicUpdateBlockTest.java --- @@ -0,0 +1,370 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.solr.update.processor; + +import java.util.Arrays; +import java.util.Collection; +import java.util.Collections; +import java.util.List; + +import org.apache.lucene.util.BytesRef; +import org.apache.solr.SolrTestCaseJ4; +import org.apache.solr.common.SolrInputDocument; +import org.apache.solr.common.SolrInputField; +import org.apache.solr.core.SolrCore; +import org.apache.solr.handler.component.RealTimeGetComponent; +import org.junit.AfterClass; +import org.junit.Before; +import org.junit.BeforeClass; +import org.junit.Test; + +public class AtomicUpdateBlockTest extends SolrTestCaseJ4 { + + private final static String VERSION = "_version_"; + private static String PREVIOUS_ENABLE_UPDATE_LOG_VALUE; + + @BeforeClass + public static void beforeTests() throws Exception { + PREVIOUS_ENABLE_UPDATE_LOG_VALUE = System.getProperty("enable.update.log"); + System.setProperty("enable.update.log", "true"); + initCore("solrconfig-block-atomic-update.xml", "schema-nest.xml"); // use "nest" schema + } + + @AfterClass + public static void afterTests() throws Exception { + // restore enable.update.log + System.setProperty("enable.update.log", PREVIOUS_ENABLE_UPDATE_LOG_VALUE); + } + + @Before + public void before() { + clearIndex(); + assertU(commit()); + } + + @Test + public void testMergeChildDoc() throws Exception { + SolrInputDocument doc = new SolrInputDocument(); + doc.setField("id", "1"); + doc.setField("cat_ss", new String[]{"aaa", "ccc"}); + doc.setField("child", Collections.singletonList(sdoc("id", "2", "cat_ss", "child"))); + addDoc(adoc(doc), "nested-rtg"); + + BytesRef rootDocId = new BytesRef("1"); + SolrCore core = h.getCore(); + SolrInputDocument block = RealTimeGetComponent.getInputDocument(core, rootDocId, true); + // assert block doc has child docs + assertTrue(block.containsKey("child")); + + assertJQ(req("q","id:1") + ,"/response/numFound==0" + ); + + // commit the changes + assertU(commit()); + + SolrInputDocument newChildDoc = sdoc("id", "3", "cat_ss", "child"); + SolrInputDocument addedDoc = sdoc("id", "1", + "cat_ss", Collections.singletonMap("add", "bbb"), + "child", Collections.singletonMap("add", sdocs(newChildDoc))); + block = RealTimeGetComponent.getInputDocument(core, rootDocId, true); + block.removeField(VERSION); + SolrInputDocument preMergeDoc = new SolrInputDocument(block); + AtomicUpdateDocumentMerger docMerger = new AtomicUpdateDocumentMerger(req()); + docMerger.merge(addedDoc, block); + assertEquals("merged document should have the same id", preMergeDoc.getFieldValue("id"), block.getFieldValue("id")); + assertDocContainsSubset(preMergeDoc, block); + assertDocContainsSubset(addedDoc, block); + assertDocContainsSubset(newChildDoc, (SolrInputDocument) ((List) block.getFieldValues("child")).get(1)); + assertEquals(doc.getFieldValue("id"), block.getFieldValue("id")); + } + + @Test + public void testBlockAtomicAdd() throws Exception { + + SolrInputDocument doc = sdoc("id", "1", + "cat_ss", new String[] {"aaa", "ccc"}, + "child1", sdoc("id", "2", "cat_ss", "child") + ); + json(doc); --- End diff -- accidentally added this json line?
--- --------------------------------------------------------------------- To unsubscribe, e-mail: dev-unsubscr...@lucene.apache.org For additional commands, e-mail: dev-h...@lucene.apache.org