~slub.team/goobi-indexserver/3.x

« back to all changes in this revision

Viewing changes to solr/solrj/src/test/org/apache/solr/client/solrj/MergeIndexesExampleTestBase.java

  • Committer: Sebastian Meyer
  • Date: 2012-08-03 09:12:40 UTC
  • Revision ID: sebastian.meyer@slub-dresden.de-20120803091240-x6861b0vabq1xror
Remove Lucene and Solr source code and add patches instead
Fix Bug #985487: Auto-suggestion for the search interface

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
/**
2
 
 * Licensed to the Apache Software Foundation (ASF) under one or more
3
 
 * contributor license agreements.  See the NOTICE file distributed with
4
 
 * this work for additional information regarding copyright ownership.
5
 
 * The ASF licenses this file to You under the Apache License, Version 2.0
6
 
 * (the "License"); you may not use this file except in compliance with
7
 
 * the License.  You may obtain a copy of the License at
8
 
 *
9
 
 *     http://www.apache.org/licenses/LICENSE-2.0
10
 
 *
11
 
 * Unless required by applicable law or agreed to in writing, software
12
 
 * distributed under the License is distributed on an "AS IS" BASIS,
13
 
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
 
 * See the License for the specific language governing permissions and
15
 
 * limitations under the License.
16
 
 */
17
 
 
18
 
package org.apache.solr.client.solrj;
19
 
 
20
 
import org.apache.solr.client.solrj.request.AbstractUpdateRequest;
21
 
import org.apache.solr.client.solrj.request.CoreAdminRequest;
22
 
import org.apache.solr.client.solrj.request.QueryRequest;
23
 
import org.apache.solr.client.solrj.request.UpdateRequest;
24
 
import org.apache.solr.client.solrj.request.UpdateRequest.ACTION;
25
 
import org.apache.solr.common.SolrInputDocument;
26
 
import org.apache.solr.core.CoreContainer;
27
 
import org.apache.solr.core.SolrCore;
28
 
import org.apache.solr.util.ExternalPaths;
29
 
 
30
 
import java.io.IOException;
31
 
 
32
 
/**
33
 
 * Abstract base class for testing merge indexes command
34
 
 *
35
 
 * @since solr 1.4
36
 
 * @version $Id: MergeIndexesExampleTestBase.java 1146191 2011-07-13 18:23:28Z sarowe $
37
 
 */
38
 
public abstract class MergeIndexesExampleTestBase extends SolrExampleTestBase {
39
 
  // protected static final CoreContainer cores = new CoreContainer();
40
 
  protected static CoreContainer cores;
41
 
 
42
 
  @Override
43
 
  public String getSolrHome() {
44
 
    return ExternalPaths.EXAMPLE_MULTICORE_HOME;
45
 
  }
46
 
 
47
 
  @Override
48
 
  public String getSchemaFile() {
49
 
    return getSolrHome() + "/core0/conf/schema.xml";
50
 
  }
51
 
 
52
 
  @Override
53
 
  public String getSolrConfigFile() {
54
 
    return getSolrHome() + "/core0/conf/solrconfig.xml";
55
 
  }
56
 
 
57
 
  @Override
58
 
  public void setUp() throws Exception {
59
 
    super.setUp();
60
 
    cores = h.getCoreContainer();
61
 
    SolrCore.log.info("CORES=" + cores + " : " + cores.getCoreNames());
62
 
    cores.setPersistent(false);
63
 
  }
64
 
 
65
 
  @Override
66
 
  protected final SolrServer getSolrServer() {
67
 
    throw new UnsupportedOperationException();
68
 
  }
69
 
 
70
 
  @Override
71
 
  protected final SolrServer createNewSolrServer() {
72
 
    throw new UnsupportedOperationException();
73
 
  }
74
 
 
75
 
  protected abstract SolrServer getSolrCore0();
76
 
 
77
 
  protected abstract SolrServer getSolrCore1();
78
 
 
79
 
  protected abstract SolrServer getSolrAdmin();
80
 
 
81
 
  protected abstract SolrServer getSolrCore(String name);
82
 
 
83
 
  protected abstract String getIndexDirCore1();
84
 
 
85
 
  private UpdateRequest setupCores() throws SolrServerException, IOException {
86
 
    UpdateRequest up = new UpdateRequest();
87
 
    up.setAction(AbstractUpdateRequest.ACTION.COMMIT, true, true);
88
 
    up.deleteByQuery("*:*");
89
 
    up.process(getSolrCore0());
90
 
    up.process(getSolrCore1());
91
 
    up.clear();
92
 
 
93
 
    // Add something to each core
94
 
    SolrInputDocument doc = new SolrInputDocument();
95
 
    doc.setField("id", "AAA");
96
 
    doc.setField("name", "core0");
97
 
 
98
 
    // Add to core0
99
 
    up.add(doc);
100
 
    up.process(getSolrCore0());
101
 
 
102
 
    // Add to core1
103
 
    doc.setField("id", "BBB");
104
 
    doc.setField("name", "core1");
105
 
    up.add(doc);
106
 
    up.process(getSolrCore1());
107
 
 
108
 
    // Now Make sure AAA is in 0 and BBB in 1
109
 
    SolrQuery q = new SolrQuery();
110
 
    QueryRequest r = new QueryRequest(q);
111
 
    q.setQuery("id:AAA");
112
 
    assertEquals(1, r.process(getSolrCore0()).getResults().size());
113
 
    assertEquals(0, r.process(getSolrCore1()).getResults().size());
114
 
 
115
 
    assertEquals(1,
116
 
        getSolrCore0().query(new SolrQuery("id:AAA")).getResults().size());
117
 
    assertEquals(0,
118
 
        getSolrCore0().query(new SolrQuery("id:BBB")).getResults().size());
119
 
 
120
 
    assertEquals(0,
121
 
        getSolrCore1().query(new SolrQuery("id:AAA")).getResults().size());
122
 
    assertEquals(1,
123
 
        getSolrCore1().query(new SolrQuery("id:BBB")).getResults().size());
124
 
 
125
 
    return up;
126
 
  }
127
 
 
128
 
  public void testMergeIndexesByDirName() throws Exception {
129
 
    UpdateRequest up = setupCores();
130
 
 
131
 
    // Now get the index directory of core1 and merge with core0
132
 
    CoreAdminRequest.mergeIndexes("core0", new String[] {getIndexDirCore1()}, new String[0], getSolrAdmin());
133
 
 
134
 
    // Now commit the merged index
135
 
    up.clear(); // just do commit
136
 
    up.process(getSolrCore0());
137
 
 
138
 
    assertEquals(1,
139
 
        getSolrCore0().query(new SolrQuery("id:AAA")).getResults().size());
140
 
    assertEquals(1,
141
 
        getSolrCore0().query(new SolrQuery("id:BBB")).getResults().size());
142
 
  }
143
 
 
144
 
  public void testMergeIndexesByCoreName() throws Exception {
145
 
    UpdateRequest up = setupCores();
146
 
    CoreAdminRequest.mergeIndexes("core0", new String[0], new String[] {"core1"}, getSolrAdmin());
147
 
 
148
 
    // Now commit the merged index
149
 
    up.clear(); // just do commit
150
 
    up.process(getSolrCore0());
151
 
 
152
 
    assertEquals(1,
153
 
        getSolrCore0().query(new SolrQuery("id:AAA")).getResults().size());
154
 
    assertEquals(1,
155
 
        getSolrCore0().query(new SolrQuery("id:BBB")).getResults().size());
156
 
  }
157
 
}