Esempio n. 1
0
 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)
Esempio n. 2
0
    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]))
Esempio n. 3
0
 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)
Esempio n. 4
0
    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
Esempio n. 5
0
    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()
Esempio n. 6
0
    def testQueryDispatch(self):
        url = "http://this.is.a.failure.fail"
        solr = Solr({"en": url}, "en")
        solr._send_solr_query = self._query_handler

        query = "what is a treeš"
        filters = {"meaning": "deep"}
        sort = ["deepness", "wideness"]
        columns = ["title", "content"]
        start = 31
        rows = 84

        results = solr.query(query,
                             filters=filters,
                             columns=columns,
                             sort=sort,
                             start=start,
                             rows=rows)

        self.assertEqual(self.query_url, "%s/select" % (url, ))
        expected = {
            'q': query,
            'json.nl': 'map',
            'fl': ",".join(columns),
            'start': str(start),
            'rows': str(rows),
            'fq':
            '%s:%s' % (list(filters.keys())[0], list(filters.values())[0]),
            'wt': 'json',
            'sort': ",".join(sort)
        }

        self.assertEqual(self.query_params, expected)
        self.assertTrue(results is not None)

        # Check results
        self.assertEqual(results.results_count, 1)
        self.assertEqual(results.query_time, 45)
        self.assertEqual(results.documents[0], {
            "title": "This is woot",
            "content": "This isn't woot."
        })
        self.assertEqual(results.facets, {"source": [("newspaper", 342)]})
        self.assertEqual(results.highlights,
                         {'ididid': {
                             "content": ["... blah blah ..."]
                         }})
Esempio n. 7
0
    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
Esempio n. 8
0
    def testQueryDispatch(self):
        url = "http://this.is.a.failure.fail"
        solr = Solr({"en": url}, "en")
        solr._send_solr_query = self._query_handler

        query = u"what is a treeš"
        filters = {"meaning": "deep"}
        sort = ["deepness", "wideness"]
        columns = ["title", "content"]
        start = 31
        rows = 84

        results = solr.query(query,
                             filters=filters,
                             columns=columns,
                             sort=sort,
                             start=start,
                             rows=rows)

        self.assertEquals(self.query_url, "%s/select" % (url,))
        expected = { 'q': query,
                     'json.nl': 'map',
                     'fl': ",".join(columns),
                     'start': str(start),
                     'rows': str(rows),
                     'fq': '%s:%s' % (filters.keys()[0], filters.values()[0]),
                     'wt': 'json',
                     'sort': ",".join(sort)}

        self.assertEqual(self.query_params, expected)
        self.assertTrue(results is not None)

        # Check results
        self.assertEquals(results.results_count, 1)
        self.assertEquals(results.query_time, 45)
        self.assertEqual(results.documents[0], {"title": "This is woot", "content": "This isn't woot."})
        self.assertEqual(results.facets, {"source": [("newspaper", 342)]})
        self.assertEqual(results.highlights, {'ididid': {"content": ["... blah blah ..."]}})
Esempio n. 9
0
    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]))
Esempio n. 10
0
    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]))
Esempio n. 11
0
    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()
Esempio n. 12
0
    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]))