def test_complete_bracket_field(self):
self.model.load(DataSourceType.VARIABLE,
{"Alice's key": {
"key": "value"
}})
_, _, options = autocomplete("$.['Alice\\\'s key']", self.model.query)
self.assertEqual(['.', '['], options)
def test_incomplete_bracket_field_after_array_wildcard(self):
self.model.load(DataSourceType.VARIABLE, [{
"key": "v1"
}, {
"key": "v2"
}])
_, _, options = autocomplete("$[*][", self.model.query)
self.assertEqual(["['key']"], options)
def test_incomplete_single_dot_after_array_wildcard(self):
self.model.load(DataSourceType.VARIABLE, [{
"key": "v1"
}, {
"key": "v2"
}])
_, _, options = autocomplete("$[*].", self.model.query)
self.assertEqual(["key"], options)
def test_incomplete_union_after_array_wildcard(self):
self.model.load(DataSourceType.VARIABLE, [{
"k1": "v1.0",
"k2": "v2.0"
}, {
"k1": "v1.1",
"k2": "v2.1"
}])
_, _, options = autocomplete("$[*]['k1','", self.model.query)
self.assertEqual(["'k2'"], options)
def test_array_index(self):
self.model.load(DataSourceType.VARIABLE, ["item1", "item2"])
is_partial_complete, _, options = autocomplete("$[", self.model.query)
self.assertEqual(["[*]", "[0]", "[1]"], options)
self.assertEqual(False, is_partial_complete)
def test_invalid_bracket(self):
_, _, options = autocomplete("$[[", self.model.query)
self.assertEqual([], options)
def test_incomplete_open_bracket(self):
self.model.load(DataSourceType.VARIABLE, {"key": "value"})
_, _, options = autocomplete("$.['k", self.model.query)
self.assertEqual(["['key']"], options)
def test_dot_complete_field(self):
self.model.load(DataSourceType.VARIABLE, {"key": "value"})
_, _, options = autocomplete("$.key", self.model.query)
self.assertEqual([".", "["], options)
def test_invalid_dots(self):
# invalid case
_, _, options = autocomplete("$...", self.model.query)
self.assertEqual([], options)
def test_double_dot(self):
self.model.load(DataSourceType.VARIABLE, {"key": "value"})
_, _, options = autocomplete("$..", self.model.query)
self.assertEqual([], options)
def test_incomplete_union_second_key(self):
self.model.load(DataSourceType.VARIABLE, {"k1": "v1", "k2": "v2"})
_, _, options = autocomplete("$.['k1','", self.model.query)
self.assertEqual(["'k2'"], options)
def test_dotted_open_filter(self):
self.model.load(DataSourceType.VARIABLE, {"key": "value"})
_, _, options = autocomplete("$.[?(", self.model.query)
self.assertEqual(["@.key"], options)
def test_nested_list_with_single_dot(self):
self.model.load(DataSourceType.VARIABLE, {"key": ["item1", "item2"]})
is_partial_complete, _, options = autocomplete("$.key.",
self.model.query)
self.assertEqual(["[*]", "[0]", "[1]"], options)
self.assertEqual(False, is_partial_complete)
def test_open_filter_with_only_question_mark(self):
self.model.load(DataSourceType.VARIABLE, {"key": "value"})
_, _, options = autocomplete("$[?", self.model.query)
self.assertEqual(["("], options)
def test_empty_string(self):
_, _, options = autocomplete("", self.model.query)
self.assertEqual(["$"], options)