def test_handle_fields(self):
processor = DiscoverProcessor(organization_id=self.org.id,
discover_query=self.discover_query)
assert processor.header_fields == ["count_id", "fake_field", "issue"]
result_list = [{"issue": self.group.id, "issue.id": self.group.id}]
new_result_list = processor.handle_fields(result_list)
assert new_result_list[0] != result_list
assert new_result_list[0]["issue"] == self.group.qualified_short_id
def test_handle_equations(self):
self.discover_query["field"] = ["count(id)", "fake(field)"]
self.discover_query["equations"] = [
"count(id) / fake(field)", "count(id) / 2"
]
processor = DiscoverProcessor(organization_id=self.org.id,
discover_query=self.discover_query)
assert processor.header_fields == [
"count_id",
"fake_field",
"count(id) / fake(field)",
"count(id) / 2",
]
result_list = [{"equation[0]": 5, "equation[1]": 8}]
new_result_list = processor.handle_fields(result_list)
assert new_result_list[0] != result_list
assert new_result_list[0]["count(id) / fake(field)"] == 5
assert new_result_list[0]["count(id) / 2"] == 8