def expected_multiple_is_empty(self, setup_objs):
"""Calculate expected values for multiple test case, is empty operator"""
# pylint: disable=no-self-use
expected_data = [
create_tuple_data(row.obj_id, None, None)
for row in setup_objs[:MULTIPLE_ITEMS_COUNT]
]
return [(expected_data, None)]
def expected_single_equal(self, setup_objs):
"""Calculate expected values for single test case, equal operator"""
values = []
for data in setup_objs:
search_obj = inflector.get_model(data.searchable_type).query.get(
data.searchable_id)
for subprop in self.Meta.subprops:
values.append(([
create_tuple_data(data.obj_id, search_obj,
self.Meta.subprops)
], getattr(search_obj, subprop)))
return values
def expected_single_equal(self, setup_objs):
"""Calculate expected values for single test case, equal operator"""
values = []
for data in setup_objs:
search_obj = inflector.get_model(data.searchable_type).query.get(
data.searchable_id
)
for subprop in self.Meta.subprops:
values.append((
[create_tuple_data(data.obj_id, search_obj, self.Meta.subprops)],
getattr(search_obj, subprop)
))
return values
def expected_multiple_not_equal(self, setup_objs):
"""Calculate expected values for multiple test case, not equal operator"""
values, expected_data = [], []
search_obj = inflector.get_model(
setup_objs[0].searchable_type).query.get(
setup_objs[0].searchable_id)
for data in setup_objs:
if data.searchable_id != setup_objs[0].searchable_id:
expected_data.append(
create_tuple_data(data.obj_id, search_obj,
self.Meta.subprops))
for subprop in self.Meta.subprops:
values.append((expected_data, getattr(search_obj, subprop)))
return values
def expected_multiple_not_equal(self, setup_objs):
"""Calculate expected values for multiple test case, not equal operator"""
values, expected_data = [], []
search_obj = inflector.get_model(setup_objs[0].searchable_type).query.get(
setup_objs[0].searchable_id
)
for data in setup_objs:
if data.searchable_id != setup_objs[0].searchable_id:
expected_data.append(
create_tuple_data(data.obj_id, search_obj, self.Meta.subprops)
)
for subprop in self.Meta.subprops:
values.append(
(expected_data, getattr(search_obj, subprop))
)
return values
def expected_single_not_equal(self, setup_objs):
"""Calculate expected values for single test case, not equal operator"""
values = []
# Think that data type of all searchable objects is the same
model = inflector.get_model(setup_objs[0].searchable_type)
id_searchables = dict(
db.session.query(model.id, model).filter(
getattr(model, "id").in_(s.searchable_id for s in setup_objs)))
for data in setup_objs:
exp_data = [so for so in setup_objs if so.obj_id != data.obj_id]
search_obj = id_searchables[data.searchable_id]
for subprop in self.Meta.subprops:
values.append(([
create_tuple_data(ed.obj_id,
id_searchables[ed.searchable_id],
self.Meta.subprops) for ed in exp_data
], getattr(search_obj, subprop)))
return values
def expected_single_not_equal(self, setup_objs):
"""Calculate expected values for single test case, not equal operator"""
values = []
# Think that data type of all searchable objects is the same
model = inflector.get_model(setup_objs[0].searchable_type)
id_searchables = dict(
db.session.query(model.id, model).filter(getattr(model, "id").in_(
s.searchable_id for s in setup_objs
))
)
for data in setup_objs:
exp_data = [so for so in setup_objs if so.obj_id != data.obj_id]
search_obj = id_searchables[data.searchable_id]
for subprop in self.Meta.subprops:
values.append((
[create_tuple_data(
ed.obj_id, id_searchables[ed.searchable_id], self.Meta.subprops
) for ed in exp_data],
getattr(search_obj, subprop))
)
return values
def expected_multiple_is_empty(self, setup_objs):
"""Calculate expected values for multiple test case, is empty operator"""
# pylint: disable=no-self-use
expected_data = [create_tuple_data(row.obj_id, None, None)
for row in setup_objs[:MULTIPLE_ITEMS_COUNT]]
return [(expected_data, None)]
