def test_register(self):
        sql_manager.register('Dinko', '123123')
        hashed_password = sql_manager.hash_password('123123')
        sql_manager.cursor.execute('SELECT Count(*)  FROM clients WHERE username = (?) AND password = (?)', ('Dinko', hashed_password))
        users_count = sql_manager.cursor.fetchone()

        self.assertEqual(users_count[0], 1)

    def start(self):
        print(messages.WELCOME_MSG)
        while True:
            command = input("$$$>")

            if command == 'register':
                username = input("Enter your username: ")
                password = input("Enter your password: ")
                try:
                    sql_manager.register(username, password)
                    print(messages.SUCCESSFUL_REG_MSG)
                except PasswordNotStrongError:
                    print(messages.STRONG_PASSWORD_ERROR_MSG)

            elif command == 'login':
                username = input("Enter your username: ")
                password = getpass.getpass("Enter your password: ")
                logged_user = sql_manager.login(username, password)

                if logged_user:
                    self.logged_menu(logged_user)
                else:
                    print(messages.LOGIN_FAIL_MSG)

            elif command == 'help':
                print("login - for logging in!")
                print("register - for creating new account!")
                print("exit - for closing program!")

            elif command == 'exit':
                break
            else:
                print("Not a valid command")

 def setUp(self):
     sql_manager.create_clients_table()
     sql_manager.register('Tester', '[email protected]')
     sql_manager.register('Tester_1', '1234Tester_1')
     sql_manager.register('Tester_3', '12345678Qw')
     sql_manager.register('Tester_2', '[email protected]')
     sql_manager.register('Tester_4', '[email protected]')

    def test_register(self):
        sql_manager.register('Dinko', STRONG_PASSWORD2)

        sql_manager.cursor.execute('SELECT Count(*)  FROM clients WHERE username = (?)', ('Dinko',))
        users_count = sql_manager.cursor.fetchone()

        self.assertEqual(users_count[0], 1)

def main_menu():
    print("Welcome to our bank service. You are not logged in. \nPlease register or login")

    while True:
        command = input("$$$>")

        if command == 'register':
            username = input("Enter your username: ")
            password = valid_password()

            sql_manager.register(username, password)

            print("Registration Successfull")

        elif command == 'login':
            username = input("Enter your username: ")
            password = input("Enter your password: ")

            logged_user = sql_manager.login(username, password)

            if logged_user:
                logged_menu(logged_user)
            else:
                print("Login failed")

        elif command == 'help':
            print("login - for logging in!")
            print("register - for creating new account!")
            print("exit - for closing program!")

        elif command == 'exit':
            break
        else:
            print("Not a valid command")

 def test_register(self):
     sql_manager.register('Dinko', '123123', "[email protected]")
     sql_manager.cursor.execute('''SELECT Count(*)  FROM clients
                                 WHERE username = (?)
                                 AND password = (?)''', ('Dinko', '123123'))
     users_count = sql_manager.cursor.fetchone()
     self.assertEqual(users_count[0], 1)

    def test_register(self):
        sql_manager.register('Dinko', 'asdaFGG45g*&')

        sql_manager.cursor.execute('SELECT Count(*)  FROM clients WHERE username = (?) AND password = (?)', ('Dinko', 'asdaFGG45g*&'))
        users_count = sql_manager.cursor.fetchone()

        self.assertEqual(users_count[0], 1)

    def test_change_password_with_sql_injection(self):
        sql_manager.register('Dinko', STRONG_PASSWORD1)
        sql_manager.register('Vladko', STRONG_PASSWORD2)

        logged_user = sql_manager.login('Dinko', STRONG_PASSWORD1)
        new_password = "1234Asdf$$$Asdf' WHERE id = 3 --"
        sql_manager.change_pass(new_password, logged_user)
        self.assertFalse(sql_manager.login('Vladko', "1234Asdf$$$Asdf"))

    def test_register(self):
        sql_manager.register('Dinko', '$$$$Dim_4o')

        sql_manager.cursor.execute(
            '''SELECT Count(*)  FROM clients WHERE username = ? AND password = ?''', ('Dinko', '$$$$Dim_4o'))
        users_count = sql_manager.cursor.fetchone()

        self.assertEqual(users_count[0], 1)

    def test_register_with_username_not_including_upper_lower_digit_special(self):
        sql_manager.register('Dinko', 'palachinkooooo!!!')

        sql_manager.cursor.execute(
            '''SELECT Count(*)  FROM clients WHERE username = ? AND password = ?''', ('Dinko', 'Dinko_23Lopatata$!!!'))
        users_count = sql_manager.cursor.fetchone()

        self.assertEqual(users_count[0], 0)

