def test_add_donor():
mr.donor_db = mr.get_donor_db()
name = "Fred Flintstone "
donor = mr.add_donor(name)
donor[1].append(300)
assert donor[0] == "Fred Flintstone"
assert donor[1] == [300]
assert mr.find_donor(name) == donor
def test_list_donors():
donor_db = mr.get_donor_db()
listing = mr.list_donors()
# hard to test this throughly -- better not to hard code the entire
# thing. But check for a few aspects -- this will catch the likely
# errors
print(repr(listing))
assert listing.startswith("\nDonor list:")
assert "Jeff Bezos" in listing
assert "William Gates III" in listing
assert len(listing.split('\n')) == 6
def test_save_letters_to_disk():
"""
This only tests that the files get created, but that's a start
Note that the contents of the letter was already
tested with test_gen_letter
"""
mr.donor_db = mr.get_donor_db()
mr.save_letters_to_disk()
assert os.path.isfile('Jeff_Bezos.txt')
assert os.path.isfile('William_Gates_III.txt')
# check that it's not empty:
with open('William_Gates_III.txt') as f:
size = len(f.read())
assert size > 0
def test_add_donation_new_donor():
"""
tests adding a dontation when the donor is new
"""
# make sure the DB is clean
mr.donor_db = mr.get_donor_db()
name = "John Smith"
donation_amount = 300.5
# make sure he's not already there
assert mr.find_donor(name) is None
mr.add_donation(name, donation_amount)
donor = mr.find_donor(name)
assert donor is not None
assert donor[1] == [donation_amount]
def test_generate_donor_report():
# make sure the the database is "clean"
mr.donor_db = mr.get_donor_db()
report = mr.generate_donor_report()
print(report) # printing so you can see it if it fails
# this is pretty tough to test
# these are not great, because they will fail if unimportant parts of the
# report are changed.
# but at least you know that code is working now.
assert report.startswith(
"Donor Name | Total Given | Num Gifts | Average Gift")
assert "Jeff Bezos $ 877.33 1 $ 877.33" in report
lines = report.split("\n")
assert len(lines) == 6
def test_find_donor_not():
"test one that's not there"
mr.donor_db = mr.get_donor_db()
donor = mr.find_donor("Jeff Bzos")
assert donor is None
def test_find_donor():
""" checks a donor that is there, but with odd case and spaces"""
mr.donor_db = mr.get_donor_db()
donor = mr.find_donor("jefF beZos ")
assert donor[0] == "Jeff Bezos"
#!/usr/bin/env python
"""
unit tests for the mailroom program
"""
import os
import pytest
import mailroom as mr
# so that it's there for the tests
# you may want to re-run this in a test that requires a
# "clean" database
mr.donor_db = mr.get_donor_db()
def test_normalize_name():
"""
make sure that the normalize function works as expected
"""
name = " A verY Odd Name. \t and long! "
assert mr.normalize_name(name) == "a very odd name. and long!"
def test_list_donors():
donor_db = mr.get_donor_db()
listing = mr.list_donors()
# hard to test this throughly -- better not to hard code the entire
# thing. But check for a few aspects -- this will catch the likely
#!/usr/bin/env python3
import mailroom
from mailroom import *
import os
import pytest
mailroom.donor_db = mailroom.get_donor_db()
def test_donor_db():
assert list_donors(donor_db) == [
'Jimmy Hendrix', 'Jack White', 'Keith Richards', 'Jimmy Page',
'Albert Hamond'
]
def test_add_donor():
new_donor_name = "Test"
add_donor(new_donor_name, donor_db)
assert list_donors(donor_db) == [
'Jimmy Hendrix', 'Jack White', 'Keith Richards', 'Jimmy Page',
'Albert Hamond', 'Test'
]
add_donation(new_donor_name, 10, donor_db)
def test_add_donation():
assert 10 in donor_db['Test']
