def test_disable_control_master_plan(self):
self.client.get('/app/OpenTestScenario/ScenarioCreator/tests/population_fixtures/Roundtrip.sqlite3/')
delete_all_outputs()
self.click_navbar_element("Controls")
parent_of(self.selenium.find_element_by_id("id_disable_all_controls")).click()
time.sleep(1)
elements = self.selenium.find_elements_by_css_selector("section > form > div")
for element in elements:
el_id = element.get_attribute("id")
if el_id == "div_id_name" or el_id == "div_id_disable_all_controls":
self.assertEqual(element.get_attribute("disabled"), None)
else:
self.assertEqual(element.get_attribute("disabled"), "true")
time.sleep(1)
setup_menu = self.find("#setupMenu")
hidden_menu_items = [
"Control Protocol",
"Protocol Assignments",
"Zone Effects",
]
for element in setup_menu.find_elements_by_tag_name("a"):
self.assertNotIn(element.text, hidden_menu_items)
def run_simulation(request):
delete_all_outputs()
delete_supplemental_folder()
create_blank_unit_stats() # create UnitStats before we risk the simulation writing to them
sim = Simulation(OutputSettings.objects.all().first().iterations)
sim.start() # starts a new thread
# time.sleep(5) # give initial threads time to initialize their db connection
return redirect('/results/')
def test_enable_control_master_plan(self):
"""
Check that re-enabling the disable all controls option
in control master plan re-enables the elements and menu
items
"""
self.setup_scenario()
delete_all_outputs()
ControlMasterPlan.objects.get_or_create(disable_all_controls=True)
self.click_navbar_element("Controls")
control_items = [
"Vaccination Triggers",
"Control Protocol",
"Assign Protocols",
"Zones",
"Zone Effects"
]
setup_menu = self.selenium.find_element_by_id("setupMenu")
actual_menu = [x.text for x in setup_menu.find_elements_by_tag_name("a")]
for controls in control_items:
self.assertNotIn(controls, actual_menu)
parent_of(self.selenium.find_element_by_id("id_disable_all_controls")).click()
time.sleep(1)
elements = self.selenium.find_elements_by_css_selector("section > form > div")
for element in elements:
self.assertEqual(element.get_attribute("disabled"), None)
setup_menu = self.selenium.find_element_by_id("setupMenu")
actual_menu = [x.text for x in setup_menu.find_elements_by_tag_name("a")]
for controls in control_items:
self.assertIn(controls, actual_menu)
