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_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([("", "")]), ), ]