mirror of
https://github.com/kennethreitz-archive/www.gittip.com.git
synced 2026-06-20 07:10:56 +00:00
Chad Whitacre
351 lines
12 KiB
Python
351 lines
12 KiB
Python
from __future__ import unicode_literals
|
|
import random
|
|
import datetime
|
|
from decimal import Decimal
|
|
|
|
import psycopg2
|
|
import pytz
|
|
from nose.tools import assert_raises
|
|
|
|
from gittip.testing import Harness
|
|
from gittip.models import Participant, Tip
|
|
from gittip.participant import Participant as OldParticipant
|
|
|
|
|
|
class Tests(Harness):
|
|
|
|
def random_restricted_username(self):
|
|
"""helper method to chooses a restricted username for testing """
|
|
from gittip import RESTRICTED_USERNAMES
|
|
random_item = random.choice(RESTRICTED_USERNAMES)
|
|
while random_item.startswith('%'):
|
|
random_item = random.choice(RESTRICTED_USERNAMES)
|
|
return random_item
|
|
|
|
def setUp(self):
|
|
super(Harness, self).setUp()
|
|
self.participant = self.make_participant('user1') # Our protagonist
|
|
|
|
|
|
def test_claiming_participant(self):
|
|
expected = now = datetime.datetime.now(pytz.utc)
|
|
self.participant.set_as_claimed(claimed_at=now)
|
|
actual = self.participant.claimed_time
|
|
assert actual == expected, actual
|
|
|
|
def test_changing_username_successfully(self):
|
|
self.participant.change_username('user2')
|
|
actual = Participant.query.get('user2')
|
|
assert self.participant == actual, actual
|
|
|
|
def test_changing_username_to_too_long(self):
|
|
with assert_raises(Participant.UsernameTooLong):
|
|
self.participant.change_username('123456789012345678901234567890123')
|
|
|
|
def test_changing_username_to_already_taken(self):
|
|
self.make_participant('user2')
|
|
with assert_raises(Participant.UsernameAlreadyTaken):
|
|
self.participant.change_username('user2')
|
|
|
|
def test_changing_username_to_already_taken_is_case_insensitive(self):
|
|
self.make_participant('UsEr2')
|
|
with assert_raises(Participant.UsernameAlreadyTaken):
|
|
self.participant.change_username('uSeR2')
|
|
|
|
def test_changing_username_to_invalid_characters(self):
|
|
with assert_raises(Participant.UsernameContainsInvalidCharacters):
|
|
self.participant.change_username(u"\u2603") # Snowman
|
|
|
|
def test_changing_username_to_restricted_name(self):
|
|
with assert_raises(Participant.UsernameIsRestricted):
|
|
self.participant.change_username(self.random_restricted_username())
|
|
|
|
def test_getting_tips_actually_made(self):
|
|
expected = Decimal('1.00')
|
|
self.make_participant('user2')
|
|
self.session.add(Tip(tipper='user1', tippee='user2', amount=expected,
|
|
ctime=datetime.datetime.now(pytz.utc)))
|
|
self.session.commit()
|
|
actual = self.participant.get_tip_to('user2')
|
|
assert actual == expected, actual
|
|
|
|
def test_getting_tips_not_made(self):
|
|
expected = Decimal('0.00')
|
|
self.make_participant('user2')
|
|
actual = self.participant.get_tip_to('user2')
|
|
assert actual == expected, actual
|
|
|
|
|
|
# id
|
|
|
|
def test_participant_gets_a_long_id(self):
|
|
actual = type(self.make_participant('alice').id)
|
|
assert actual == long, actual
|
|
|
|
|
|
# set_tip_to - stt
|
|
|
|
def test_stt_sets_tip_to(self):
|
|
alice = self.make_participant('alice', last_bill_result='')
|
|
self.make_participant('bob')
|
|
alice.set_tip_to('bob', '1.00')
|
|
|
|
actual = alice.get_tip_to('bob')
|
|
assert actual == Decimal('1.00'), actual
|
|
|
|
def test_stt_returns_a_Decimal_and_a_boolean(self):
|
|
alice = self.make_participant('alice', last_bill_result='')
|
|
self.make_participant('bob')
|
|
actual = alice.set_tip_to('bob', '1.00')
|
|
assert actual == (Decimal('1.00'), True), actual
|
|
|
|
def test_stt_returns_False_for_second_time_tipper(self):
|
|
alice = self.make_participant('alice', last_bill_result='')
|
|
self.make_participant('bob')
|
|
alice.set_tip_to('bob', '1.00')
|
|
actual = alice.set_tip_to('bob', '2.00')
|
|
assert actual == (Decimal('2.00'), False), actual
|
|
|
|
def test_stt_doesnt_allow_self_tipping(self):
|
|
alice = self.make_participant('alice', last_bill_result='')
|
|
assert_raises( OldParticipant.NoSelfTipping
|
|
, alice.set_tip_to
|
|
, 'alice'
|
|
, '1000000.00'
|
|
)
|
|
|
|
def test_stt_doesnt_allow_just_any_ole_amount(self):
|
|
alice = self.make_participant('alice', last_bill_result='')
|
|
self.make_participant('bob')
|
|
assert_raises( OldParticipant.BadAmount
|
|
, alice.set_tip_to
|
|
, 'bob'
|
|
, '1000000.00'
|
|
)
|
|
|
|
def test_stt_fails_to_tip_unknown_people(self):
|
|
alice = self.make_participant('alice', last_bill_result='')
|
|
assert_raises( psycopg2.IntegrityError
|
|
, alice.set_tip_to
|
|
, 'bob'
|
|
, '1.00'
|
|
)
|
|
|
|
|
|
# get_dollars_receiving - gdr
|
|
|
|
def test_gdr_only_sees_latest_tip(self):
|
|
alice = self.make_participant('alice', last_bill_result='')
|
|
bob = self.make_participant('bob')
|
|
alice.set_tip_to('bob', '12.00')
|
|
alice.set_tip_to('bob', '3.00')
|
|
self.session.commit()
|
|
|
|
expected = Decimal('3.00')
|
|
actual = bob.get_dollars_receiving()
|
|
assert actual == expected, actual
|
|
|
|
|
|
def test_gdr_includes_tips_from_accounts_with_a_working_card(self):
|
|
alice = self.make_participant('alice', last_bill_result='')
|
|
bob = self.make_participant('bob')
|
|
alice.set_tip_to('bob', '3.00')
|
|
self.session.commit()
|
|
|
|
expected = Decimal('3.00')
|
|
actual = bob.get_dollars_receiving()
|
|
assert actual == expected, actual
|
|
|
|
def test_gdr_ignores_tips_from_accounts_with_no_card_on_file(self):
|
|
alice = self.make_participant('alice', last_bill_result=None)
|
|
bob = self.make_participant('bob')
|
|
alice.set_tip_to('bob', '3.00')
|
|
self.session.commit()
|
|
|
|
expected = Decimal('0.00')
|
|
actual = bob.get_dollars_receiving()
|
|
assert actual == expected, actual
|
|
|
|
def test_gdr_ignores_tips_from_accounts_with_a_failing_card_on_file(self):
|
|
alice = self.make_participant('alice', last_bill_result="Fail!")
|
|
bob = self.make_participant('bob')
|
|
alice.set_tip_to('bob', '3.00')
|
|
self.session.commit()
|
|
|
|
expected = Decimal('0.00')
|
|
actual = bob.get_dollars_receiving()
|
|
assert actual == expected, actual
|
|
|
|
|
|
def test_gdr_includes_tips_from_whitelisted_accounts(self):
|
|
alice = self.make_participant( 'alice'
|
|
, last_bill_result=''
|
|
, is_suspicious=False
|
|
)
|
|
bob = self.make_participant('bob')
|
|
alice.set_tip_to('bob', '3.00')
|
|
self.session.commit()
|
|
|
|
expected = Decimal('3.00')
|
|
actual = bob.get_dollars_receiving()
|
|
assert actual == expected, actual
|
|
|
|
def test_gdr_includes_tips_from_unreviewed_accounts(self):
|
|
alice = self.make_participant( 'alice'
|
|
, last_bill_result=''
|
|
, is_suspicious=None
|
|
)
|
|
bob = self.make_participant('bob')
|
|
alice.set_tip_to('bob', '3.00')
|
|
self.session.commit()
|
|
|
|
expected = Decimal('3.00')
|
|
actual = bob.get_dollars_receiving()
|
|
assert actual == expected, actual
|
|
|
|
def test_gdr_ignores_tips_from_blacklisted_accounts(self):
|
|
alice = self.make_participant( 'alice'
|
|
, last_bill_result=''
|
|
, is_suspicious=True
|
|
)
|
|
bob = self.make_participant('bob')
|
|
alice.set_tip_to('bob', '3.00')
|
|
self.session.commit()
|
|
|
|
expected = Decimal('0.00')
|
|
actual = bob.get_dollars_receiving()
|
|
assert actual == expected, actual
|
|
|
|
|
|
# get_number_of_backers - gnob
|
|
|
|
def test_gnob_gets_number_of_backers(self):
|
|
alice = self.make_participant('alice', last_bill_result='')
|
|
bob = self.make_participant('bob', last_bill_result='')
|
|
clancy = self.make_participant('clancy')
|
|
|
|
alice.set_tip_to('clancy', '3.00')
|
|
bob.set_tip_to('clancy', '1.00')
|
|
|
|
self.session.commit()
|
|
|
|
actual = clancy.get_number_of_backers()
|
|
assert actual == 2, actual
|
|
|
|
|
|
def test_gnob_includes_backers_with_a_working_card_on_file(self):
|
|
alice = self.make_participant('alice', last_bill_result='')
|
|
bob = self.make_participant('bob')
|
|
alice.set_tip_to('bob', '3.00')
|
|
self.session.commit()
|
|
|
|
actual = bob.get_number_of_backers()
|
|
assert actual == 1, actual
|
|
|
|
def test_gnob_ignores_backers_with_no_card_on_file(self):
|
|
alice = self.make_participant('alice', last_bill_result=None)
|
|
bob = self.make_participant('bob')
|
|
alice.set_tip_to('bob', '3.00')
|
|
self.session.commit()
|
|
|
|
actual = bob.get_number_of_backers()
|
|
assert actual == 0, actual
|
|
|
|
def test_gnob_ignores_backers_with_a_failing_card_on_file(self):
|
|
alice = self.make_participant('alice', last_bill_result="Fail!")
|
|
bob = self.make_participant('bob')
|
|
alice.set_tip_to('bob', '3.00')
|
|
self.session.commit()
|
|
|
|
actual = bob.get_number_of_backers()
|
|
assert actual == 0, actual
|
|
|
|
|
|
def test_gnob_includes_whitelisted_backers(self):
|
|
alice = self.make_participant( 'alice'
|
|
, last_bill_result=''
|
|
, is_suspicious=False
|
|
)
|
|
bob = self.make_participant('bob')
|
|
alice.set_tip_to('bob', '3.00')
|
|
self.session.commit()
|
|
|
|
actual = bob.get_number_of_backers()
|
|
assert actual == 1, actual
|
|
|
|
def test_gnob_includes_unreviewed_backers(self):
|
|
alice = self.make_participant( 'alice'
|
|
, last_bill_result=''
|
|
, is_suspicious=None
|
|
)
|
|
bob = self.make_participant('bob')
|
|
alice.set_tip_to('bob', '3.00')
|
|
self.session.commit()
|
|
|
|
actual = bob.get_number_of_backers()
|
|
assert actual == 1, actual
|
|
|
|
def test_gnob_ignores_blacklisted_backers(self):
|
|
alice = self.make_participant( 'alice'
|
|
, last_bill_result=''
|
|
, is_suspicious=True
|
|
)
|
|
bob = self.make_participant('bob')
|
|
alice.set_tip_to('bob', '3.00')
|
|
self.session.commit()
|
|
|
|
actual = bob.get_number_of_backers()
|
|
assert actual == 0, actual
|
|
|
|
|
|
def test_gnob_ignores_backers_where_tip_is_zero(self):
|
|
alice = self.make_participant('alice', last_bill_result='')
|
|
bob = self.make_participant('bob')
|
|
alice.set_tip_to('bob', '0.00')
|
|
self.session.commit()
|
|
|
|
actual = bob.get_number_of_backers()
|
|
assert actual == 0, actual
|
|
|
|
def test_gnob_looks_at_latest_tip_only(self):
|
|
alice = self.make_participant('alice', last_bill_result='')
|
|
bob = self.make_participant('bob')
|
|
|
|
alice.set_tip_to('bob', '1.00')
|
|
alice.set_tip_to('bob', '12.00')
|
|
alice.set_tip_to('bob', '3.00')
|
|
alice.set_tip_to('bob', '6.00')
|
|
alice.set_tip_to('bob', '0.00')
|
|
|
|
self.session.commit()
|
|
|
|
actual = bob.get_number_of_backers()
|
|
assert actual == 0, actual
|
|
|
|
|
|
# get_age_in_seconds - gais
|
|
|
|
def test_gais_gets_age_in_seconds(self):
|
|
now = datetime.datetime.now(pytz.utc)
|
|
alice = self.make_participant('alice', claimed_time=now)
|
|
actual = alice.get_age_in_seconds()
|
|
assert 0 < actual < 1, actual
|
|
|
|
def test_gais_returns_negative_one_if_None(self):
|
|
alice = self.make_participant('alice', claimed_time=None)
|
|
actual = alice.get_age_in_seconds()
|
|
assert actual == -1, actual
|
|
|
|
# def get_details(self):
|
|
# def resolve_unclaimed(self):
|
|
# def set_as_claimed(self):
|
|
# def change_username(self, suggested):
|
|
# def get_accounts_elsewhere(self):
|
|
# def get_tip_to(self, tippee):
|
|
# def get_dollars_receiving(self):
|
|
# def get_dollars_giving(self):
|
|
# def get_chart_of_receiving(self):
|
|
# def get_giving_for_profile(self, db=None):
|
|
# def get_tips_and_total(self, for_payday=False, db=None):
|
|
# def take_over(self, account_elsewhere, have_confirmation=False):
|