import random
import pandas as pd
import utils

def generate_passed_usecase():
    usernames = ['admin']
    passwords = ['admin']
    df = pd.DataFrame()
    df['username'] = usernames
    df['password'] = passwords
    df.to_csv('./cases/login-001.csv', header=0, index=0)


def generate_uname_with_null_char():
    usernames = []
    passwords = []
    for i in range(0, 6):
        uname = utils.right_username[0:i] + ' ' + utils.right_username[i:5]
        usernames.append(uname)
        passwords.append(uname)
    df = pd.DataFrame()
    df['username'] = usernames
    df['password'] = passwords
    df.to_csv('./cases/login-002.csv', header=0, index=0)


def generate_passwd_with_null_char():
    usernames = []
    passwords = []
    for i in range(0, 6):
        uname = utils.right_passwd[0:i] + ' ' + utils.right_passwd[i:5]
        usernames.append(uname)
        passwords.append(uname)
    df = pd.DataFrame()
    df['username'] = usernames
    df['password'] = passwords
    df.to_csv('./cases/login-003.csv',header=0, index=0)


def generate_uname_with_capital():
    usernames = []
    passwords = []
    for i in range(0, 6):
        uname = utils.right_username[0:i].upper() + utils.right_username[i:5]
        usernames.append(uname)
        passwords.append(utils.right_passwd)
    df = pd.DataFrame()
    df['username'] = usernames
    df['password'] = passwords
    df.to_csv('./cases/login-004.csv', header=0, index=0)


def generate_passwd_with_capital():
    usernames = []
    passwords = []
    for i in range(0, 6):
        pswd = utils.right_passwd[0:i].upper() + utils.right_passwd[i:5]
        usernames.append(utils.right_username)
        passwords.append(pswd)
    df = pd.DataFrame()
    df['username'] = usernames
    df['password'] = passwords
    df.to_csv('./cases/login-005.csv', header=0, index=0)


def generate_unknown_usernames():
    usernames = []
    passwords = []
    for i in range(0, 6):
        uname = ""
        for j in range(0, random.randint(1, 5)):
            uname += random.choice(utils.unknown)
        usernames.append(uname)
        passwords.append(utils.right_passwd)
    df = pd.DataFrame()
    df['username'] = usernames
    df['password'] = passwords
    df.to_csv('./cases/login-006.csv', header=0, index=0)


def generate_not_in_db():
    usernames = []
    passwords = []
    for i in range(0, 6):
        uname = ""
        pswd = ""
        for j in range(0, random.randint(1, 5)):
            uname += random.choice(utils.alphabet)
            pswd += random.choice(utils.alphabet)
        if uname != utils.right_username:
            usernames.append(uname)
            passwords.append(pswd)
    df = pd.DataFrame()
    df['username'] = usernames
    df['password'] = passwords
    df.to_csv('./cases/login-007.csv', header=0, index=0)


def generate_null_case():
    usernames = [' ', '  ']
    passwords = [' ', '  ']
    df = pd.DataFrame()
    df['username'] = usernames
    df['password'] = passwords
    df.to_csv('./cases/login-008.csv', header=0, index=0)


def generate_null_uname():
    usernames = []
    passwords = []
    for i in range(5):
        usernames.append(' ')
        passwd = ''
        for j in range(random.randint(8, 14)):
            passwd += random.choice(utils.alphabet)
        passwords.append(passwd)
    df = pd.DataFrame()
    df['username'] = usernames
    df['password'] = passwords
    df.to_csv('./cases/login-009.csv', header=0, index=0)


def generate_null_password():
    usernames = []
    passwords = []
    for i in range(5):
        passwords.append(' ')
        uname = ''
        for j in range(random.randint(8, 14)):
            uname += random.choice(utils.alphabet)
        usernames.append(uname)
    df = pd.DataFrame()
    df['username'] = usernames
    df['password'] = passwords
    df.to_csv('./cases/login-010.csv', header=0, index=0)


if __name__ == '__main__':
    generate_passed_usecase()
    generate_uname_with_null_char()
    generate_passwd_with_null_char()
    generate_uname_with_capital()
    generate_passwd_with_capital()
    generate_unknown_usernames()
    generate_not_in_db()
    generate_null_case()
    generate_null_uname()
    generate_null_password()