def setUp(self):
reload(test_data)
self.set_up_database()
self.charity_repo = CharityRepository()
self.da_repo = DonationAmountRepository()
self.dp_repo = DonationProportionRepository()
self.payment_repo = PaymentRepository(self.dp_repo)
self.user_repo = UserRepository()
self.session_repo = SessionRepository(self.user_repo)
def post(self):
user_repo = UserRepository()
session_repo = SessionRepository(user_repo=user_repo)
user_repo.set_password(
email=self.request.POST['email'],
password=self.request.POST['password'],
confirmation_code=self.request.POST['confirmation_code'])
session = session_repo.log_in(email=self.request.POST['email'],
password=self.request.POST['password'])
response = self.redirect('/')
response.set_cookie('sessionid', session.id, httponly=True)
return response
def post(self):
user_repo = UserRepository()
session_repo = SessionRepository(user_repo=user_repo)
user_repo.set_password(
email=self.request.POST['email'],
password=self.request.POST['password'],
confirmation_code=self.request.POST['confirmation_code'])
session = session_repo.log_in(
email=self.request.POST['email'],
password=self.request.POST['password'])
response = self.redirect('/')
response.set_cookie('sessionid', session.id, httponly=True)
return response
def post(self):
session = SessionRepository(UserRepository()).log_in(
email=self.request.POST['email'],
password=self.request.POST['passwd'])
response = self.redirect('/')
response.set_cookie('sessionid', session.id, httponly=True)
return response
def test_sends_email_with_confirmation_link(self):
email = '*****@*****.**' % self.get_random_value()
self._assert_no_confirmed_user(email=email)
self._assert_no_unconfirmed_user(email=email)
request = webapp2.Request.blank(Url().for_page('signup'))
request.method = 'POST'
request.POST['email'] = email
class MockEmailService(object):
def send_user_confirmation_mail(self, user):
self.user = user
email_service = MockEmailService()
user_repo = UserRepository(email_service=email_service)
SignupPage(request=request, response=webapp2.Response(), user_repository=user_repo).post()
user = self.user_repo.get_unconfirmed_user(email=email)
self.assertEquals(email_service.user, user)
def __init__(self, request, response, user_repository=None):
BasePage.__init__(self, request, response)
self.user_repository = user_repository if user_repository is not None else UserRepository()
class TestCase(unittest.TestCase):
def setUp(self):
reload(test_data)
self.set_up_database()
self.charity_repo = CharityRepository()
self.da_repo = DonationAmountRepository()
self.dp_repo = DonationProportionRepository()
self.payment_repo = PaymentRepository(self.dp_repo)
self.user_repo = UserRepository()
self.session_repo = SessionRepository(self.user_repo)
def tearDown(self):
self.tear_down_database()
def set_up_database(self):
self.testbed = testbed.Testbed()
self.testbed.activate()
policy = datastore_stub_util.PseudoRandomHRConsistencyPolicy(probability=1)
self.testbed.init_datastore_v3_stub(consistency_policy=policy)
self.testbed.init_mail_stub()
def tear_down_database(self):
self.testbed.deactivate()
def add_charities(self, charities):
for charity in charities:
self.charity_repo.add_or_update_charity(charity)
def add_donation_proportions(self, dps):
for dp in dps:
self.dp_repo.add_donation_proportion(dp)
def add_confirmed_users(self, users):
for user in users:
self.user_repo.create_confirmed_user_FOR_TEST(user)
def create_ready_to_use_user(self):
email = '%s@%s.com' % (uuid.uuid4(), TestCase.get_and_increment_random_index())
password = str(uuid.uuid4())
new_user = User(email=email)
self.user_repo.create_unconfirmed_user(new_user)
self.user_repo.confirm_user(email=new_user.email, confirmation_code=new_user.confirmation_code)
self.user_repo.set_password(email=new_user.email, password=password, confirmation_code=new_user.confirmation_code)
return TestUser(email, password)
def add_payments(self, payments, payment_repository=None):
for payment in payments:
self.payment_repo.add_payment(payment)
def add_incoming_payments(self, incoming_payments):
for incoming_payment in incoming_payments:
self.payment_repo.add_incoming_payment(incoming_payment)
def setup_one_incoming_payment(self, users_and_amounts):
# users_and_amounts: {user: amount}
for (user, amount) in users_and_amounts.items():
self.setup_one_mismatched_incoming_payment({user: (amount, amount)})
def setup_one_mismatched_incoming_payment(self, users_to_payments):
# users_to_payments : {user: (expected, actual)}
for (user, (expected, actual)) in users_to_payments.items():
self.da_repo.set_donation_amount(user=user, amount_GBPennies=expected)
self.payment_repo.add_incoming_payment(IncomingPayment(user=user, amount_GBPennies=actual))
def setup_one_missing_payment(self, users_to_payments):
# users_and_amounts: {user: amount}
for (user, amount) in users_to_payments.items():
self.da_repo.set_donation_amount(user=user, amount_GBPennies=amount)
def get_only_sent_email(self, to):
mail_stub = self.testbed.get_stub(testbed.MAIL_SERVICE_NAME)
messages = mail_stub.get_sent_messages(to=to)
self.assertEquals(len(messages), 1)
return messages[0]
@classmethod
def get_and_increment_random_index(cls):
if not hasattr(cls, 'random_index'):
cls.random_index = -1
cls.random_index += 1
return cls.random_index
def get_random_value(self):
return str(uuid.uuid4())