# -*- coding: utf-8 -*-"

#
# file    tst__main_treatment_heparin
# date    2022/03/03
# author  Akshay Rajaram Dhawan
# author  Shweta Policepatil
# NOTE:
# This test contradicts verification of Heparin Delivery data and Heparin States from Main Treatment 

import names
from configuration import config, utility
from dialin.ui import utils
from dialin.ui.hd_simulator import HDSimulator


HIGHER_RANGE = 20                    
STEP_VALUE = 1
LOWER_RANGE = 0

hd_simulator = HDSimulator()

def verification_of_cumulative_value(expected_cumulative_value):
    """
    Verifying cumulative values from heparin delivery section.
    @param expected_cumulative_value - (str) expected cumulative value.           
    """ 
    expected_cumulative_value = str(expected_cumulative_value).replace(".0","")
    heparin_value = waitForObjectExists(utility.expected_heparin_value(expected_cumulative_value))
    test.compare(heparin_value.text, expected_cumulative_value, "Actual heparin cumulative value: {} should be equal to expected heparin cumulative value: {}".format(heparin_value.text, expected_cumulative_value))                                                    
      
def constants_for_cumulative_value():
    """
    Method is used to verify the constants for cumulative value.
    """ 
    test.startSection("Verifying Heparin State 3 HEPARIN_INITIAL_BOLUS values")
    hd_simulator.cmd_set_treatment_states_data(sub_mode= 2, uf_state= 0, saline_state= 0, heparin_state= 3,
                                      rinseback_state= 0, recirculate_state= 0, blood_prime_state= 0,
                                      treatment_end_state= 0, treatment_stop_state=0, dialysis_state= 0)
    test.compare(str(waitForObjectExists(names.o_state0_heparin_delivery_text).text), config.HEPARIN_DELIVERY_TEXT, "Heparin Delivery Text should be {}".format(config.HEPARIN_DELIVERY_TEXT))
    test.compare((waitForObjectExists(names.o_state0_heparin_delivery_text)).enabled, False, "Heparin Delivery should be disabled")
    initial_bolus_text = waitForObjectExists(names.o_state3_heparin_state_initial_bolus_text).text
    test.compare(str(initial_bolus_text), config.HEPARIN_BOLUS_ACTIVE_TEXT, "Notification Message should be {}".format(config.HEPARIN_BOLUS_ACTIVE_TEXT))
    
    HIGHER_VALUE = HIGHER_RANGE * 10    #conversion of decaliter to liter
    for heparin_cumulative in range(LOWER_RANGE, HIGHER_VALUE+1, STEP_VALUE):
        heparin_cumulative = format(heparin_cumulative/10, ".1f")
        test.log(str(heparin_cumulative))
        heparin_cumulative = float(heparin_cumulative)
        hd_simulator.cmd_set_treatment_heparin_data(heparin_cumulative)  
        verification_of_cumulative_value(heparin_cumulative)  
        utils.waitForGUI(1)
    test.endSection()
      
def heparin_stop_states_in_main_treatment():
    """
    Verifying the heparin states when heparin cumulative is '0' at 
    HEPARIN STATE 0 -> HEPARIN_STATE_OFF
    HEPARIN STATE 1 -> HEPARIN_STATE_STOPPED
    HEPARIN STATE 5 -> HEPARIN_STATE_COMPLETED
    HEPARIN STATE 6 -> HEPARIN_STATE_EMPTY
    """
    test.log("Verifying Heparin State 0 - HEPARIN_STATE_OFF")
    hd_simulator.cmd_set_treatment_states_data(sub_mode= 2, uf_state= 0, saline_state= 0, heparin_state= 0,
                                               rinseback_state= 0, recirculate_state= 0, blood_prime_state= 0,
                                               treatment_end_state= 0, treatment_stop_state= 0, dialysis_state= 0)
    test.compare(str(waitForObjectExists(names.o_state0_heparin_delivery_text).text), config.HEPARIN_DELIVERY_TEXT, "Heparin Delivery Text should be {}".format(config.HEPARIN_DELIVERY_TEXT))
    test.compare((waitForObjectExists(names.o_state0_heparin_delivery_text)).enabled, False, "Heparin Delivery should be disabled")
    test.log("Verifying Notification Message")
    test.compare(str(findObject(names.o_state0_heparin_delivery_Off_text).text ), config.HEPARIN_DELIVERY_OFF_TEXT,"Notification Message should be {}".format(config.HEPARIN_DELIVERY_OFF_TEXT))
    utils.waitForGUI(1)
    
    test.log("Verifying Heparin State 1 - HEPARIN_STATE_STOPPED")
    hd_simulator.cmd_set_treatment_states_data(sub_mode= 2, uf_state= 0, saline_state=0, heparin_state= 1,
                                     rinseback_state= 0, recirculate_state= 0, blood_prime_state= 0,
                                      treatment_end_state=0, treatment_stop_state= 0, dialysis_state= 0) 
    test.compare(str(waitForObjectExists(names.o_state0_heparin_delivery_text).text), config.HEPARIN_DELIVERY_TEXT, "Heparin Delivery Text should be {}".format(config.HEPARIN_DELIVERY_TEXT))
    test.compare((waitForObjectExists(names.o_state0_heparin_delivery_text)).enabled, False, "Heparin Delivery should be disabled")
    test.log("Verifying Notification Message")
    test.compare(str(waitForObjectExists(names.o_state_one_text).text), config.HEPARIN_STOP_TEXT,"Notification Message should be {}".format(config.HEPARIN_STOP_TEXT))   
    utils.waitForGUI(1)
    
    constants_for_cumulative_value()
    heparin_pause_resume_states()
    
    test.log("Verifying Heparin State 5 - HEPARIN_STATE_COMPLETED")
    hd_simulator.cmd_set_treatment_states_data(sub_mode= 2, uf_state= 0, saline_state=0, heparin_state= 5,
                                     rinseback_state= 0, recirculate_state= 0, blood_prime_state= 0,
                                     treatment_end_state=0, treatment_stop_state= 0, dialysis_state= 0) 
    
    test.compare(str(waitForObjectExists(names.o_state0_heparin_delivery_text).text), config.HEPARIN_DELIVERY_TEXT, "Heparin Delivery Text should be {}".format(config.HEPARIN_DELIVERY_TEXT))
    test.compare((waitForObjectExists(names.o_state0_heparin_delivery_text)).enabled, False, "Heparin Delivery should be disabled")
    test.log("Verifying Notification Message")
    test.compare(str(waitForObjectExists(names.o_state5_heparin_state_completed_text).text), config.HEPARIN_MAXIMUM_VOLUME_DELIVERED_TEXT,"Notification Message should be {}".format(config.HEPARIN_MAXIMUM_VOLUME_DELIVERED_TEXT))   
    utils.waitForGUI(1)
    
    test.log("Verifying Heparin State 6 - HEPARIN_STATE_EMPTY") 
    hd_simulator.cmd_set_treatment_states_data(sub_mode= 2, uf_state= 0, saline_state=0, heparin_state= 6,
                                     rinseback_state= 0, recirculate_state= 0, blood_prime_state= 0,
                                     treatment_end_state=0, treatment_stop_state= 0, dialysis_state= 0) 
   
    test.compare(str(waitForObjectExists(names.o_state0_heparin_delivery_text).text), config.HEPARIN_DELIVERY_TEXT, "Heparin Delivery Text should be {}".format(config.HEPARIN_DELIVERY_TEXT))
    test.compare((waitForObjectExists(names.o_state0_heparin_delivery_text)).enabled, False, "Heparin Delivery should be disabled")
    test.log("Verifying Notification Message")
    test.compare(str(waitForObjectExists(names.o_state6_heparin_state_empty_text).text), config.HEPARIN_SYRINGE_EMPTY_TEXT,"Notification Message should be {}".format(config.HEPARIN_SYRINGE_EMPTY_TEXT))   
    utils.waitForGUI(1)
    
