def test_authentication_error(tmp_app): # NOQA
from dtool_lookup_server import AuthenticationError
from dtool_lookup_server_annotation_filter_plugin.utils import (
get_annotation_key_info_by_user
)
with pytest.raises(AuthenticationError):
get_annotation_key_info_by_user("dont_exist", {})
def test_authorization(tmp_app): # NOQA
from dtool_lookup_server_annotation_filter_plugin.utils import (
get_annotation_key_info_by_user
)
# The "sleepy" user has not got access to any datasets.
actual_key_info = get_annotation_key_info_by_user(username="******", filters={}) # NOQA
assert actual_key_info == {}
def test_get_annotation_keys_by_user_no_filter(tmp_app): # NOQA
from dtool_lookup_server_annotation_filter_plugin.utils import (
get_annotation_key_info_by_user
)
expected_key_info = {"color": 2, "pattern": 2}
actual_key_info = get_annotation_key_info_by_user(username="******", filters={}) # NOQA
assert expected_key_info == actual_key_info
def test_get_annotation_keys_by_user_base_uri_filter(tmp_app): # NOQA
from dtool_lookup_server_annotation_filter_plugin.utils import (
get_annotation_key_info_by_user
)
filters = {"base_uris": ["s3://mr-men"]}
expected_key_info = {"color": 1, "pattern": 1}
actual_key_info = get_annotation_key_info_by_user("grumpy", filters) # NOQA
assert expected_key_info == actual_key_info
def test_get_annotation_keys_by_user_key_filter(tmp_app): # NOQA
from dtool_lookup_server_annotation_filter_plugin.utils import (
get_annotation_key_info_by_user
)
filters = {"annotation_keys": ["color"]}
actual_key_info = get_annotation_key_info_by_user("grumpy", filters) # NOQA
expected_key_info = {"color": 2, "pattern": 1}
assert expected_key_info == actual_key_info
def test_get_annotation_keys_by_user_base_complex_filter(tmp_app): # NOQA
from dtool_lookup_server_annotation_filter_plugin.utils import (
get_annotation_key_info_by_user
)
filters = {
"annotation_keys": ["pattern"],
"annotations": {"pattern": "stripey"}
}
expected_key_info = {"pattern": 1}
actual_key_info = get_annotation_key_info_by_user("grumpy", filters) # NOQA
assert expected_key_info == actual_key_info
filters = {
"annotation_keys": ["color"],
"annotations": {"pattern": "stripey"}
}
expected_key_info = {}
actual_key_info = get_annotation_key_info_by_user("grumpy", filters) # NOQA
assert expected_key_info == actual_key_info