def register():
    username = input("Enter your username: ")
    password = input("Enter your password: ")
    try:
        sql_manager.register(username, password)
        return "Registration siccessfull"
    except StrongPasswordException as e:
        print(e)

def main_menu():
    failed_log_users = [1]
    fail_log_counter = 0
    wrong_pass_limit = 5
    block_message = "you have 5 wrong password inputs\
                    \n and you`re blocked for 5 minutes"

    print("Welcome to our bank service. You are not logged in.\
          \nPlease register or login")

    while True:
        command = input("$$$>")
        if command == 'register':
            username = input("Enter your username: ")
            password = pass_conditions(username)
            hashed_pass = hash_password(password)
            email = input("Enter your emal: ")
            sql_manager.register(username, hashed_pass, email)

            print("Registration Successfull")

        elif command == 'login':
            username = input("Enter your username: ")
            password = getpass.getpass('Password:')
            input_pass = hash_password(password)

            logged_user = sql_manager.login(username)
            existing_pass = sql_manager.get_user_pass(username)
            is_pass_eq = sql_manager.check_password(input_pass, existing_pass) is True
            if logged_user and is_pass_eq and can_login(username) is True:
                fail_log_counter = 0
                sql_manager.unbclock_user(username)
                logged_menu(logged_user)
                failed_log_users = [1]
            else:
                print("Login failed")
                last_user = failed_log_users[len(failed_log_users) - 1]
                if fail_log_counter >= wrong_pass_limit:
                    print(block_message)
                if username == last_user:
                    fail_log_counter += 1
                    if fail_log_counter == wrong_pass_limit:
                        sql_manager.block_user(username)
                        fail_log_counter = 0
                        print(block_message)
                else:
                    fail_log_counter = 1
                failed_log_users.append(username)

        elif command == 'help':
            print("login - for logging in!")
            print("register - for creating new account!")
            print("exit - for closing program!")

        elif command == 'exit':
            break
        else:
            print("Not a valid command")

    def test_hash(self):
        username = 'tdhris'
        password = 'blahblahblah'
        sql_manager.register(username, password)

        sql_manager.cursor.execute("SELECT Count(*) FROM clients\
            WHERE username = ? AND password = ?", (username, password))

        self.assertEqual(0, sql_manager.cursor.fetchone()[0])

def main_menu():
    print("Welcome to our bank service. You are not logged in. \nPlease register or login")

    while True:
        command = input("$$$>").split(" ")

        if command[0] == 'register':
            username = input("Enter your username: ")
            password = getpass.getpass("Enter your password: ")
            while not sql_manager.is_strong(username, password):
                print("Password is not strong enough!")
                password = getpass.getpass("Enter your password: ")
            email = input("Enter your email: ")
            sql_manager.register(username, password, email)

            print("Registration Successfull")

        elif command[0] == 'login':
            username = input("Enter your username: ")
            password = getpass.getpass("Enter your password: ")

            logged_user = sql_manager.login(username, password)

            if logged_user:
                logged_menu(logged_user)
            else:
                print("Login failed")

        elif command[0] == "reset-password" and len(command) >1:
            reset_hash = hashlib.sha1(str(random.random()).encode()).hexdigest()
            sql_manager.update_reset_hash(command[1], reset_hash)
            send_email.send_email(sql_manager.get_email(command[1]), reset_hash)
            print(command[1])


        elif command[0] == "send-reset-password" and len(command) > 1:
            reset_hash = input("Enter reset hash: ")
            if reset_hash == sql_manager.get_reset_hash(command[1]):
                logged_user = sql_manager.login_with_hash(command[1], reset_hash)
                if logged_user:
                    logged_menu(logged_user)
                else:
                    print("Login failed")


        elif command[0] == 'help':
            print("login - for logging in!")
            print("register - for creating new account!")
            print("exit - for closing program!")
            print("reset-password <name> - to reset password")
            print("send-reset-password <name> - to enter reset password")

        elif command[0] == 'exit':
            break
        else:
            print("Not a valid command")

    def test_register(self):
        sql_manager.register('Dinko', '123123', '[email protected]',)

        sql_manager.cursor.execute(
            'SELECT Count(*)  FROM clients WHERE username = (?) AND password = (?)',
            ('Dinko', '123123')
        )
        users_count = sql_manager.cursor.fetchone()

        self.assertEqual(users_count[0], 1)

