Skip to content
This repository has been archived by the owner on May 23, 2019. It is now read-only.

Commit

Permalink
Reduce repetition in test_completer.py 🚧
Browse files Browse the repository at this point in the history
  • Loading branch information
@nicolewhite
nicolewhite committed Jan 29, 2016
1 parent b34ddc6 commit 18f1d95
Showing 1 changed file with 35 additions and 178 deletions.
213 changes: 35 additions & 178 deletions tests/test_completer.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -17,309 +17,166 @@ def complete_event():
from mock import Mock
return Mock()

def test_empty_string_completion(completer, complete_event):
text = ""
position = 0
def get_completions(completer, complete_event, text):
position = len(text)

result = list(completer.get_completions(
return list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

def test_empty_string_completion(completer, complete_event):
text = ""
result = get_completions(completer, complete_event, text)
assert result == []

def test_label_completion(completer, complete_event):
text = "MATCH (p:P"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="Person", start_position=-1)]

def test_label_completion_after_rel(completer, complete_event):
text = "MATCH (p:Person)-[:ACTED_IN]->(m:M"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="Movie", start_position=-1)]

def test_label_completion_with_backticks(completer, complete_event):
text = "MATCH (p:`P"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="Person", start_position=-1)]

def test_label_completion_with_underscores(completer, complete_event):
text = "MATCH (u:Under_S"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="Under_Score", start_position=-7)]

def test_label_completion_with_backticks_and_spaces(completer, complete_event):
text = "MATCH (s:`Spa C"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="Spa Ce", start_position=-5)]

def test_label_completion_with_backticks_and_underscores(completer, complete_event):
text = "MATCH (u:`Under_S"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="Under_Score", start_position=-7)]

def test_rel_completion(completer, complete_event):
text = "MATCH (p:Person)-[:A"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="ACTED_IN", start_position=-1)]

def test_rel_completion_with_backticks(completer, complete_event):
text = "MATCH (p:Person)-[:`A"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="ACTED_IN", start_position=-1)]

def test_rel_completion_with_underscores(completer, complete_event):
text = "MATCH (p:Person)-[:ACTED_I"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="ACTED_IN", start_position=-7)]

def test_rel_completion_with_backsticks_and_spaces(completer, complete_event):
text = "MATCH (p:Person)-[`:SPA C"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="SPA CE", start_position=-5)]

def test_rel_completion_with_backticks_and_underscores(completer, complete_event):
text = "MATCH (p:Person)-[:`ACTED_I"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="ACTED_IN", start_position=-7)]

def test_property_completion(completer, complete_event):
text = "MATCH (p:Person) WHERE p.n"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="name", start_position=-1)]

def test_property_completion_with_backticks(completer, complete_event):
text = "MATCH (p:Person) WHERE p.`n"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="name", start_position=-1)]

def test_property_completion_with_underscores(completer, complete_event):
text = "MATCH (p:Person) WHERE p.under_s"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="under_score", start_position=-7)]

def test_property_completion_with_backticks_and_underscores(completer, complete_event):
text = "MATCH (p:Person) WHERE p.`under_s"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="under_score", start_position=-7)]

def test_property_completion_with_backticks_and_spaces(completer, complete_event):
text = "MATCH (p:Person) WHERE p.`spa c"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="spa ce", start_position=-5)]

def test_no_property_completion_in_float(completer, complete_event):
text = "RETURN 5."
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == []

def test_no_property_completion_in_float_after_export(completer, complete_event):
text = "export k=5."
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == []

def test_no_cypher_completions_after_non_alpha(completer, complete_event):
text = "MATCH p WHERE p.name ="
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == []

def property_completion_in_identifier_with_number(completer, complete_event):
text = "MATCH (p1:Person) WHERE p1."
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="name", start_position=0), Completion(text="title", start_position=0)]

def test_no_completions_while_inside_string(completer, complete_event):
text = "MATCH (p:Person) WHERE p.name = 'Tom H"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == []

def test_match_keyword_completion(completer, complete_event):
text = "MAT"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="MATCH", start_position=-3)]

def test_nested_function_completion(completer, complete_event):
text = "MATCH n RETURN LENGTH(COL"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result == [Completion(text="COLLECT", start_position=-3)]

def test_markov_w_after_match(completer, complete_event):
# Out of the words starting with W, WHERE is most commonly used after MATCH.
text = "MATCH n W"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result[0] == Completion(text="WHERE", start_position=-1)

def test_markov_w_after_case(completer, complete_event):
# Out of the words starting with W, WHEN is most commonly used after CASE.
text = "MATCH n RETURN CASE W"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result[0] == Completion(text="WHEN", start_position=-1)

def test_markov_w_after_nothing(completer, complete_event):
# Out of the words starting with W, WITH is most commonly the first word in a query.
text = "W"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result[0] == Completion(text="WITH", start_position=-1)

def test_markov_m_after_nothing(completer, complete_event):
# Out of the words starting with M, MATCH and MERGE are most commonly the first words in a query, in that order.
text = "M"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result[0] == Completion(text="MATCH", start_position=-1)
assert result[1] == Completion(text="MERGE", start_position=-1)

def test_markov_a_after_where(completer, complete_event):
# Out of the words starting with A, AND is most commonly used after WHERE.
text = "MATCH (p:Person) WHERE p.born > 1950 A"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result[0] == Completion(text="AND", start_position=-1)

def test_markov_a_after_return(completer, complete_event):
# Out of the words starting with A, AS is most commonly used after RETURN.
text = "MATCH (p:Person) RETURN p.name A"
position = len(text)

result = list(completer.get_completions(
Document(text=text, cursor_position=position),
complete_event))

result = get_completions(completer, complete_event, text)
assert result[0] == Completion(text="AS", start_position=-1)

0 comments on commit 18f1d95

Please sign in to comment.