Index: tst_pre_treatment_water_sample/test.py
===================================================================
diff -u -r6a95bfa0ed2eb2c3358631768e936cef69d648bc -r88021b78150355dab24a99207a98cf16801b1c1b
--- tst_pre_treatment_water_sample/test.py	(.../test.py)	(revision 6a95bfa0ed2eb2c3358631768e936cef69d648bc)
+++ tst_pre_treatment_water_sample/test.py	(.../test.py)	(revision 88021b78150355dab24a99207a98cf16801b1c1b)
@@ -24,24 +24,26 @@
     Method to navigate to sub mode under pre-treatment screen
     Navigating to Filter Flush screen
     """
+    test.startSection("Verify the Filter Flush screen")
     hd_simulator.cmd_set_hd_operation_mode_data(HDOpModes.MODE_PRET.value,PreTreatmentSampleWaterStates.SAMPLE_WATER_SETUP_STATE.value)
     hd_simulator.cmd_send_pre_treatment_state_data(sub_mode=HDOpSubModes.SUBMODE_WAIT_FOR_TREATMENT.value, water_sample_state=PreTreatmentSampleWaterStates.SAMPLE_WATER_SETUP_STATE.value,consumables_self_test_state=PreTreatmentConsumableSelfTestStates.CONSUMABLE_SELF_TESTS_INSTALL_STATE.value,
                                                     no_cartridge_self_test_state=PreTreatmentNoCartSelfTestsStates.NO_CART_SELF_TESTS_START_STATE.value,installation_state=PreTreatmentCartridgeInstallStates.CARTRIDGE_INSTALL_STATE.value, dry_self_test_state=PreTreatmentDrySelfTestsStates.DRY_SELF_TESTS_START_STATE.value,
                                                     prime_state=PreTreatmentPrimeStates.HD_PRIME_START_STATE.value,recirculate_state=PreTreatmentRecircStates.PRE_TREATMENT_RECIRC_STATE.value, patient_connection_state=PreTreatmentPatientConnectionStates.PATIENT_CONNECTION_STATE.value)
-    test.startSection("Verify the Filter Flush screen")
     utility.verify_page_step_indicator(names.o_preTreatmentWaterSampleStack_PreTreatmentBase_TreatmentFlowBase_2,PRE_TREATMENT_STEP)
-    utility.verify_countdown(config.FILTER_FLUSH_TITLE,hd_simulator, dg_simulator)
+    utility.verify_countdown(config.FILTER_FLUSH_TITLE, config.TIME_OUT_DURATION_100, hd_simulator, dg_simulator)
+    utility.verify_countdown(config.FILTER_FLUSH_TITLE, config.TIME_OUT_DURATION_200, hd_simulator, dg_simulator)
+    utility.verify_countdown(config.FILTER_FLUSH_TITLE, config.TIME_OUT_DURATION_300, hd_simulator, dg_simulator)
     test.endSection()
 
 def verify_sample_water_state_screens():
     """
     Method to verify the text and current state of the buttons and verify navigation within screen
     """
+    test.startSection("Verify the water sample main screen")
     hd_simulator.cmd_set_hd_operation_mode_data(HDOpModes.MODE_PRET.value,PreTreatmentSampleWaterStates.SAMPLE_WATER_SETUP_STATE.value)    
     hd_simulator.cmd_send_pre_treatment_state_data(sub_mode=HDOpSubModes.SUBMODE_WAIT_FOR_TREATMENT.value, water_sample_state=PreTreatmentSampleWaterStates.SAMPLE_WATER_STATE.value,consumables_self_test_state=PreTreatmentConsumableSelfTestStates.CONSUMABLE_SELF_TESTS_INSTALL_STATE.value,
                                                     no_cartridge_self_test_state=PreTreatmentNoCartSelfTestsStates.NO_CART_SELF_TESTS_START_STATE.value,installation_state=PreTreatmentCartridgeInstallStates.CARTRIDGE_INSTALL_STATE.value, dry_self_test_state=PreTreatmentDrySelfTestsStates.DRY_SELF_TESTS_START_STATE.value,
                                                     prime_state=PreTreatmentPrimeStates.HD_PRIME_START_STATE.value,recirculate_state=PreTreatmentRecircStates.PRE_TREATMENT_RECIRC_STATE.value, patient_connection_state=PreTreatmentPatientConnectionStates.PATIENT_CONNECTION_STATE.value)
-    test.startSection("Verify the water sample main screen")
     utility.verify_page_step_indicator(names.o_preTreatmentWaterSampleStack_PreTreatmentBase_TreatmentFlowBase_2,PRE_TREATMENT_STEP)
     test.compare(str(waitForObjectExists(names.o_water_sample_button).text), config.WATER_SAMPLE_BUTTON_TEXT,"Water Sample button text should be {}".format(config.WATER_SAMPLE_BUTTON_TEXT))
     test.compare(waitForObjectExists(names.o_water_sample_button).enabled , True, "Water Sample button should be enabled")
@@ -73,23 +75,20 @@
     test.compare(str(waitForObjectExists(names.o_fail_button).text), config.FAIL_TEXT,"Fail button text should be {}".format(config.FAIL_TEXT))
     test.compare(waitForObjectExists(names.o_fail_button).enabled , True, "Fail button should be enabled")
     mouseClick(waitForObject(names.o_fail_button))
-    
     utility.verify_page_step_indicator(names.o_preTreatmentWaterSampleStack_PreTreatmentBase_TreatmentFlowBase_2,PRE_TREATMENT_STEP)
     test.compare(str(waitForObjectExists(names.o_ok_button).text), config.OK_TEXT,"OK button text should be {}".format(config.OK_TEXT))
     test.compare(waitForObjectExists(names.o_ok_button).enabled , True, "OK button should be enabled")
     mouseClick(waitForObject(names.o_ok_button))
     test.compare(waitForObjectExists(names.o_stephead_indicator).enabled, True,"Verifying Water result failed indicator should be enabled")
+    hd_simulator.cmd_send_pre_treatment_water_sample_response(accepted=config.ACCEPTED, reason=0)
     test.endSection()
- 
 
+
 def main():
     utils.tstStart(__file__)
     startApplication(config.AUT_NAME)
     verify_pretreatment_filter_flush_screen()
     verify_sample_water_state_screens()
     utils.tstDone()
     
-    
-    
-    
     
\ No newline at end of file