def register():
    username = input("Enter your username: ")
    password = getpass.getpass('Enter your password:')

    valid_password = ValidateRegistration(username, password)
    is_valid = valid_password.is_valid()

    if is_valid is True:
        sql_manager.register(username, password)
        print("Registration Successfull")
    else:
        print(is_valid)

    def test_register(self):
        username = 'Dinko'
        password = '123123$$GGhhhh'
        hashed_pass = sql_manager.hash_function(password)
        sql_manager.register(username, password)

        sql_manager.cursor.execute('SELECT Count(*)  FROM clients\
            WHERE username = (?) AND password = (?)', (username,
                                                       hashed_pass))
        users_count = sql_manager.cursor.fetchone()

        self.assertEqual(users_count[0], 1)

def main_menu():
    print("Welcome to our bank service. You are not logged in. \nPlease register or login")

    while True:
        command = input("$$$>")
        command_array = command.split(" ")
        if command == 'register':
            username = input("Enter your username: ")
            email = input('Enter your email: ')
            while True:
                password = getpass.getpass()
                if len(password) < 8:
                    print("Your Pasword is to short, it must be 8 symbols")
                elif not have_upper_letters(password):
                    print("Your password must have upper letters")
                elif not number_function(password):
                    print("Your password must have numbers")
                elif not have_special(password):
                    print("It must have special symbols")
                elif username in password:
                    print("Don`t use your name in your password")
                else:
                    sql_manager.register(username, sql_manager.get_crypted(password), email)
                    break
            print("Registration Successfull")

        elif command == 'login':
            username = input("Enter your username: ")

            logged_user = sql_manager.login(username, sql_manager.wrong_password_count(username))

            if logged_user:
                logged_menu(logged_user)
            else:
                print("Login failed")

        elif command == 'send-reset-password':
            username = input("Enter your username: ")
            print(sql_manager.send_reset_password(username))

        elif command_array[0] == 'reset-password':
            print(sql_manager.reset_password(command_array[1]))

        elif command == 'help':
            print("login - for logging in!")
            print("register - for creating new account!")
            print("exit - for closing program!")

        elif command == 'exit':
            break
        else:
            print("Not a valid command")

    def test_register(self):
        sql_manager.register('Dinko', '123aB#23123', '[email protected]')
        m = hashlib.sha1()
        passw = '123aB#23123'
        m.update(passw.encode('utf-8'))
        passw = m.hexdigest()

        sql_manager.cursor.execute('''SELECT Count(*)  FROM clients
                                      WHERE username = (?)
                                      AND password = (?)''', ('Dinko', passw))
        users_count = sql_manager.cursor.fetchone()

        self.assertEqual(users_count[0], 1)

def main_menu():
    print("Welcome to our bank service. You are not logged in.")
    print("Please register or login")

    while True:
        command = input("$$$>")

        if command == "register":
            username = input("Enter your username: ")
            password = getpass.getpass("Enter your password: ")
            email = input("Enter your email: ")

            sql_manager.register(username, password, email)

        elif command == "login":
            username = input("Enter your username: ")
            password = getpass.getpass("Enter your password: ")

            logged_user = sql_manager.login(username, password)

            if logged_user:
                logged_menu(logged_user)
            else:
                print("Login failed")

        elif command == "send-reset-password":
            username = input("Enter your username: ")
            sql_manager.send_reset_pass_email(username)

        elif command == "reset-password":
            username = input("Enter your username: ")
            hash = input("Enter the hash code from your email: ")
            sql_manager.reset_password(username, hash)

        elif command == "help":
            print("login - for logging in!")
            print("register - for creating new account!")
            print("send-reset-password - to send a " "password reset code to your email!")
            print(
                "reset-password - to reset your password using a "
                "code from your email, to get it use the command above!"
            )
            print("exit - for closing the program!")

        elif command == "exit":
            break

        else:
            print("Not a valid command")