Updated admin report download logic to include all quiz completions f…

…or the user bureaus when admin selects agency only (#741)

training/repositories/user.py

@@ -147,7 +147,11 @@ def get_admin_smartpay_training_report(self, filter: SmartPayTrainingReportFilte
         if filter.bureau_id is not None:
             query = query.filter(models.User.agency_id == filter.bureau_id)
         elif filter.agency_id is not None:
-            query = query.filter(models.User.agency_id == filter.agency_id)
+            # if agency is selected and not the bureau, return all records associated to agency/bureau
+            all_agencies = self._session.query(models.Agency).all()
+            selected_agency = [agency for agency in all_agencies if agency.id == filter.agency_id][0]
+            selected_agency_bureaus_ids = [agency.id for agency in all_agencies if agency.name == selected_agency.name]
+            query = query.filter(models.User.agency_id.in_(selected_agency_bureaus_ids))
 
         if filter.completion_date_start is not None:
             query = query.filter(models.QuizCompletion.submit_ts >= filter.completion_date_start)