def testSolrHandlers(self):
document1 = {u"name": u"Joe", u"surname": u"Satriani", u"booboo": 12}
document2 = {u"name": u"Joanna", u"surname": u"S šuuumnikiiiič!", u"booboo": 12}
solr = Solr('http://example/solr/core1/')
solr._send_solr_command = self._command_handler
solr.add([document1])
solr.commit()
solr = Solr('http://example/solr/core2/')
solr._send_solr_command = self._command_handler
self.assertEquals(len(solr._add_batch), 0)
solr.add([document2])
solr.commit()
def testUpdateDispatch(self):
url = "http://this.is.a.failure.fail"
document1 = {"name": "Joe", "surname": "Satriani", "booboo": 12}
document2 = {
"name": "Joanna",
"surname": "S šuuumnikiiiič!",
"booboo": 12
}
solr = Solr({"en": url}, "en")
solr._send_solr_command = self._command_handler
solr.add([document1])
solr.commit()
self.assertEqual(self.req_urls[0], url)
self.assertEqual(self.req_urls[1], url)
self.assertEqual({"add": {
"doc": document1
}}, json.loads(self.req_commands[0]))
self.assertEqual({"commit": {}}, json.loads(self.req_commands[1]))
self._clear_handler()
solr.add([document1, document2])
solr.commit()
self.assertEqual(self.req_urls[0], url)
self.assertEqual(self.req_urls[1], url)
self.assertEqual(
self.req_commands[0],
"{\"add\":{\"doc\": %s},\"add\":{\"doc\": %s}}" %
(json.dumps(document1), json.dumps(document2)))
self.assertEqual({"commit": {}}, json.loads(self.req_commands[1]))
def testSolrInterface(self):
# Check configuration with string only
url = "http://this.is.a.mock.url"
solr = Solr(url)
solr._send_solr_command = self._command_handler
solr.commit()
self.assertEqual(self.req_urls[0], url)
def testUpdateDispatch(self):
url = "http://this.is.a.failure.fail"
document1 = {u"name": u"Joe", u"surname": u"Satriani", u"booboo": 12}
document2 = {u"name": u"Joanna", u"surname": u"S šuuumnikiiiič!", u"booboo": 12}
solr = Solr({"en": url}, "en")
solr._send_solr_command = self._command_handler
solr.add([document1])
solr.commit()
self.assertEquals(self.req_urls[0], url)
self.assertEquals(self.req_urls[1], url)
self.assertEqual({"add": {"doc": document1}}, json.loads(self.req_commands[0]))
self.assertEqual({"commit": {}}, json.loads(self.req_commands[1]))
self._clear_handler()
solr.add([document1, document2])
solr.commit()
self.assertEquals(self.req_urls[0], url)
self.assertEquals(self.req_urls[1], url)
self.assertEquals(self.req_commands[0], u"{\"add\":{\"doc\": %s},\"add\":{\"doc\": %s}}" % (json.dumps(document1),
json.dumps(document2)))
self.assertEqual({"commit": {}}, json.loads(self.req_commands[1]))
def testSolrInterface(self):
# Check configuration with string only
url = "http://this.is.a.mock.url"
solr = Solr(url)
solr._send_solr_command = self._command_handler
solr.commit()
self.assertEquals(self.req_urls[0], url)
def testAddFlushBatch(self):
url = "http://this.is.a.mock.url"
document1 = {u"name": u"Joe", u"surname": u"Satriani", u"booboo": 12}
document2 = {
u"name": u"Joanna",
u"surname": u"S šuuumnikiiiič!",
u"booboo": 10
}
document3 = {u"name": u"Lester", u"surname": u"Burnham", u"booboo": 7}
import pysolarized.solr
oldaddbatch = pysolarized.solr.SOLR_ADD_BATCH
pysolarized.solr.SOLR_ADD_BATCH = 1
solr = Solr({"en": url}, "en")
solr._send_solr_command = self._command_handler
solr.add(document1)
solr.add(document2)
solr.add(document3)
solr.commit()
self.assertEqual(
'{"add":{"doc": {"surname": "Satriani", "name": "Joe", "booboo": 12}},"add":{"doc": {"surname": "S \\u0161uuumnikiiii\\u010d!", "name": "Joanna", "booboo": 10}}}',
self.req_commands[0])
self.assertEqual({'add': {
'doc': document3
}}, json.loads(self.req_commands[1]))
self.assertEqual({"commit": {}}, json.loads(self.req_commands[2]))
pysolarized.solr.SOLR_ADD_BATCH = oldaddbatch
def testSolrHandlers(self):
document1 = {"name": "Joe", "surname": "Satriani", "booboo": 12}
document2 = {
"name": "Joanna",
"surname": "S šuuumnikiiiič!",
"booboo": 12
}
solr = Solr('http://example/solr/core1/')
solr._send_solr_command = self._command_handler
solr.add([document1])
solr.commit()
solr = Solr('http://example/solr/core2/')
solr._send_solr_command = self._command_handler
self.assertEqual(len(solr._add_batch), 0)
solr.add([document2])
solr.commit()
def testUpdateDocBoost(self):
url = "http://this.is.a.mock.url"
document1 = {u"name": u"Joe", u"surname": u"Satriani", u"booboo": 12}
solr = Solr({"en": url}, "en")
solr._send_solr_command = self._command_handler
solr.add(document1, boost=10.0)
solr.commit()
self.assertEquals(self.req_urls[0], url)
self.assertEquals(self.req_urls[1], url)
self.assertEqual({"add": {"doc": document1, "boost": 10.0}}, json.loads(self.req_commands[0]))
self.assertEqual({"commit": {}}, json.loads(self.req_commands[1]))
def testUpdateDocBoost(self):
url = "http://this.is.a.mock.url"
document1 = {"name": "Joe", "surname": "Satriani", "booboo": 12}
solr = Solr({"en": url}, "en")
solr._send_solr_command = self._command_handler
solr.add(document1, boost=10.0)
solr.commit()
self.assertEqual(self.req_urls[0], url)
self.assertEqual(self.req_urls[1], url)
self.assertEqual({"add": {
"doc": document1,
"boost": 10.0
}}, json.loads(self.req_commands[0]))
self.assertEqual({"commit": {}}, json.loads(self.req_commands[1]))
def testAddFlushBatch(self):
url = "http://this.is.a.mock.url"
document1 = {u"name": u"Joe", u"surname": u"Satriani", u"booboo": 12}
document2 = {u"name": u"Joanna", u"surname": u"S šuuumnikiiiič!", u"booboo": 10}
document3 = {u"name": u"Lester", u"surname": u"Burnham", u"booboo": 7}
import pysolarized.solr
oldaddbatch = pysolarized.solr.SOLR_ADD_BATCH
pysolarized.solr.SOLR_ADD_BATCH = 1
solr = Solr({"en": url}, "en")
solr._send_solr_command = self._command_handler
solr.add(document1)
solr.add(document2)
solr.add(document3)
solr.commit()
self.assertEqual('{"add":{"doc": {"surname": "Satriani", "name": "Joe", "booboo": 12}},"add":{"doc": {"surname": "S \\u0161uuumnikiiii\\u010d!", "name": "Joanna", "booboo": 10}}}', self.req_commands[0])
self.assertEqual({'add': {'doc': document3}}, json.loads(self.req_commands[1]))
self.assertEqual({"commit": {}}, json.loads(self.req_commands[2]))
pysolarized.solr.SOLR_ADD_BATCH = oldaddbatch
