use of com.yahoo.feedapi.MessagePropertyProcessor in project vespa by vespa-engine.
the class GetSearcherTestCase method testQueryPassThroughAndGet.
/* Test that a query will contain both document hits and hits from a searcher
* further down the chain, iff the searcher returns a DocumentHit.
*/
@Test
public void testQueryPassThroughAndGet() throws Exception {
Document doc1 = new Document(docType, new DocumentId("userdoc:kittens:1234:foo"));
doc1.setFieldValue("name", new StringFieldValue("megacat"));
doc1.setFieldValue("description", new StringFieldValue("supercat"));
doc1.setFieldValue("fluffiness", new IntegerFieldValue(10000));
GetDocumentReply[] replies = new GetDocumentReply[] { new GetDocumentReply(doc1) };
DocumentSessionFactory factory = new DocumentSessionFactory(docType, null, false, replies);
GetSearcher searcher = new GetSearcher(new FeedContext(new MessagePropertyProcessor(defFeedCfg, defLoadTypeCfg), factory, docMan, new ClusterList(), new NullFeedMetric()));
DocumentHit backendHit = new DocumentHit(new Document(docType, new DocumentId("userdoc:kittens:5678:bar")), 5);
Chain<Searcher> searchChain = new Chain<>(searcher, new MockBackend(backendHit));
Result result = new Execution(searchChain, Execution.Context.createContextStub()).search(newQuery("?query=flarn&id=userdoc:kittens:1234:foo"));
assertEquals(1, factory.messages.size());
assertEquals(2, result.hits().size());
assertNotNull(result.hits().get("userdoc:kittens:5678:bar"));
assertNotNull(result.hits().get("userdoc:kittens:1234:foo"));
}
use of com.yahoo.feedapi.MessagePropertyProcessor in project vespa by vespa-engine.
the class GetSearcherTestCase method testConfigChanges.
@Test
public void testConfigChanges() throws Exception {
String config = "raw:timeout 37\nroute \"riksveg18\"\nretryenabled true";
DocumentSessionFactory factory = new DocumentSessionFactory(docType);
GetSearcher searcher = new GetSearcher(new FeedContext(new MessagePropertyProcessor(new FeederConfig(new FeederConfig.Builder().timeout(58).route("riksveg18").retryenabled(true)), defLoadTypeCfg), factory, docMan, new ClusterList(), new NullFeedMetric()));
Chain<Searcher> searchChain = new Chain<>(searcher);
new Execution(searchChain, Execution.Context.createContextStub()).search(newQuery("?id=doc:batman:dahnahnahnah"));
assertEquals(1, factory.messages.size());
assertEquals(1, factory.getSessionsCreated());
{
Message m = factory.messages.get(0);
assertEquals(DocumentProtocol.MESSAGE_GETDOCUMENT, m.getType());
GetDocumentMessage gdm = (GetDocumentMessage) m;
DocumentId d = gdm.getDocumentId();
assertEquals("doc:batman:dahnahnahnah", d.toString());
assertEquals("[all]", gdm.getFieldSet());
assertEquals(Route.parse("riksveg18"), gdm.getRoute());
assertTrue(gdm.getRetryEnabled());
assertTrue(58000 >= gdm.getTimeRemaining());
}
factory.messages.clear();
FeederConfig newConfig = new FeederConfig(new FeederConfig.Builder().timeout(123).route("e6").retryenabled(false));
searcher.getMessagePropertyProcessor().configure(newConfig, defLoadTypeCfg);
new Execution(searchChain, Execution.Context.createContextStub()).search(newQuery("?id=doc:spiderman:does_whatever_a_spider_can"));
// riksveg18 is created again, and e6 is created as well.
assertEquals(3, factory.getSessionsCreated());
assertEquals(1, factory.messages.size());
{
Message m = factory.messages.get(0);
assertEquals(DocumentProtocol.MESSAGE_GETDOCUMENT, m.getType());
GetDocumentMessage gdm = (GetDocumentMessage) m;
DocumentId d = gdm.getDocumentId();
assertEquals("doc:spiderman:does_whatever_a_spider_can", d.toString());
assertEquals("[all]", gdm.getFieldSet());
assertEquals(Route.parse("e6"), gdm.getRoute());
assertFalse(gdm.getRetryEnabled());
assertTrue(123000 >= gdm.getTimeRemaining());
}
}
use of com.yahoo.feedapi.MessagePropertyProcessor in project vespa by vespa-engine.
the class GetSearcherTestCase method testResultWithSingleError.
@Test
public void testResultWithSingleError() throws Exception {
com.yahoo.messagebus.Error err = new com.yahoo.messagebus.Error(32, "Alas, it went poorly");
DocumentSessionFactory factory = new DocumentSessionFactory(docType, err, true);
GetSearcher searcher = new GetSearcher(new FeedContext(new MessagePropertyProcessor(defFeedCfg, defLoadTypeCfg), factory, docMan, new ClusterList(), new NullFeedMetric()));
Chain<Searcher> searchChain = new Chain<>(searcher);
Result result = new Execution(searchChain, Execution.Context.createContextStub()).search(newQuery("?id[0]=userdoc:kittens:1:2&id[1]=userdoc:kittens:3:4"));
assertNotNull(result.hits().getErrorHit());
assertRendered("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" + "<result>\n" + "<errors>\n" + "<error type=\"messagebus\" code=\"32\" message=\"Alas, it went poorly\"/>\n" + "</errors>\n" + "</result>\n", result);
}
use of com.yahoo.feedapi.MessagePropertyProcessor in project vespa by vespa-engine.
the class GetSearcherTestCase method testGetMultipleDocumentsQueryAndGZippedPOST.
@Test
public void testGetMultipleDocumentsQueryAndGZippedPOST() throws Exception {
DocumentSessionFactory factory = new DocumentSessionFactory(docType);
GetSearcher searcher = new GetSearcher(new FeedContext(new MessagePropertyProcessor(defFeedCfg, defLoadTypeCfg), factory, docMan, new ClusterList(), new NullFeedMetric()));
Chain<Searcher> searchChain = new Chain<>(searcher);
String data = "userdoc:kittens:5:6\nuserdoc:kittens:7:8\nuserdoc:kittens:9:10";
// Create with automatic GZIP encoding
MockHttpRequest request = new MockHttpRequest(data.getBytes("utf-8"), "/get/?id[0]=userdoc:kittens:1:2&id[1]=userdoc:kittens:3:4", true);
Query query = new Query(request.toRequest());
Result result = new Execution(searchChain, Execution.Context.createContextStub()).search(query);
assertEquals(5, factory.messages.size());
assertEquals(5, result.hits().size());
assertNull(result.hits().getErrorHit());
assertHits(result.hits(), "userdoc:kittens:1:2", "userdoc:kittens:3:4", "userdoc:kittens:5:6", "userdoc:kittens:7:8", "userdoc:kittens:9:10");
}
use of com.yahoo.feedapi.MessagePropertyProcessor in project vespa by vespa-engine.
the class GetSearcherTestCase method testQueryPassThrough.
/* Test that a query without any ids is passed through to the next chain */
@Test
public void testQueryPassThrough() throws Exception {
DocumentSessionFactory factory = new DocumentSessionFactory(docType);
GetSearcher searcher = new GetSearcher(new FeedContext(new MessagePropertyProcessor(defFeedCfg, defLoadTypeCfg), factory, docMan, new ClusterList(), new NullFeedMetric()));
HitGroup hits = new HitGroup("mock");
hits.add(new Hit("blernsball"));
Chain<Searcher> searchChain = new Chain<>(searcher, new MockBackend(hits));
Result result = new Execution(searchChain, Execution.Context.createContextStub()).search(newQuery("?flarn=blern"));
assertEquals(0, factory.messages.size());
assertEquals(1, result.hits().size());
assertNotNull(result.hits().get("blernsball"));
}
Aggregations