WIP: Retrieve the groupsummary with the dates

Author: ogiorgis

tests/conftest.py

@@ -970,7 +970,7 @@ def mock_get_summary_groups_request(monkeypatch):
     Mock get_summary_groups()
     """
 
-    def _get_summary_groups():
+    def _get_summary_groups(start_day, end_day):
         tests_folder = os.path.dirname(__file__)
         file_name = "summary_groups.json"
         data_path = os.path.join(tests_folder, "sample_data", file_name)

treeherder/intermittents_commenter/commenter.py

@@ -80,7 +80,7 @@ def generate_bug_changes(self, startday, endday, alt_startday, alt_endday):
 
         bug_ids, bugs = self.get_bugs(startday, endday)
         option_collection_map = OptionCollection.objects.get_option_collection_map()
-        bug_map = self.build_bug_map(bugs, option_collection_map)
+        bug_map = self.build_bug_map(bugs, option_collection_map, startday, endday)
 
         alt_date_bug_totals = self.get_alt_date_bug_totals(alt_startday, alt_endday, bug_ids)
 
@@ -387,10 +387,12 @@ def get_bug_run_info(self, bug):
             info.build_type = "unknown build"
         return info
 
-    def get_task_labels_and_count(self, manifest):
+    def get_task_labels_and_count(self, manifest, start_day, end_day):
         tasks_and_count = {}
         summary_groups = (
-            fetch.get_summary_groups() if self.summary_groups is None else self.summary_groups
+            fetch.get_summary_groups(start_day, end_day)
+            if self.summary_groups is None
+            else self.summary_groups
         )
         all_task_labels = summary_groups["job_type_names"]
         for tasks_by_manifest in summary_groups["manifests"]:
@@ -402,7 +404,7 @@ def get_task_labels_and_count(self, manifest):
                         tasks_and_count[task_label] += count
         return tasks_and_count
 
-    def build_bug_map(self, bugs, option_collection_map):
+    def build_bug_map(self, bugs, option_collection_map, start_day, end_day):
         """Build bug_map
          eg:
         {
@@ -434,7 +436,7 @@ def build_bug_map(self, bugs, option_collection_map):
         for bug in bugs:
             bug_id = bug["bug__bugzilla_id"]
             manifest = self.get_test_manifest(bug["bug__summary"])
-            task_labels = self.get_task_labels_and_count(manifest)
+            task_labels = self.get_task_labels_and_count(manifest, start_day, end_day)
             print(task_labels)
             bug_testrun_matrix = []
             if manifest:

treeherder/intermittents_commenter/fetch.py

@@ -24,7 +24,7 @@ def fetch_testrun_matrix():
     return response.json()
 
 
-def get_summary_groups():
+def get_summary_groups(start_day, end_day):
     url = "https://treeherder.mozilla.org/api/groupsummary/"
     response = requests.get(url, headers={"User-agent": "mach-test-info/1.0"})
     return response.json()