mirror of
https://github.com/kennethreitz-archive/www.gittip.com.git
synced 2026-06-05 23:30:19 +00:00
Selena Deckelmann
123 lines
4.7 KiB
Python
123 lines
4.7 KiB
Python
from datetime import datetime
|
|
from decimal import Decimal
|
|
|
|
from aspen.utils import utcnow
|
|
from mock import patch
|
|
from nose.tools import assert_equals
|
|
|
|
from gittip import wireup
|
|
from gittip.billing.payday import Payday
|
|
from gittip.models import Tip
|
|
from gittip.participant import Participant
|
|
from gittip.testing import Harness, load_simplate
|
|
from gittip.testing.client import TestClient
|
|
|
|
|
|
class TestCommaize(Harness):
|
|
# XXX This really ought to be in helper methods test file
|
|
def setUp(self):
|
|
super(Harness, self).setUp()
|
|
simplate = load_simplate('/about/stats.html')
|
|
self.commaize = simplate.pages[0]['commaize']
|
|
|
|
def test_commaize_commaizes(self):
|
|
actual = self.commaize(1000.0)
|
|
assert actual == "1,000", actual
|
|
|
|
def test_commaize_commaizes_and_obeys_decimal_places(self):
|
|
actual = self.commaize(1000, 4)
|
|
assert actual == "1,000.0000", actual
|
|
|
|
|
|
class TestChartOfReceiving(Harness):
|
|
def setUp(self):
|
|
super(Harness, self).setUp()
|
|
for participant in ['alice', 'bob']:
|
|
self.make_participant(participant, last_bill_result='')
|
|
|
|
def test_get_tip_distribution_handles_a_tip(self):
|
|
Participant(u'alice').set_tip_to('bob', '3.00')
|
|
expected = ([[Decimal('3.00'), 1, Decimal('3.00'), 1.0, Decimal('1')]],
|
|
1.0, Decimal('3.00'))
|
|
actual = Participant(u'bob').get_tip_distribution()
|
|
assert_equals(actual, expected)
|
|
|
|
def test_get_tip_distribution_handles_a_non_standard_amount(self):
|
|
tip = Tip(tipper='alice', tippee='bob', amount='5.37', ctime=utcnow())
|
|
self.session.add(tip)
|
|
self.session.commit()
|
|
expected = ([[-1, 1, Decimal('5.37'), 1.0, Decimal('1')]],
|
|
1.0, Decimal('5.37'))
|
|
actual = Participant(u'bob').get_tip_distribution()
|
|
assert_equals(actual, expected)
|
|
|
|
def test_get_tip_distribution_handles_no_tips(self):
|
|
expected = ([], 0.0, Decimal('0.00'))
|
|
actual = Participant(u'foo').get_tip_distribution()
|
|
assert_equals(actual, expected)
|
|
|
|
def test_get_tip_distribution_handles_multiple_tips(self):
|
|
self.make_participant('carl', last_bill_result='')
|
|
Participant(u'alice').set_tip_to('bob', '1.00')
|
|
Participant(u'carl').set_tip_to('bob', '3.00')
|
|
expected = ([
|
|
[Decimal('1.00'), 1L, Decimal('1.00'), 0.5, Decimal('0.25')],
|
|
[Decimal('3.00'), 1L, Decimal('3.00'), 0.5, Decimal('0.75')]
|
|
], 2.0, Decimal('4.00'))
|
|
actual = Participant(u'bob').get_tip_distribution()
|
|
assert_equals(actual, expected)
|
|
|
|
def test_get_tip_distribution_ignores_bad_cc(self):
|
|
self.make_participant('bad_cc', last_bill_result='Failure!')
|
|
Participant(u'alice').set_tip_to('bob', '1.00')
|
|
Participant(u'bad_cc').set_tip_to('bob', '3.00')
|
|
expected = ([[Decimal('1.00'), 1L, Decimal('1.00'), 1, Decimal('1')]],
|
|
1.0, Decimal('1.00'))
|
|
actual = Participant(u'bob').get_tip_distribution()
|
|
assert_equals(actual, expected)
|
|
|
|
def test_get_tip_distribution_ignores_missing_cc(self):
|
|
self.make_participant('missing_cc', last_bill_result=None)
|
|
Participant(u'alice').set_tip_to('bob', '1.00')
|
|
Participant(u'missing_cc').set_tip_to('bob', '3.00')
|
|
expected = ([[Decimal('1.00'), 1L, Decimal('1.00'), 1, Decimal('1')]],
|
|
1.0, Decimal('1.00'))
|
|
actual = Participant(u'bob').get_tip_distribution()
|
|
assert_equals(actual, expected)
|
|
|
|
|
|
class TestRenderingStatsPage(Harness):
|
|
def get_stats_page(self):
|
|
return TestClient().get('/about/stats.html').body
|
|
|
|
@patch('datetime.datetime')
|
|
def test_stats_description_accurate_during_payday_run(self, mock_datetime):
|
|
"""Test that stats page takes running payday into account.
|
|
|
|
This test was originally written to expose the fix required for
|
|
https://github.com/gittip/www.gittip.com/issues/92.
|
|
"""
|
|
a_thursday = datetime(2012, 8, 9, 11, 00, 01)
|
|
mock_datetime.utcnow.return_value = a_thursday
|
|
|
|
wireup.billing()
|
|
payday = Payday(self.postgres)
|
|
payday.start()
|
|
|
|
body = self.get_stats_page()
|
|
assert "is changing hands <b>right now!</b>" in body, body
|
|
payday.end()
|
|
|
|
@patch('datetime.datetime')
|
|
def test_stats_description_accurate_outside_of_payday(self, mock_datetime):
|
|
"""Test stats page outside of the payday running"""
|
|
a_monday = datetime(2012, 8, 6, 11, 00, 01)
|
|
mock_datetime.utcnow.return_value = a_monday
|
|
|
|
payday = Payday(self.postgres)
|
|
payday.start()
|
|
|
|
body = self.get_stats_page()
|
|
assert "is ready for <b>this Thursday</b>" in body, body
|
|
payday.end()
|