| Index: tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/cloudsearch/test_search.py
|
| diff --git a/tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/cloudsearch/test_search.py b/tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/cloudsearch/test_search.py
|
| deleted file mode 100755
|
| index e8097640a0e8bc6e9667b547915758c96e89b609..0000000000000000000000000000000000000000
|
| --- a/tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/cloudsearch/test_search.py
|
| +++ /dev/null
|
| @@ -1,428 +0,0 @@
|
| -#!/usr/bin env python
|
| -
|
| -from tests.compat import mock, unittest
|
| -from httpretty import HTTPretty
|
| -
|
| -import json
|
| -import requests
|
| -
|
| -from boto.cloudsearch.search import SearchConnection, SearchServiceException
|
| -from boto.compat import six, map
|
| -
|
| -HOSTNAME = "search-demo-userdomain.us-east-1.cloudsearch.amazonaws.com"
|
| -FULL_URL = 'http://%s/2011-02-01/search' % HOSTNAME
|
| -
|
| -
|
| -class CloudSearchSearchBaseTest(unittest.TestCase):
|
| -
|
| - hits = [
|
| - {
|
| - 'id': '12341',
|
| - 'title': 'Document 1',
|
| - },
|
| - {
|
| - 'id': '12342',
|
| - 'title': 'Document 2',
|
| - },
|
| - {
|
| - 'id': '12343',
|
| - 'title': 'Document 3',
|
| - },
|
| - {
|
| - 'id': '12344',
|
| - 'title': 'Document 4',
|
| - },
|
| - {
|
| - 'id': '12345',
|
| - 'title': 'Document 5',
|
| - },
|
| - {
|
| - 'id': '12346',
|
| - 'title': 'Document 6',
|
| - },
|
| - {
|
| - 'id': '12347',
|
| - 'title': 'Document 7',
|
| - },
|
| - ]
|
| -
|
| - content_type = "text/xml"
|
| - response_status = 200
|
| -
|
| - def get_args(self, requestline):
|
| - (_, request, _) = requestline.split(b" ")
|
| - (_, request) = request.split(b"?", 1)
|
| - args = six.moves.urllib.parse.parse_qs(request)
|
| - return args
|
| -
|
| - def setUp(self):
|
| - HTTPretty.enable()
|
| - body = self.response
|
| -
|
| - if not isinstance(body, bytes):
|
| - body = json.dumps(body).encode('utf-8')
|
| -
|
| - HTTPretty.register_uri(HTTPretty.GET, FULL_URL,
|
| - body=body,
|
| - content_type=self.content_type,
|
| - status=self.response_status)
|
| -
|
| - def tearDown(self):
|
| - HTTPretty.disable()
|
| -
|
| -class CloudSearchSearchTest(CloudSearchSearchBaseTest):
|
| - response = {
|
| - 'rank': '-text_relevance',
|
| - 'match-expr': "Test",
|
| - 'hits': {
|
| - 'found': 30,
|
| - 'start': 0,
|
| - 'hit': CloudSearchSearchBaseTest.hits
|
| - },
|
| - 'info': {
|
| - 'rid': 'b7c167f6c2da6d93531b9a7b314ad030b3a74803b4b7797edb905ba5a6a08',
|
| - 'time-ms': 2,
|
| - 'cpu-time-ms': 0
|
| - }
|
| -
|
| - }
|
| -
|
| - def test_cloudsearch_qsearch(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(q='Test')
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b'q'], [b"Test"])
|
| - self.assertEqual(args[b'start'], [b"0"])
|
| - self.assertEqual(args[b'size'], [b"10"])
|
| -
|
| - def test_cloudsearch_bqsearch(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(bq="'Test'")
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b'bq'], [b"'Test'"])
|
| -
|
| - def test_cloudsearch_search_details(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(q='Test', size=50, start=20)
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b'q'], [b"Test"])
|
| - self.assertEqual(args[b'size'], [b"50"])
|
| - self.assertEqual(args[b'start'], [b"20"])
|
| -
|
| - def test_cloudsearch_facet_single(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(q='Test', facet=["Author"])
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b'facet'], [b"Author"])
|
| -
|
| - def test_cloudsearch_facet_multiple(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(q='Test', facet=["author", "cat"])
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b'facet'], [b"author,cat"])
|
| -
|
| - def test_cloudsearch_facet_constraint_single(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(
|
| - q='Test',
|
| - facet_constraints={'author': "'John Smith','Mark Smith'"})
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b'facet-author-constraints'],
|
| - [b"'John Smith','Mark Smith'"])
|
| -
|
| - def test_cloudsearch_facet_constraint_multiple(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(
|
| - q='Test',
|
| - facet_constraints={'author': "'John Smith','Mark Smith'",
|
| - 'category': "'News','Reviews'"})
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b'facet-author-constraints'],
|
| - [b"'John Smith','Mark Smith'"])
|
| - self.assertEqual(args[b'facet-category-constraints'],
|
| - [b"'News','Reviews'"])
|
| -
|
| - def test_cloudsearch_facet_sort_single(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(q='Test', facet_sort={'author': 'alpha'})
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b'facet-author-sort'], [b'alpha'])
|
| -
|
| - def test_cloudsearch_facet_sort_multiple(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(q='Test', facet_sort={'author': 'alpha',
|
| - 'cat': 'count'})
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b'facet-author-sort'], [b'alpha'])
|
| - self.assertEqual(args[b'facet-cat-sort'], [b'count'])
|
| -
|
| - def test_cloudsearch_top_n_single(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(q='Test', facet_top_n={'author': 5})
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b'facet-author-top-n'], [b'5'])
|
| -
|
| - def test_cloudsearch_top_n_multiple(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(q='Test', facet_top_n={'author': 5, 'cat': 10})
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b'facet-author-top-n'], [b'5'])
|
| - self.assertEqual(args[b'facet-cat-top-n'], [b'10'])
|
| -
|
| - def test_cloudsearch_rank_single(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(q='Test', rank=["date"])
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b'rank'], [b'date'])
|
| -
|
| - def test_cloudsearch_rank_multiple(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(q='Test', rank=["date", "score"])
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b'rank'], [b'date,score'])
|
| -
|
| - def test_cloudsearch_result_fields_single(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(q='Test', return_fields=['author'])
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b'return-fields'], [b'author'])
|
| -
|
| - def test_cloudsearch_result_fields_multiple(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(q='Test', return_fields=['author', 'title'])
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b'return-fields'], [b'author,title'])
|
| -
|
| - def test_cloudsearch_t_field_single(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(q='Test', t={'year': '2001..2007'})
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b't-year'], [b'2001..2007'])
|
| -
|
| - def test_cloudsearch_t_field_multiple(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - search.search(q='Test', t={'year': '2001..2007', 'score': '10..50'})
|
| -
|
| - args = self.get_args(HTTPretty.last_request.raw_requestline)
|
| -
|
| - self.assertEqual(args[b't-year'], [b'2001..2007'])
|
| - self.assertEqual(args[b't-score'], [b'10..50'])
|
| -
|
| - def test_cloudsearch_results_meta(self):
|
| - """Check returned metadata is parsed correctly"""
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - results = search.search(q='Test')
|
| -
|
| - # These rely on the default response which is fed into HTTPretty
|
| - self.assertEqual(results.rank, "-text_relevance")
|
| - self.assertEqual(results.match_expression, "Test")
|
| -
|
| - def test_cloudsearch_results_info(self):
|
| - """Check num_pages_needed is calculated correctly"""
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - results = search.search(q='Test')
|
| -
|
| - # This relies on the default response which is fed into HTTPretty
|
| - self.assertEqual(results.num_pages_needed, 3.0)
|
| -
|
| - def test_cloudsearch_results_matched(self):
|
| - """
|
| - Check that information objects are passed back through the API
|
| - correctly.
|
| - """
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| - query = search.build_query(q='Test')
|
| -
|
| - results = search(query)
|
| -
|
| - self.assertEqual(results.search_service, search)
|
| - self.assertEqual(results.query, query)
|
| -
|
| - def test_cloudsearch_results_hits(self):
|
| - """Check that documents are parsed properly from AWS"""
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - results = search.search(q='Test')
|
| -
|
| - hits = list(map(lambda x: x['id'], results.docs))
|
| -
|
| - # This relies on the default response which is fed into HTTPretty
|
| - self.assertEqual(
|
| - hits, ["12341", "12342", "12343", "12344",
|
| - "12345", "12346", "12347"])
|
| -
|
| - def test_cloudsearch_results_iterator(self):
|
| - """Check the results iterator"""
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - results = search.search(q='Test')
|
| - results_correct = iter(["12341", "12342", "12343", "12344",
|
| - "12345", "12346", "12347"])
|
| - for x in results:
|
| - self.assertEqual(x['id'], next(results_correct))
|
| -
|
| -
|
| - def test_cloudsearch_results_internal_consistancy(self):
|
| - """Check the documents length matches the iterator details"""
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - results = search.search(q='Test')
|
| -
|
| - self.assertEqual(len(results), len(results.docs))
|
| -
|
| - def test_cloudsearch_search_nextpage(self):
|
| - """Check next page query is correct"""
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| - query1 = search.build_query(q='Test')
|
| - query2 = search.build_query(q='Test')
|
| -
|
| - results = search(query2)
|
| -
|
| - self.assertEqual(results.next_page().query.start,
|
| - query1.start + query1.size)
|
| - self.assertEqual(query1.q, query2.q)
|
| -
|
| -class CloudSearchSearchFacetTest(CloudSearchSearchBaseTest):
|
| - response = {
|
| - 'rank': '-text_relevance',
|
| - 'match-expr': "Test",
|
| - 'hits': {
|
| - 'found': 30,
|
| - 'start': 0,
|
| - 'hit': CloudSearchSearchBaseTest.hits
|
| - },
|
| - 'info': {
|
| - 'rid': 'b7c167f6c2da6d93531b9a7b314ad030b3a74803b4b7797edb905ba5a6a08',
|
| - 'time-ms': 2,
|
| - 'cpu-time-ms': 0
|
| - },
|
| - 'facets': {
|
| - 'tags': {},
|
| - 'animals': {'constraints': [{'count': '2', 'value': 'fish'}, {'count': '1', 'value': 'lions'}]},
|
| - }
|
| - }
|
| -
|
| - def test_cloudsearch_search_facets(self):
|
| - #self.response['facets'] = {'tags': {}}
|
| -
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - results = search.search(q='Test', facet=['tags'])
|
| -
|
| - self.assertTrue('tags' not in results.facets)
|
| - self.assertEqual(results.facets['animals'], {u'lions': u'1', u'fish': u'2'})
|
| -
|
| -
|
| -class CloudSearchNonJsonTest(CloudSearchSearchBaseTest):
|
| - response = b'<html><body><h1>500 Internal Server Error</h1></body></html>'
|
| - response_status = 500
|
| - content_type = 'text/xml'
|
| -
|
| - def test_response(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - with self.assertRaises(SearchServiceException):
|
| - search.search(q='Test')
|
| -
|
| -
|
| -class CloudSearchUnauthorizedTest(CloudSearchSearchBaseTest):
|
| - response = b'<html><body><h1>403 Forbidden</h1>foo bar baz</body></html>'
|
| - response_status = 403
|
| - content_type = 'text/html'
|
| -
|
| - def test_response(self):
|
| - search = SearchConnection(endpoint=HOSTNAME)
|
| -
|
| - with self.assertRaisesRegexp(SearchServiceException, 'foo bar baz'):
|
| - search.search(q='Test')
|
| -
|
| -
|
| -class FakeResponse(object):
|
| - status_code = 405
|
| - content = b''
|
| -
|
| -
|
| -class CloudSearchConnectionTest(unittest.TestCase):
|
| - cloudsearch = True
|
| -
|
| - def setUp(self):
|
| - super(CloudSearchConnectionTest, self).setUp()
|
| - self.conn = SearchConnection(
|
| - endpoint='test-domain.cloudsearch.amazonaws.com'
|
| - )
|
| -
|
| - def test_expose_additional_error_info(self):
|
| - mpo = mock.patch.object
|
| - fake = FakeResponse()
|
| - fake.content = b'Nopenopenope'
|
| -
|
| - # First, in the case of a non-JSON, non-403 error.
|
| - with mpo(requests, 'get', return_value=fake) as mock_request:
|
| - with self.assertRaises(SearchServiceException) as cm:
|
| - self.conn.search(q='not_gonna_happen')
|
| -
|
| - self.assertTrue('non-json response' in str(cm.exception))
|
| - self.assertTrue('Nopenopenope' in str(cm.exception))
|
| -
|
| - # Then with JSON & an 'error' key within.
|
| - fake.content = json.dumps({
|
| - 'error': "Something went wrong. Oops."
|
| - }).encode('utf-8')
|
| -
|
| - with mpo(requests, 'get', return_value=fake) as mock_request:
|
| - with self.assertRaises(SearchServiceException) as cm:
|
| - self.conn.search(q='no_luck_here')
|
| -
|
| - self.assertTrue('Unknown error' in str(cm.exception))
|
| - self.assertTrue('went wrong. Oops' in str(cm.exception))
|
|
|