###########################################################################
#
#  Copyright (c) 2019-2019 Diality Inc. - All Rights Reserved.
#
#  THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN
#  WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
#
#  @file hd_tests_script.py
#
#  @date 7-Apr-2019
#  @author P. Lucia
#
#  @brief This is an example test script for the HD.
#
############################################################################
import sys
sys.path.append("..")
from dialin.hd.constants import RESET, NO_RESET
from dialin.hd.hemodialysis_device import HD
from time import sleep
import unittest


class Test(unittest.TestCase):

    # @unittest.skip("Skipping test_hd_1.")
    def test_hd_1(self):
        hd = HD()
        sleep(2)

        if hd.basics.cmd_log_in_to_hd() == 0:
            exit(1)

        hd.bloodflow.cmd_blood_flow_broadcast_interval_override(RESET, 0)

        sleep(2)
        print("Blood Flow Target = {}".format(hd.bloodflow.target_blood_flow_rate))
        print("Blood Pump Current = {}".format(hd.bloodflow.measured_blood_pump_mc_current))
        sleep(5)
        print("Blood Pump Current = {}".format(hd.bloodflow.measured_blood_pump_mc_current))

        hd.bloodflow.cmd_blood_pump_measured_current_override(NO_RESET, 140)

        sleep(1)
        print("Blood Pump Current= {}".format(hd.bloodflow.measured_blood_pump_mc_current))
        sleep(5)
        hd.bloodflow.cmd_blood_pump_measured_current_override(RESET, 0)

        i = 0
        while True:
            sleep(0.5)
            print("Measured Flow = {} mL/min".format(hd.bloodflow.measured_blood_flow_rate))
            if i > 0 and i % 60 == 0:
                resp = input("Press 'Enter' to continue or 'q' to quit: ")
                if resp.lower() == "q":
                    break
            i += 1
        tgtRate = 0
        hd.bloodflow.cmd_blood_flow_broadcast_interval_override(NO_RESET, 2000)

        i = 0
        while True:
            if hd.bloodflow.target_blood_flow_rate == 0:
                if tgtRate != 0:
                    hd.bloodflow.cmd_blood_flow_broadcast_interval_override(NO_RESET, 2000)
                    tgtRate = 0
            else:
                if tgtRate == 0:
                    hd.bloodflow.cmd_blood_flow_broadcast_interval_override(NO_RESET, 200)
                    tgtRate = hd.bloodflow.target_blood_flow_rate
            if i > 0 and i % 60 == 0:
                resp = input("Press 'Enter' to continue or 'q' to quit: ")
                if resp.lower() == "q":
                    break
            i += 1

    #    hd.bloodflow.cmd_blood_flow_broadcast_interval_override(RESET,0)

        # TODO: Update passing criteria
        self.assertTrue(True)


if __name__ == '__main__':
    unittest.main(verbosity=2)