def heparin_pause_resume_states():
    """
        Verifying pause/resume delivery when HEPARIN DELIVERY button is active or deactive using heparin states
        HEPARIN STATE 2 -> HEPARIN_STATE_PAUSED
        HEPARIN STATE 4 -> HEPARIN_STATE_DISPENSING
    """
    hd_simulator.cmd_set_heparin_pause_resume_response(accepted = 1, reason = 1) 
    hd_simulator.cmd_set_treatment_heparin_data(cumulative = 10)
    hd_simulator.cmd_set_treatment_states_data(sub_mode= 2, uf_state= 0, saline_state=0, heparin_state= 2,
                                    rinseback_state= 0, recirculate_state= 0, blood_prime_state= 0,
                                    treatment_end_state=0, treatment_stop_state= 0, dialysis_state= 0)
    test.compare(str(waitForObjectExists(names.o_state2_heparin_state_paused_text).text),config.HEPARIN_RESUME_DELIVERY_TEXT,"Heparin Resume Delivery Text should be {}".format(config.HEPARIN_RESUME_DELIVERY_TEXT))
    test.compare( waitForObjectExists(names.o_state2_heparin_state_paused_text).enabled, True, "Resume Delivery Button is enabled")
    
    test.log("Heparin delivery is paused, resume delivery to start")
    utils.waitForGUI(1)
    hd_simulator.cmd_set_treatment_heparin_data(cumulative = 10.1)
    hd_simulator.cmd_set_treatment_states_data(sub_mode= 2, uf_state= 0, saline_state=0, heparin_state= 4,
                                    rinseback_state= 0, recirculate_state= 0, blood_prime_state= 0,
                                    treatment_end_state=0, treatment_stop_state= 0, dialysis_state= 0) 
    test.compare(str(waitForObjectExists(names.o_state4_heparin_state_dispensing_text).text), config.HEPARIN_PAUSE_DELIVERY_TEXT,"Heparin Pause Delivery Text should be {}".format(config.HEPARIN_PAUSE_DELIVERY_TEXT))
    test.compare( waitForObjectExists(names.o_state4_heparin_state_dispensing_text).enabled , True, "Pause Delivery Button is enabled")
    test.log("Heparin delivery is in dispensing state")
    utils.waitForGUI(1)
    
def verify_the_rejection_msg():
    """
    Method to verify rejection messages 
    
    """ 
    test.startSection("Verifying Rejection Messages")
    hd_simulator.cmd_set_treatment_states_data(sub_mode= 2, uf_state= 0, saline_state=0, heparin_state= 4,
                                    rinseback_state= 0, recirculate_state= 0, blood_prime_state= 0,
                                    treatment_end_state=0, treatment_stop_state= 0, dialysis_state= 0) 
    for reason_id in range(1, config.NUM_OF_REQUEST_REJECT_REASONS+1):
        hd_simulator.cmd_set_heparin_pause_resume_response(accepted=0, reason=reason_id) 
        rejection_msg = waitForObjectExists(utility.rejection_msg(text=config.REJECTION_REASON[reason_id]))
        test.compare(rejection_msg.text, config.REJECTION_REASON[reason_id], "The rejection message should be {}".format(config.REJECTION_REASON[reason_id]))
    test.endSection()

def main():
    
    utils.tstStart(__file__)            
    startApplication(config.AUT_NAME)
    heparin_stop_states_in_main_treatment()
    verify_the_rejection_msg()
    
    utils.tstDone()