from .test_payloads.payload_fees_worksheet_router import *

# ======================================================= #
# =============== LOAN OFFICER TEST CASES =============== #
# ======================================================= #


# get loan cost types
def test_get_loan_cost_types(test_client, loan_officer_header):
    url = "/fees/loan_cost_types"
    response = test_client.get(url, headers=loan_officer_header)

    assert response.status_code == 200
    response = response.json()
    assert response["status"] == 200
    assert response["message"] == "Loan Cost data fetched successfully!"


# get closing cost calculations
def test_get_closing_cost_calculations(test_client, application_id, loan_officer_header):
    url = f"/fees/closing_cost_calculation/{application_id}"
    response = test_client.get(url, headers=loan_officer_header)

    if response.status_code == 404:
        assert response.status_code == 404
        response = response.json()
        assert response["status"] == 404
        assert response["message"] == "Loan Information not found"
    else:
        assert response.status_code == 200
        response = response.json()
        assert response["status"] == 200
        assert response["message"] == "SettleWise data fetched successfully!"


#  get user fees worksheet
def test_get_user_fees_worksheet(test_client, application_id, loan_officer_header):
    url = f"/fees/{application_id}"
    response = test_client.get(url, headers=loan_officer_header)

    if response.status_code == 404:
        assert response.status_code == 404
        response = response.json()
        assert response["status"] == 404
        assert response["message"] == "Applications not found"
    else:
        assert response.status_code == 200
        response = response.json()
        assert response["status"] == 200
        assert response["message"] == "Fees Worksheet data fetch successfully!"


#  add user fees worksheet
def test_add_user_fees_worksheet(test_client, application_id, loan_officer_header):
    payload = add_fees_worksheet_payload
    url = f"/fees/{application_id}"
    response = test_client.post(url, json=payload, headers=loan_officer_header)

    if response.status_code == 404:
        assert response.status_code == 404
        response = response.json()
        assert response["status"] == 404
        assert response["message"] == "Applications already exist"
    else:
        assert response.status_code == 200
        response = response.json()
        assert response["status"] == 200
        assert response["message"] == "User Fees Details saved successfully!"


#  update user fees worksheet
def test_update_user_fees_worksheet(test_client, application_id, loan_officer_header):
    payload = update_fees_worksheet_payload
    url = f"/fees/{application_id}"
    response = test_client.put(url, json=payload, headers=loan_officer_header)

    if response.status_code == 404:
        assert response.status_code == 404
        response = response.json()
        assert response["status"] == 404
        assert response["message"] == "Applications not found"
    else:
        assert response.status_code == 200
        response = response.json()
        assert response["status"] == 200
        assert response["message"] == "User Fees Details updated successfully!"


# ======================================================= #
# ============ LOAN OFFICER ADMIN TEST CASES ============ #
# ======================================================= #


# get loan cost types
def test_get_loan_cost_types_admin(test_client, loan_officer_admin_header):
    url = "/fees/loan_cost_types"
    response = test_client.get(url, headers=loan_officer_admin_header)

    assert response.status_code == 200
    response = response.json()
    assert response["status"] == 200
    assert response["message"] == "Loan Cost data fetched successfully!"


# get closing cost calculations
def test_get_closing_cost_calculations_admin(test_client, application_id, loan_officer_header):
    url = f"/fees/closing_cost_calculation/{application_id}"
    response = test_client.get(url, headers=loan_officer_header)
    assert response.status_code == 200

    if response.status_code == 404:
        assert response.status_code == 404
        response = response.json()
        assert response["status"] == 404
        assert response["message"] == "Loan Information not found"
    else:
        assert response.status_code == 200
        response = response.json()
        assert response["status"] == 200
        assert response["message"] == "SettleWise data fetched successfully!"


#  get user fees worksheet
def test_get_user_fees_worksheet_admin(test_client, application_id, loan_officer_admin_header):
    url = f"/fees/{application_id}"
    response = test_client.get(url, headers=loan_officer_admin_header)

    if response.status_code == 404:
        assert response.status_code == 404
        response = response.json()
        assert response["status"] == 404
        assert response["message"] == "Applications not found"
    else:
        assert response.status_code == 200
        response = response.json()
        assert response["status"] == 200
        assert response["message"] == "Fees Worksheet data fetch successfully!"


#  add user fees worksheet
def test_add_user_fees_worksheet_admin(test_client, application_id, loan_officer_admin_header):
    payload = add_fees_worksheet_payload
    url = f"/fees/{application_id}"
    response = test_client.post(url, json=payload, headers=loan_officer_admin_header)

    if response.status_code == 404:
        assert response.status_code == 404
        response = response.json()
        assert response["status"] == 404
        assert response["message"] == "Applications already exist"
    else:
        assert response.status_code == 200
        response = response.json()
        assert response["status"] == 200
        assert response["message"] == "User Fees Details saved successfully!"


#  update user fees worksheet
def test_update_user_fees_worksheet_admin(test_client, application_id, loan_officer_admin_header):
    payload = update_fees_worksheet_payload
    url = f"/fees/{application_id}"
    response = test_client.put(url, json=payload, headers=loan_officer_admin_header)

    if response.status_code == 404:
        assert response.status_code == 404
        response = response.json()
        assert response["status"] == 404
        assert response["message"] == "Applications not found"
    else:
        assert response.status_code == 200
        response = response.json()
        assert response["status"] == 200
        assert response["message"] == "User Fees Details updated successfully!"