def test_get_projects(self):
project = DiscoverProcessor.get_projects(
organization_id=self.org.id, query={"project": [self.project1.id]})
assert isinstance(project, list)
assert project[0] == self.project1
projects = DiscoverProcessor.get_projects(
organization_id=self.org.id,
query={"project": [self.project1.id, self.project2.id]})
assert sorted(p.id for p in projects) == sorted(
[self.project1.id, self.project2.id])
with self.assertRaises(ExportError):
DiscoverProcessor.get_projects(organization_id=self.org.id,
query={"project": [-1]})
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