def test_get_token_range(self):
"""Should return a range around the word"""
# Given a string with some words
words = ['create', 'table', 'T1'] # indexes: 0-5 7-11 13-14
line = ' '.join(words)
start_col_to_word: dict = {
0: 0,
3: 0,
6: 0, # if on space between words, should select previous word
8: 1,
10: 1,
12: 1, # on space between words, should select previous
13: 2, # 'c' -> last indent
14: 2
}
# When I lookup the word range for a given input
for start, expected_word_index in start_col_to_word.items():
expected_word = words[expected_word_index]
expected_line = 2
pos = Position.from_data(2, start)
text_range: Range = TextUtilities.get_token_range(pos, line)
# Then I expect the line value to always match the Positions line
self.assertEqual(text_range.start.line, expected_line)
self.assertEqual(text_range.end.line, expected_line)
# ... and I expect the word to be the one covered by the indent
actual_word = line[text_range.start.character:text_range.end.
character]
self.assertEqual(
actual_word, expected_word,
'For start {0} expected "{1}" actual "{2}"'.format(
start, expected_word, actual_word)) # noqa
def test_next_delim_not_found(self):
"""Should return length of line if delim does not exist"""
line = '01 34textnospace'
start_cols = [3, 5, len(line) - 1]
for col in start_cols:
pos = TextUtilities.next_delimiter_pos(line, col)
self.assertEqual(pos, len(line))
def test_prev_delim_not_found(self):
"""Should return 0 if delim does not exist"""
line = '0123456789'
start_cols = [0, 1, 5, len(line) - 1]
for col in start_cols:
pos = TextUtilities.prev_delimiter_pos(line, col)
self.assertEqual(
pos, 0,
'For start {0} with value "{1}" expected {2} actual {3}'.
format(col, line[col], 0, pos))
def _send_default_completions(self, request_context: RequestContext,
script_file: ScriptFile,
params: TextDocumentPosition) -> bool:
response = []
line: str = script_file.get_line(params.position.line)
(token_text,
text_range) = TextUtilities.get_text_and_range(params.position, line)
if token_text:
completions = self._completion_helper.get_matches(
token_text, text_range, self.should_lowercase)
response = completions
request_context.send_response(response)
return True
def test_next_delim_found(self):
"""Should return length of line if delim does not exist"""
line = '01 345\t789\na/cd'
start_col_to_delim: dict = {
0: 2,
3: 6,
4: 6,
8: 10,
10:
10, # when cursor is on a delimiter, return it as the next delimiter
12: 12, # '/'
13: 15 # 'c' -> end of line
}
for start, expected_col in start_col_to_delim.items():
pos = TextUtilities.next_delimiter_pos(line, start)
self.assertEqual(
pos, expected_col,
'For start {0} with value "{1}" expected {2} actual {3}'.
format(start, line[start], expected_col, pos))
def test_prev_delim_found(self):
"""Should return 0 if on first word or the first index after the delimiter for all others"""
line = '01 345\t789\na/cd'
start_col_to_delim: dict = {
0: 0,
3: 3,
4: 3,
8: 7,
# when on a cursor that is a delimiter, search for previous delimiter.
# this is important as otherwise, if at the end of a word the range would be
# the empty string instead of the previous word
10: 7,
11: 11,
12: 11, # '/' return previous since its an indent
13: 13
}
for start, expected_col in start_col_to_delim.items():
pos = TextUtilities.prev_delimiter_pos(line, start)
self.assertEqual(
pos, expected_col,
'For start {0} with value "{1}" expected {2} actual {3}'.
format(start, line[start], expected_col, pos))