def test_build_query_for_uri(models):
"""'uri' args are returned in the query dict in a "match" clause.
This is what happens when you open the sidebar on a page and it loads
all the annotations of that page.
"""
models.Document.get_by_uri.return_value = None
query1 = search.build_query(request_params=multidict.NestedMultiDict(
{"uri": "http://example.com/"}))
query2 = search.build_query(request_params=multidict.NestedMultiDict(
{"uri": "http://whitehouse.gov/"}))
assert query1["query"] == {
"bool": {
"must": [{
"match": {
"uri": "http://example.com/"
}
}]
}
}
assert query2["query"] == {
"bool": {
"must": [{
"match": {
"uri": "http://whitehouse.gov/"
}
}]
}
}
def test_build_query_sort_is_by_updated():
"""Sort defaults to "updated"."""
query = search.build_query(request_params=multidict.NestedMultiDict())
sort = query["sort"]
assert len(sort) == 1
assert sort[0].keys() == ["updated"]
def test_build_query_sort_includes_ignore_unmapped():
"""'ignore_unmapped': True is used in the sort clause."""
query = search.build_query(
request_params=multidict.NestedMultiDict())
sort = query["sort"]
assert sort[0]["updated"]["ignore_unmapped"] == True
def test_build_query_with_custom_order():
"""'order' params are returned in the query dict if given."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"order": "asc"}))
sort = query["sort"]
assert sort[0]["updated"]["order"] == "asc"
def test_build_query_with_custom_sort():
"""Custom sorts are returned in the query dict."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"sort": "title"}))
sort = query["sort"]
assert sort == [{'title': {'ignore_unmapped': True, 'order': 'desc'}}]
def test_build_query_for_uri_with_multiple_representations(models):
"""It should search for any URI returned by the Document class.
If models.Document.get_by_uri() returns multiple documents for the URI then
build_query() should return a query that finds annotations that match one
or more of these documents' URIs.
"""
doc = mock.MagicMock()
doc.uris.return_value = [
"http://example.com/", "http://example2.com/", "http://example3.com/"]
models.Document.get_by_uri.return_value = doc
query = search.build_query(
request_params=multidict.NestedMultiDict(
{"uri": "http://example.com/"}))
assert query["query"] == {
"bool": {
"must": [
{
"bool": {
"minimum_should_match": 1,
"should": [
{"match": {"uri": "http://example.com/"}},
{"match": {"uri": "http://example2.com/"}},
{"match": {"uri": "http://example3.com/"}}
]
}
}
]
}
}
def test_build_query_for_user():
"""'user' params returned in the query dict in "match" clauses."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"user": "******"}))
assert query["query"] == {
"bool": {"must": [{"match": {"user": "******"}}]}}
def test_build_query_with_invalid_limit():
"""Invalid 'limit' params should be ignored."""
for invalid_limit in ("foo", '', ' ', "-23", "32.7"):
query = search.build_query(
request_params=multidict.NestedMultiDict({"limit": invalid_limit}))
assert query["size"] == 20 # (20 is the default value.)
def test_build_query_with_invalid_offset():
"""Invalid 'offset' params should be ignored."""
for invalid_offset in ("foo", '', ' ', "-23", "32.7"):
query = search.build_query(request_params=multidict.NestedMultiDict(
{"offset": invalid_offset}))
assert query["from"] == 0
def test_build_query_order_defaults_to_desc():
"""'order': "desc" is returned in the query dict by default."""
query = search.build_query(
request_params=multidict.NestedMultiDict())
sort = query["sort"]
assert sort[0]["updated"]["order"] == "desc"
def test_build_query_for_tag():
"""'tags' params are returned in the query dict in "match" clauses."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"tags": "foo"}))
assert query["query"]["filtered"]["query"] == {
"bool": {"must": [{"match": {"tags": "foo"}}]}}
def test_build_query_with_custom_order():
"""'order' params are returned in the query dict if given."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"order": "asc"}))
sort = query["sort"]
assert sort[0]["updated"]["order"] == "asc"
def test_build_query_returns_nipsa_filter(nipsa_filter):
"""_build_query() returns a nipsa-filtered query."""
nipsa_filter.return_value = "foobar!"
query = search.build_query(multidict.NestedMultiDict())
assert query["query"]["filtered"]["filter"] == "foobar!"
def test_build_query_for_uri_with_multiple_representations(uri):
"""It should expand the search to all URIs.
If h.api.uri.expand returns multiple documents for the URI then
build_query() should return a query that finds annotations that match one
or more of these documents' URIs.
"""
results = ["http://example.com/",
"http://example2.com/",
"http://example3.com/"]
uri.expand.side_effect = lambda x: results
uri.normalise.side_effect = lambda x: x
query = search.build_query(
request_params=multidict.NestedMultiDict(
{"uri": "http://example.com/"}))
assert query["query"]["filtered"]["query"] == {
"bool": {
"must": [
{
"bool": {
"minimum_should_match": 1,
"should": [
{"match": {"uri": "http://example.com/"}},
{"match": {"uri": "http://example2.com/"}},
{"match": {"uri": "http://example3.com/"}}
]
}
}
]
}
}
def test_build_query_with_single_quote_param():
"""'quote' params are returned in the query dict in "match" clauses."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"quote": "foobar"}))
assert query["query"]["filtered"]["query"] == {
"bool": {"must": [{"match": {"quote": "foobar"}}]}}
def test_build_query_with_invalid_limit():
"""Invalid 'limit' params should be ignored."""
for invalid_limit in ("foo", '', ' ', "-23", "32.7"):
query = search.build_query(
request_params=multidict.NestedMultiDict({"limit": invalid_limit}))
assert query["size"] == 20 # (20 is the default value.)
def test_build_query_query_defaults_to_match_all():
"""If no query params are given a "match_all": {} query is returned."""
query = search.build_query(
request_params=multidict.NestedMultiDict())
assert query["query"]["filtered"]["query"] == {
"bool": {"must": [{"match_all": {}}]}}
def test_build_query_with_custom_sort():
"""Custom sorts are returned in the query dict."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"sort": "title"}))
sort = query["sort"]
assert sort == [{'title': {'ignore_unmapped': True, 'order': 'desc'}}]
def test_build_query_with_invalid_offset():
"""Invalid 'offset' params should be ignored."""
for invalid_offset in ("foo", '', ' ', "-23", "32.7"):
query = search.build_query(
request_params=multidict.NestedMultiDict(
{"offset": invalid_offset}))
assert query["from"] == 0
def test_build_query_sort_is_by_updated():
"""Sort defaults to "updated"."""
query = search.build_query(
request_params=multidict.NestedMultiDict())
sort = query["sort"]
assert len(sort) == 1
assert sort[0].keys() == ["updated"]
def test_build_query_with_evil_arguments():
params = multidict.NestedMultiDict({
"offset": "3foo",
"limit": '\' drop table annotations'
})
query = search.build_query(request_params=params)
assert query["query"] == {'bool': {'must': [{'match_all': {}}]}}
def test_build_query_with_evil_arguments():
params = multidict.NestedMultiDict({
"offset": "3foo",
"limit": '\' drop table annotations'
})
query = search.build_query(request_params=params)
assert query["query"] == {'bool': {'must': [{'match_all': {}}]}}
def test_build_query_with_combined_user_and_tag_query():
"""A 'user' and a 'param' at the same time are handled correctly."""
query = search.build_query(
request_params=multidict.NestedMultiDict(
{"user": "******", "tags": "foo"}))
assert query["query"]["filtered"]["query"] == {
"bool": {"must": [
{"match": {"user": "******"}},
{"match": {"tags": "foo"}},
]}}
def test_build_query_for_uri(uri):
"""'uri' args are returned in the query dict in a "match" clause.
This is what happens when you open the sidebar on a page and it loads
all the annotations of that page.
"""
uri.expand.side_effect = lambda x: [x]
uri.normalise.side_effect = lambda x: x
query1 = search.build_query(
request_params=multidict.NestedMultiDict(
{"uri": "http://example.com/"}))
query2 = search.build_query(
request_params=multidict.NestedMultiDict(
{"uri": "http://whitehouse.gov/"}))
assert query1["query"]["filtered"]["query"] == {
"bool": {"must": [{"match": {"uri": "http://example.com/"}}]}}
assert query2["query"]["filtered"]["query"] == {
"bool": {"must": [{"match": {"uri": "http://whitehouse.gov/"}}]}}
def test_build_query_for_uri():
"""'uri' args are returned in the query dict in a "match" clause.
This is what happens when you open the sidebar on a page and it loads
all the annotations of that page.
"""
query = search.build_query(
request_params=multidict.NestedMultiDict(
{"uri": "http://example.com/"}))
assert query["query"] == {
"bool": {"must": [{"match": {"uri": "http://example.com/"}}]}}
def test_build_query_with_single_quote_param():
"""'quote' params are returned in the query dict in "match" clauses."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"quote": "foobar"}))
assert query["query"] == {
"bool": {
"must": [{
"match": {
"quote": "foobar"
}
}]
}
}
def test_build_query_with_multiple_quote_params():
"""Multiple "quote" request params produce multiple "match" clauses."""
params = multidict.MultiDict()
params.add("quote", "foo")
params.add("quote", "bar")
query = search.build_query(request_params=params)
assert query["query"]["filtered"]["query"] == {
"bool": {
"must": [
{"match": {"quote": "foo"}},
{"match": {"quote": "bar"}}
]
}
}
def test_build_query_with_multiple_keywords():
"""Multiple keywords at once are handled correctly."""
params = multidict.MultiDict()
params.add("any", "howdy")
params.add("any", "there")
query = search.build_query(request_params=params)
assert query["query"]["filtered"]["query"] == {
"bool": {"must": [{"multi_match": {
"fields": ["quote", "tags", "text", "uri.parts", "user"],
"query": ["howdy", "there"],
"type": "cross_fields"
}}]}
}
def test_build_query_for_multiple_tags():
"""Multiple "tags" params go into multiple "match" dicts."""
params = multidict.MultiDict()
params.add("tags", "foo")
params.add("tags", "bar")
query = search.build_query(request_params=params)
assert query["query"]["filtered"]["query"] == {
"bool": {
"must": [
{"match": {"tags": "foo"}},
{"match": {"tags": "bar"}}
]
}
}
def test_build_query_for_multiple_users():
"""Multiple "user" params go into multiple "match" dicts."""
params = multidict.MultiDict()
params.add("user", "fred")
params.add("user", "bob")
query = search.build_query(request_params=params)
assert query["query"]["filtered"]["query"] == {
"bool": {
"must": [
{"match": {"user": "******"}},
{"match": {"user": "******"}}
]
}
}
def test_build_query_with_keyword():
"""Keywords are returned in the query dict in a "multi_match" clause."""
params = multidict.MultiDict()
params.add("any", "howdy")
query = search.build_query(request_params=params)
assert query["query"] == {
"bool": {
"must": [{
"multi_match": {
"fields": ["quote", "tags", "text", "uri.parts", "user"],
"query": ["howdy"],
"type": "cross_fields"
}
}]
}
}
def test_build_query_with_multiple_keywords():
"""Multiple keywords at once are handled correctly."""
params = multidict.MultiDict()
params.add("any", "howdy")
params.add("any", "there")
query = search.build_query(request_params=params)
assert query["query"] == {
"bool": {
"must": [{
"multi_match": {
"fields": ["quote", "tags", "text", "uri.parts", "user"],
"query": ["howdy", "there"],
"type": "cross_fields"
}
}]
}
}
def test_build_query_with_multiple_quote_params():
"""Multiple "quote" request params produce multiple "match" clauses."""
params = multidict.MultiDict()
params.add("quote", "foo")
params.add("quote", "bar")
query = search.build_query(request_params=params)
assert query["query"] == {
"bool": {
"must": [{
"match": {
"quote": "foo"
}
}, {
"match": {
"quote": "bar"
}
}]
}
}
def test_build_query_for_multiple_users():
"""Multiple "user" params go into multiple "match" dicts."""
params = multidict.MultiDict()
params.add("user", "fred")
params.add("user", "bob")
query = search.build_query(request_params=params)
assert query["query"] == {
"bool": {
"must": [{
"match": {
"user": "******"
}
}, {
"match": {
"user": "******"
}
}]
}
}
def test_build_query_with_arbitrary_params():
"""You can pass any ?foo=bar param to the search API.
Arbitrary parameters that aren't handled specially are simply passed to
Elasticsearch as match clauses. This way search queries can match against
any fields in the Elasticsearch object.
"""
params = multidict.NestedMultiDict({"foo.bar": "arbitrary"})
query = search.build_query(request_params=params)
assert query["query"] == {
'bool': {
'must': [{
'match': {
'foo.bar': 'arbitrary'
}
}]
}
}
def test_build_query_with_keyword():
"""Keywords are returned in the query dict in a "multi_match" clause."""
params = multidict.MultiDict()
params.add("any", "howdy")
query = search.build_query(request_params=params)
assert query["query"]["filtered"]["query"] == {
"bool": {
"must": [
{
"multi_match": {
"fields": ["quote", "tags", "text", "uri.parts",
"user"],
"query": ["howdy"],
"type": "cross_fields"
}
}
]
}
}
def test_build_query_with_arbitrary_params():
"""You can pass any ?foo=bar param to the search API.
Arbitrary parameters that aren't handled specially are simply passed to
Elasticsearch as match clauses. This way search queries can match against
any fields in the Elasticsearch object.
"""
params = multidict.NestedMultiDict({"foo.bar": "arbitrary"})
query = search.build_query(request_params=params)
assert query["query"]["filtered"]["query"] == {
'bool': {
'must': [
{
'match': {'foo.bar': 'arbitrary'}
}
]
}
}
def test_build_query_for_multiple_tags():
"""Multiple "tags" params go into multiple "match" dicts."""
params = multidict.MultiDict()
params.add("tags", "foo")
params.add("tags", "bar")
query = search.build_query(request_params=params)
assert query["query"] == {
"bool": {
"must": [{
"match": {
"tags": "foo"
}
}, {
"match": {
"tags": "bar"
}
}]
}
}
def test_build_query_for_uri_with_multiple_representations(models):
"""It should search for any URI returned by the Document class.
If models.Document.get_by_uri() returns multiple documents for the URI then
build_query() should return a query that finds annotations that match one
or more of these documents' URIs.
"""
doc = mock.MagicMock()
doc.uris.return_value = [
"http://example.com/", "http://example2.com/", "http://example3.com/"
]
models.Document.get_by_uri.return_value = doc
query = search.build_query(request_params=multidict.NestedMultiDict(
{"uri": "http://example.com/"}))
assert query["query"] == {
"bool": {
"must": [{
"bool": {
"minimum_should_match":
1,
"should": [{
"match": {
"uri": "http://example.com/"
}
}, {
"match": {
"uri": "http://example2.com/"
}
}, {
"match": {
"uri": "http://example3.com/"
}
}]
}
}]
}
}
def test_build_query_with_combined_user_and_tag_query():
"""A 'user' and a 'param' at the same time are handled correctly."""
query = search.build_query(
request_params=multidict.NestedMultiDict({
"user": "******",
"tags": "foo"
}))
assert query["query"] == {
"bool": {
"must": [
{
"match": {
"user": "******"
}
},
{
"match": {
"tags": "foo"
}
},
]
}
}
def test_build_query_sort_includes_ignore_unmapped():
"""'ignore_unmapped': True is used in the sort clause."""
query = search.build_query(request_params=multidict.NestedMultiDict())
sort = query["sort"]
assert sort[0]["updated"]["ignore_unmapped"] == True
def test_build_query_for_tag():
"""'tags' params are returned in the query dict in "match" clauses."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"tags": "foo"}))
assert query["query"] == {"bool": {"must": [{"match": {"tags": "foo"}}]}}
def test_build_query_limit_strings_are_converted_to_ints():
"""String values for limit should be converted to ints."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"limit": "17"}))
assert query["size"] == 17
def test_build_query_offset_defaults_to_0():
"""If no offset is given then "from": 0 is used in the query by default."""
query = search.build_query(
request_params=multidict.NestedMultiDict())
assert query["from"] == 0
def test_build_query_users_own_annotations_are_not_filtered():
query = search.build_query(multidict.NestedMultiDict(), userid="fred")
assert {'term': {'user': '******'}} in (
query["query"]["filtered"]["filter"]["bool"]["should"])
def test_build_query_custom_limits_are_passed_in():
"""If a limit is given it's returned in the query dict as "size"."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"limit": 7}))
assert query["size"] == 7
def test_build_query_custom_limits_are_passed_in():
"""If a limit is given it's returned in the query dict as "size"."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"limit": 7}))
assert query["size"] == 7
def test_build_query_limit_defaults_to_20():
"""If no limit is given "size": 20 is used in the query by default."""
query = search.build_query(
request_params=multidict.NestedMultiDict())
assert query["size"] == 20
def test_build_query_custom_offsets_are_passed_in():
"""If an offset is given it's returned in the query dict."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"offset": 7}))
assert query["from"] == 7
def test_build_query_order_defaults_to_desc():
"""'order': "desc" is returned in the query dict by default."""
query = search.build_query(request_params=multidict.NestedMultiDict())
sort = query["sort"]
assert sort[0]["updated"]["order"] == "desc"
def test_build_query_limit_strings_are_converted_to_ints():
"""String values for limit should be converted to ints."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"limit": "17"}))
assert query["size"] == 17
def test_build_query_offset_defaults_to_0():
"""If no offset is given then "from": 0 is used in the query by default."""
query = search.build_query(request_params=multidict.NestedMultiDict())
assert query["from"] == 0
def test_build_query_offset_string_is_converted_to_int():
"""'offset' arguments should be converted from strings to ints."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"offset": "23"}))
assert query["from"] == 23
def test_build_query_query_defaults_to_match_all():
"""If no query params are given a "match_all": {} query is returned."""
query = search.build_query(request_params=multidict.NestedMultiDict())
assert query["query"] == {"bool": {"must": [{"match_all": {}}]}}
def test_build_query_limit_defaults_to_20():
"""If no limit is given "size": 20 is used in the query by default."""
query = search.build_query(request_params=multidict.NestedMultiDict())
assert query["size"] == 20
def test_build_query_for_user():
"""'user' params returned in the query dict in "match" clauses."""
query = search.build_query(
request_params=multidict.NestedMultiDict({"user": "******"}))
assert query["query"] == {"bool": {"must": [{"match": {"user": "******"}}]}}
def test_build_query_does_pass_userid_to_nipsa_filter(nipsa_filter):
search.build_query(multidict.NestedMultiDict(), userid="fred")
assert nipsa_filter.call_args[1]["userid"] == "fred"
