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)