Index: tst_main_treatment_ultrafiltration/test.py
===================================================================
diff -u -r6008f9330e1369fc6389162dabaf752b4ca704f8 -r6dea78f296538eaec9a3478608f5ba0be9e9c954
--- tst_main_treatment_ultrafiltration/test.py	(.../test.py)	(revision 6008f9330e1369fc6389162dabaf752b4ca704f8)
+++ tst_main_treatment_ultrafiltration/test.py	(.../test.py)	(revision 6dea78f296538eaec9a3478608f5ba0be9e9c954)
@@ -20,7 +20,7 @@
 from dialin.ui import utils
 from dialin.common.ui_defs import TXStates as uf_states
 from dialin.ui.hd_simulator import HDSimulator
-from dialin.common.hd_defs import HDStandbyStates, TreatmentStates
+from dialin.common.hd_defs import HDStandbyStates, TreatmentStates, BloodLeakZeroingStates
 from configuration import config
 
 hd_simulator    = HDSimulator()
@@ -76,7 +76,7 @@
     hd_simulator.cmd_set_treatment_parameter_ranges(MIN_TREATMENT_DURATION, MAX_TREATMENT_DURATION, min_uf, max_uf, MIN_DIALYSATE_FLOW_RATE, MAX_DIALYSATE_FLOW_RATE)
     
     for uf_value in range(min_uf, (max_uf)+1, 1):
-        hd_simulator.cmd_set_treatment_ultrafiltration_outlet_flow_data(uf_value, uf_value, ROT_SPEED, MOT_SPEED, MC_SPEED, MC_CURRENT, PWM, 0,0,0,0,0)
+        hd_simulator.cmd_set_treatment_ultrafiltration_outlet_flow_data(uf_value, uf_value, ROT_SPEED, MOT_SPEED, MC_SPEED, MC_CURRENT, PWM, 0,0,0,0,0,0)
         utils.waitForGUI(0.2)
         test.compare(utils.l2ml(waitForObjectExists(names.o_treatmentStart_ultrafiltrationTouchArea_TreatmentUltrafiltration).value), uf_value, "uf set value should be :" + str(uf_value))
         hd_simulator.cmd_send_uf_treatment_response(accepted = True, reason = 0, volume = 500)
@@ -93,7 +93,7 @@
     @param : N/A
     @return: N/A
     """
-    hd_simulator.cmd_set_treatment_ultrafiltration_outlet_flow_data(REF_UF_VOL, MEASURED_UF_VOL, ROT_SPEED, MOT_SPEED, MC_SPEED, MC_CURRENT, PWM, 0,0,0,0,0) 
+    hd_simulator.cmd_set_treatment_ultrafiltration_outlet_flow_data(REF_UF_VOL, MEASURED_UF_VOL, ROT_SPEED, MOT_SPEED, MC_SPEED, MC_CURRENT, PWM, 0,0,0,0,0,0) 
     test.compare(waitForObjectExists(names.o_treatmentStart_ultrafiltrationTouchArea_TreatmentUltrafiltration).value, config.UF_MINIMUM_VALUE, "uf treatment reset to 0")
 
 
@@ -113,7 +113,7 @@
     # Integration team needs to determine if the new UF parameters to the simulator need to be exercised for verification.
     hd_simulator.cmd_set_treatment_ultrafiltration_outlet_flow_data(ultrafiltration_flow_parameter[REF_UF_VOL_INDEX], ultrafiltration_flow_parameter[MEASURED_UF_VOL_INDEX],
                                                            ultrafiltration_flow_parameter[ROT_SPEED_INDEX], ultrafiltration_flow_parameter[MOT_SPEED_INDEX], ultrafiltration_flow_parameter[MC_SPEED_INDEX],
-                                                           ultrafiltration_flow_parameter[MC_CURRENT_INDEX], ultrafiltration_flow_parameter[PWM_INDEX],0,0,0,0,0) 
+                                                           ultrafiltration_flow_parameter[MC_CURRENT_INDEX], ultrafiltration_flow_parameter[PWM_INDEX],0,0,0,0,0,0) 
     test.log("test to verify ultrafiltration flow parameters.")
     
 
@@ -302,9 +302,17 @@
     startApplication(config.AUT_NAME)
     
     #navigate to dialysis state
-    hd_simulator.cmd_set_treatment_states_data(sub_mode=HDStandbyStates.STANDBY_WAIT_FOR_DISINFECT_STATE.value, uf_state=TreatmentStates.TREATMENT_START_STATE.value, saline_state=TreatmentStates.TREATMENT_START_STATE.value, heparin_state=TreatmentStates.TREATMENT_START_STATE.value,
-                                      rinseback_state=TreatmentStates.TREATMENT_START_STATE.value, recirculate_state=TreatmentStates.TREATMENT_START_STATE.value, blood_prime_state=TreatmentStates.TREATMENT_START_STATE.value,
-                                      treatment_end_state=TreatmentStates.TREATMENT_START_STATE.value, treatment_stop_state=TreatmentStates.TREATMENT_START_STATE.value, dialysis_state=TreatmentStates.TREATMENT_START_STATE.value) 
+    hd_simulator.cmd_set_treatment_states_data(sub_mode=HDStandbyStates.STANDBY_WAIT_FOR_DISINFECT_STATE.value, 
+                                               uf_state=TreatmentStates.TREATMENT_START_STATE.value, 
+                                               saline_state=TreatmentStates.TREATMENT_START_STATE.value, 
+                                               heparin_state=TreatmentStates.TREATMENT_START_STATE.value,
+                                               rinseback_state=TreatmentStates.TREATMENT_START_STATE.value, 
+                                               recirculate_state=TreatmentStates.TREATMENT_START_STATE.value, 
+                                               blood_prime_state=TreatmentStates.TREATMENT_START_STATE.value,
+                                               treatment_end_state=TreatmentStates.TREATMENT_START_STATE.value, 
+                                               treatment_stop_state=TreatmentStates.TREATMENT_START_STATE.value, 
+                                               dialysis_state=TreatmentStates.TREATMENT_START_STATE.value,
+                                               bloodLeakZeoringState = BloodLeakZeroingStates.BLD_ZEROING_IDLE_STATE.value) 
     
     verification_of_uf_from_main_treatment()