def test_newbie_mode_complete_without_meta_dict():
fake_document = MagicMock()
fake_document.text_before_cursor = fake_document.text = "GEOR"
completer = IRedisCompleter(hint=False)
completions = list(completer.get_completions(fake_document, None))
assert [word.text for word in completions] == ["GEORADIUS", "GEORADIUSBYMEMBER"]
assert [word.display_meta for word in completions] == [
FormattedText([("", "")]),
FormattedText([("", "")]),
]
def test_username_completer():
completer = IRedisCompleter()
completer.update_completer_for_input("acl deluser laixintao")
completer.update_completer_for_input("acl deluser antirez")
fake_document = MagicMock()
fake_document.text_before_cursor = fake_document.text = "acl deluser "
completions = list(completer.get_completions(fake_document, None))
assert sorted([completion.text for completion in completions]) == [
"antirez",
"laixintao",
]
def test_completion_casing():
c = IRedisCompleter(completion_casing="auto")
fake_document = MagicMock()
fake_document.text = fake_document.text_before_cursor = "ge"
assert [
completion.text
for completion in c.get_completions(fake_document, None)
] == [
"get",
"getset",
"getbit",
"geopos",
"geoadd",
"geohash",
"geodist",
"getrange",
"georadius",
"georadiusbymember",
]
c = IRedisCompleter(completion_casing="auto")
fake_document.text = fake_document.text_before_cursor = "GET"
assert [
completion.text
for completion in c.get_completions(fake_document, None)
] == ["GET", "GETSET", "GETBIT", "GETRANGE"]
c = IRedisCompleter(completion_casing="upper")
fake_document.text = fake_document.text_before_cursor = "get"
assert [
completion.text
for completion in c.get_completions(fake_document, None)
] == ["GET", "GETSET", "GETBIT", "GETRANGE"]
c = IRedisCompleter(completion_casing="lower")
fake_document.text = fake_document.text_before_cursor = "GET"
assert [
completion.text
for completion in c.get_completions(fake_document, None)
] == ["get", "getset", "getbit", "getrange"]
def test_newbie_mode_complete_with_meta_dict():
fake_document = MagicMock()
fake_document.text_before_cursor = "GEOR"
completer = IRedisCompleter(hint=True)
completions = list(completer.get_completions(fake_document, None))
assert sorted([completion.display_meta for completion in completions]) == [
FormattedText([(
"",
"Query a sorted set representing a geospatial index to fetch members matching a given maximum distance from a member", # noqa
)]),
FormattedText([(
"",
"Query a sorted set representing a geospatial index to fetch members matching a given maximum distance from a point", # noqa
)]),
]
def test_group_completer():
fake_document = MagicMock()
previous_commands = ["xgroup create abc world 123", "xgroup setid abc hello 123"]
fake_document.text = fake_document.text_before_cursor = "XGROUP DESTROY key "
completer = IRedisCompleter()
for command in previous_commands:
completer.update_completer_for_input(command)
completions = list(completer.get_completions(fake_document, None))
assert completions == [
Completion(
text="hello",
start_position=0,
display=FormattedText([("", "hello")]),
display_meta=FormattedText([("", "")]),
),
Completion(
text="world",
start_position=0,
display=FormattedText([("", "world")]),
display_meta=FormattedText([("", "")]),
),
]
