def test_list_donors():
donors = mr.Donors()
donor = mr.Donor()
# Test formatting and sorting of the dictionary when listing donors
assert str(donors.list_donors()).split("\n")[0] == "No. Name "
assert str(donors.list_donors()).split("\n")[1] == "==== ======="
assert str(donors.list_donors()).split("\n")[2] == "1 Arya Stark"
def test_file_created():
donors = mr.Donors()
donor = mr.Donor()
donors.send_letter_to_all()
file_to_test = tempfile.gettempdir() + "/James_Butts.txt"
# Make sure letters are getting created (written to temp)
assert os.path.exists(file_to_test)
def test_show_donors():
donors = mr.Donors()
donor = mr.Donor()
# Check calculations
# Number of donations
assert len(donors.donor_db["Melisandre"]) == 4
# Total donations
assert sum(donors.donor_db["Melisandre"]) == 300019.00
# Average donation
assert sum(donors.donor_db["Melisandre"]) / len(donors.donor_db["Melisandre"]) == 75004.75
def test_add_contribution():
donors = mr.Donors()
donor = mr.Donor()
donors.add_contribution("James Butts", 100)
assert donors.donor_db["James Butts"][-1:] == [100]
# Test that we are summing total contributions correctly
assert sum(donors.donor_db["James Butts"]) == 100
# Pass a contribution to a recurring donor
donors.add_contribution("Daenerys Targaryen", 100)
assert donors.donor_db["Daenerys Targaryen"][-1:] == [100]
# Test that we are summing total contributions correctly
assert sum(donors.donor_db["Daenerys Targaryen"]) == 8355.00
# Pass multiple contributions to a recurring donor
donors.add_contribution("Daenerys Targaryen", [100, 200])
assert donors.donor_db["Daenerys Targaryen"][-2:] == [100, 200]
# Test that we are summing total contributions correctly for multiple contributions
assert sum(donors.donor_db["Daenerys Targaryen"]) == 8655.00
def test_format_email():
donors = mr.Donors()
donor = mr.Donor()
# test normal case for generating mail text
assert donor.format_email("James Butts", 100.00, 1000.00) == ''.join((
"Dear James Butts,\n\n",
"Thank you for your recent contribution of $100.00.\n\n",
"We appreciate your generosity in support of our mission.\n\n",
"Thank you for your lifetime contributions of $1000.00.\n\n"
"Warmest Regards,\n\n",
"Charity Staff\n"))
# test for number formatting in mail text
assert donor.format_email("Joe Biden", 10000.9999, 1000.1) == ''.join((
"Dear Joe Biden,\n\n",
"Thank you for your recent contribution of $10001.00.\n\n",
"We appreciate your generosity in support of our mission.\n\n",
"Thank you for your lifetime contributions of $1000.10.\n\n"
"Warmest Regards,\n\n",
"Charity Staff\n"))
def test_get_letter_text():
donors = mr.Donors()
donor = mr.Donor()
donors.send_letter_to_all()
letter_text = ""
tempfilepath = tempfile.gettempdir() + "/James_Butts.txt"
with open(tempfilepath, "r") as letter_file:
letter_text = letter_file.readlines()
letter_text = "".join(letter_text)
print(letter_text)
print(''.join(("Dear James Butts,\n\n",
"Thank you for your recent contribution of $100.00.\n\n",
"We appreciate your generosity in support of our mission.\n\n",
"Thank you for your lifetime contributions of $100.00.\n\n",
"Warmest Regards,\n\n",
"Charity Staff\n")))
assert letter_text == ''.join(("Dear James Butts,\n\n",
"Thank you for your recent contribution of $100.00.\n\n",
"We appreciate your generosity in support of our mission.\n\n",
"Thank you for your lifetime contributions of $100.00.\n\n",
"Warmest Regards,\n\n",
"Charity Staff\n"))
def test_get_donors_details():
donors = mr.Donors()
donor = mr.Donor()
# Check output and formating of get_donor_details()
donor_list = donors.get_donors_details()
assert donor_list[1] == 'Arya Stark $10899.92 5 $2179.98 \n'
input("Donor {} doesn't exist, add them (y/n)?:".format(
donor_name)))
if response.lower() == 'y':
donors.add_contribution(donor_name, amount)
print(
donor.format_email(donor_name, amount,
sum(donors.donor_db[donor_name])))
# return donor_db
def main():
'''
main loop, continually present users with menu options
'''
while True:
switcher(int(input(print_menu())))
if __name__ == '__main__':
# We're not getting imported, run main():
# donor_db = {} # dict that contains user/donation records
donors = mr.Donors()
donor = mr.Donor()
donors.populate_data
main()
else:
# donor_db = {}
donors = mr.Donors()
donor = mr.Donor()
donors.populate_data