Merge pull request #115 from unicef/feature/test-export-generation

Add tests for export generation of households and individuals

Author: patsatsia

tests/functional/test_f_household.py

@@ -1,11 +1,14 @@
 from typing import TYPE_CHECKING
+from unittest.mock import patch
 
 import pytest
 from django.core.management import call_command
 from testutils.factories import OfficeFactory
 from testutils.factories.program import BeneficiaryGroupFactory
 from testutils.factories import CountryProgramFactory
 
+from tests.extras.testutils.selenium import CountryWorkspaceSeleniumTC
+
 if TYPE_CHECKING:
     from country_workspace.workspaces.models import CountryHousehold
 
@@ -79,3 +82,40 @@ def test_list_household_select_all_fields(browser, household: "CountryHousehold"
     checkboxes = browser.find_elements('input[type="checkbox"][name="fields"]')
     for cb in checkboxes:
         assert cb.is_selected()
+
+
+def _test_export_generation(browser: CountryWorkspaceSeleniumTC, household: "CountryHousehold", link: str):
+    browser.login_as_user()
+    browser.select_option_by_text("select[name=tenant]", household.program.country_office.name)
+    browser.select2_select("id_program", household.program.name)
+    browser.click_link(link)
+
+    browser.click("#action-toggle")
+    browser.select_option_by_text("select[name=action]", "Export records as .xlsx for bulk updates")
+
+    browser.click("button[name='index'][value='0']")
+    browser.click("#select-all")
+    browser.click("input[name='_export']")
+
+    browser.click('//a[div[text()="Async Jobs"]]')
+    browser.wait_for_element("table#result_list", timeout=10)
+    rows = browser.find_elements("table#result_list tbody tr")
+    assert any("Export records as .xlsx for bulk updates" in row.text for row in rows)
+
+
+@pytest.mark.selenium
+@patch("country_workspace.workspaces.admin.cleaners.bulk_update.bulk_update_export_template")
+def test_households_export_generation(
+    mocked_exporter, browser: CountryWorkspaceSeleniumTC, household: "CountryHousehold"
+):
+    mocked_exporter.return_value = None
+    _test_export_generation(browser=browser, household=household, link="Households")
+
+
+@pytest.mark.selenium
+@patch("country_workspace.workspaces.admin.cleaners.bulk_update.bulk_update_export_template")
+def test_individuals_export_generation(
+    mocked_exporter, browser: CountryWorkspaceSeleniumTC, household: "CountryHousehold"
+):
+    mocked_exporter.return_value = None
+    _test_export_generation(browser=browser, household=household, link="Individuals")