test_views.py

import pytest
from donate.models import (
    Account,
    Currency,
    Project,
)
from donate.routes import (
    get_donation_params,
    model_stripe_data,
)
from sqlalchemy.orm.exc import NoResultFound
from unittest.mock import patch


def validate_main_page_data(data):
    assert "donation-form" in data
    assert 'charge[amount]' in data
    assert 'cc-number' in data
    assert 'data-stripe' in data
    assert 'cc-exp' in data
    assert 'cvc' in data
    assert 'charge[recurring]' in data
    assert 'donor[email]' in data
    assert 'donation-form-name' in data
    assert 'donation-form-anonymous' in data
    assert 'donor[anonymous]' in data
    assert 'donor[name]' in data
    assert 'bitcoin' in data


def test_view_main(testapp):
    app = testapp
    result = app.get('/')
    assert result.status_code == 200
    data = result.data.decode('utf-8')
    validate_main_page_data(data)

    result = app.get('/index')
    assert result.status_code == 200
    data = result.data.decode('utf-8')
    validate_main_page_data(data)


def test_view_thanks(testapp):
    app = testapp

    result = app.get('/thanks')

    data = result.data.decode('utf-8')
    assert result.status_code == 200
    assert "humbled" in data
    assert "thankful" in data


def test_view_new_project_get(testapp):
    app = testapp

    result = app.get('/new/project')
    assert result.status_code == 200
    data = result.data.decode('utf-8')
    assert '<form action="/new/project" method="POST">' in data
    assert "Project Name" in data
    assert '<input class="project-name"' in data
    assert "Goal" in data
    assert '<input class="project-goal"' in data
    assert "Currency" in data
    assert '<input class="project-currency"' in data
    assert "Description" in data
    assert '<input class="project-description' in data
    assert '<input type="submit"' in data


def test_view_new_project_post(testapp, db):
    app = testapp

    project_name = "Test Project"
    ccy_code = "USD"
    ccy_name = "US Dollar"
    amt = 4000
    proj_desc = "A test project"

    ccy = Currency(code=ccy_code, name=ccy_name)
    db.session.add(ccy)

    db.session.commit()

    data = {
        'goal': amt,
        'ccy': ccy_code,
        'desc': proj_desc,
        'project_name': project_name,
    }
    url = "/new/project"
    response = app.post(url, data=data)

    project = db.session.query(Project).filter_by(name=project_name).one()
    assert project is not None
    assert project.name == project_name
    assert project.desc == proj_desc
    assert project.goal == amt

    accounts = project.accounts
    assert accounts is not None
    assert len(accounts) == 1

    acct = accounts[0]
    assert acct.name == "{}_{}_acct".format(project_name, ccy_code)

    ccy = acct.ccy
    assert ccy.code == ccy_code
    assert ccy.name == ccy_name

    assert response.status_code == 200


@pytest.mark.usefixtures('test_db_project')
def test_projects_page(testapp, db):
    app = testapp

    response = app.get('/projects')
    assert response.status_code == 200

    proj = db.session.query(Project).one()

    data = response.data.decode('utf-8')
    assert proj.name in data


@pytest.mark.usefixtures('test_db_project')
def test_one_project_page(testapp, db):
    app = testapp

    proj = db.session.query(Project).one()
    response = app.get('/projects/{}'.format(proj.name))

    assert response.status_code == 200

    data = response.data.decode('utf-8')

    assert proj.name in data
    assert str(proj.goal) in data


def test_get_donation_params(testapp, test_form):
    form = test_form

    result = get_donation_params(form)

    assert result['charge'] == form.amt
    assert result['email'] == form.vals["donor[email]"]
    assert result['name'] == form.vals["donor[name]"]
    assert result['stripe_token'] == form.vals["donor[stripe_token]"]
    assert result['recurring'] == ("charge[recurring]" in form.vals)
    assert result['anonymous'] == ("donor[anonymous]" in form.vals)
    assert result['project_select'] == form.vals['project_select']


def test_get_donation_params_error(testapp, test_form):
    form = test_form
    form.vals["donor[email]"] = ''
    with pytest.raises(KeyError):
        get_donation_params(form)


@pytest.mark.usefixtures('test_db_project')
def test_model_stripe_data(testapp, test_form, db):
    req_data = {
        'charge': "100",
        'email': "rando@thewhatever.org",
        'name': "Jeru the Damaga",
        'stripe_token': "abcd1234",
        'recurring': False,
        'anonymous': False,
        'project_select': "Throws Error"
    }

    with pytest.raises(NoResultFound):
        tx = model_stripe_data(req_data)

    proj = db.session.query(Project).one()
    acct = proj.accounts[0]

    req_data['project_select'] = proj.name

    tx = model_stripe_data(req_data)
    db.session.add(tx)

    payer = db.session.query(Account).filter_by(name=req_data['email']).one()

    assert tx.amount == int(req_data['charge']) * 100
    assert tx.payer == payer
    assert tx.recvr == acct
    assert tx.payer.ccy_id == tx.recvr.ccy_id


@patch('donate.routes.get_donation_params')
@patch('donate.routes.create_charge')
@patch('donate.routes.flash')
@patch('donate.routes.get_recaptcha_resp')
def test_donation_post(get_recaptcha_resp, flash, create_charge, get_params, testapp,
                       test_form, test_db_project, db):
    app = testapp
    proj = db.session.query(Project).one()

    donation_params = {
        'charge': 100,
        'email': 'bobloblaw@lawblog.com',
        'name': 'Bob Loblaw',
        'stripe_token': 'abs123',
        'recurring': True,
        'anonymous': False,
        'project_select': proj.name}

    test_form.vals['charge[amount]'] = [" ", str(test_form.amt), ""]
    test_form.vals['project_select'] = proj.name
    create_charge.return_value = {'charge_id': 1, 'plan_id': 2, 'customer_id': 3}
    get_recaptcha_resp.return_value = {"score": 1.0}

    vals = {}
    get_params.side_effect = KeyError(['test'])
    response = app.post("/donation", data=vals)
    assert get_recaptcha_resp.called
    assert response.status_code == 302

    get_params.side_effect = ValueError(['test'])
    response = app.post("/donation", data=vals)
    assert response.status_code == 302

    get_params.side_effect = None
    get_params.return_value = donation_params

    create_charge.return_value = {'charge_id': 1, 'plan_id': 2, 'customer_id': 3}
    response = app.post("/donation", data=test_form, follow_redirects=True)

    assert create_charge.called
    assert response.status_code == 200

    donation_params['recurring'] = False
    get_params.return_value = donation_params

    create_charge.return_value = {'charge_id': 0, 'customer_id': 1}
    response = app.post("/donation", data=test_form, follow_redirects=True)

    create_charge.side_effect = ValueError("bob@email.tld")
    response = app.post("/donation", data=test_form)
    assert response.status_code == 302