def link_generator(campaign):
char_list = string.ascii_letters + string.digits
for i in range(campaign.count):
while (True):
new_char_url = ''.join(random.choice(char_list) for j in range(12))
try:
Voucher.objects.get(char_url=new_char_url)
except:
v = Voucher(campaign=campaign,char_url=new_char_url)
v.save()
break
def validate_voucher():
response = {'validity': 'invalid'}
data = request.json or request.form
code = data.get('text')
phone = data.get('phone')
flow = data.get('flow')
try:
Voucher.redeem(code, phone, flow)
response['validity'] = 'valid'
except VoucherException as e:
response['reason'] = str(e)
return create_response(response)
def is_valid_voucher_key(key_value):
from models import Voucher
clean_key_value = key_value.replace(' ', '').upper()
voucher = Voucher.by_voucher_key(clean_key_value)
# validate key... right now, we support
# thinkup promotional claim codes, which are in the form:
# [A-Za-z0-9]{12} (whitespace is stripped)
#if voucher is None:
# if re.match(r"^[A-Z0-9]{12}$", clean_key_value):
# # verify this is a valid code using ThinkUP API
# http = tornado.httpclient.HTTPClient()
# try:
# url = 'https://www.thinkup.com/join/api/bundle/?code=%s' % \
# clean_key_value
# response = http.fetch(url)
# data = {}
# if response.code == 200:
# data = json.loads(response.body)
# if 'is_valid' in data and data['is_valid'] == True:
# return Voucher(voucher_key=data['code'],
# promotion_id=options.thinkup_promotion_id)
# except tornado.httpclient.HTTPError as e:
# pass
# finally:
# http.close()
if voucher and voucher.claimed_by_user_id > 0:
return None
return voucher
def test_redeem_voucher_with_good_voucher(self):
self.sign_out()
user = test.factories.user(name="free_user",
email="*****@*****.**")
user.is_paid = 0
user.save()
self.sign_in(user.name, "password")
response = self.fetch_url("/account/settings")
# verify this account is currently free
self.assertTrue(
response.body.find("You are currently using a free account.") > -1)
arguments = {"key": "unclaimed"}
# this will post and redirect to the settings page which should
# then reflect that we are a paid user with 5 years of credit
response = self.post_url("/account/redeem", arguments)
self.assertTrue(response.body.find("5 Years") > -1)
payments = PaymentLog.where("user_id=%s", user.id)
self.assertEquals(len(payments), 1)
self.assertEquals(payments[0].operation, "redeem")
self.assertEquals(payments[0].status, "credit")
self.assertEquals(payments[0].reference_id, str(self.promotion.id))
self.assertEquals(payments[0].transaction_id, arguments['key'])
self.assertEquals(payments[0].buyer_email, user.email)
self.assertEquals(payments[0].buyer_name, user.name)
# self.assertEquals(payments[0].next_transaction_date, )
voucher = Voucher.get("claimed_by_user_id=%s", user.id)
self.assertEquals(voucher.promotion_id, self.promotion.id)
self.assertEquals(voucher.claimed_by_user_id, user.id)
self.assertEquals(voucher.offered_by_user_id, self.admin.id)
def setUp(self):
super(VoucherTests, self).setUp()
self.admin = test.factories.user()
self.sign_in("admin", "password")
self.shake = Shake(user_id=self.admin.id,
name='promotion-shake',
title='Promotion Shake',
type='group')
self.shake.save()
self.expired_promotion = Promotion(name="Expired Promotion",
membership_months=60,
expires_at=datetime.utcnow() -
timedelta(seconds=50),
promotion_shake_id=0)
self.expired_promotion.save()
self.promotion = Promotion(name="Unit Test Sale",
membership_months=60,
promotion_shake_id=self.shake.id,
expires_at=datetime.utcnow() +
timedelta(seconds=60 * 60 * 24 * 365))
self.promotion.save()
self.used_voucher = Voucher(offered_by_user_id=0,
promotion_id=self.promotion.id,
voucher_key="abc123")
# apply_to_user saves the voucher object (because it touches the
# claimed_by_user_id and dates) and also the user object (by
# updating the is_paid status)
self.used_voucher.apply_to_user(self.admin)
self.unused_voucher = Voucher(offered_by_user_id=0,
claimed_by_user_id=0,
promotion_id=self.promotion.id,
voucher_key="unclaimed")
self.unused_voucher.save()
tornado.httpclient.HTTPClient()
def get(self):
vid = self.get_argument('vid', None)
shake = None
promotion = None
voucher = None
user = self.get_current_user_object()
if vid is not None:
voucher = Voucher.get("id=%s", vid)
if voucher is not None and voucher.claimed_by_user_id == user.id:
if voucher.promotion_id:
promotion = Promotion.get("id=%s", voucher.promotion_id)
if promotion is not None:
shake = promotion.shake()
return self.render(
"account/confirm.html", promotion=promotion,
promotion_shake=shake, voucher=voucher, current_user_obj=user)
class VoucherTests(test.base.BaseAsyncTestCase):
def setUp(self):
super(VoucherTests, self).setUp()
self.admin = test.factories.user()
self.sign_in("admin", "password")
self.shake = Shake(user_id=self.admin.id,
name='promotion-shake',
title='Promotion Shake',
type='group')
self.shake.save()
self.expired_promotion = Promotion(name="Expired Promotion",
membership_months=60,
expires_at=datetime.utcnow() -
timedelta(seconds=50),
promotion_shake_id=0)
self.expired_promotion.save()
self.promotion = Promotion(name="Unit Test Sale",
membership_months=60,
promotion_shake_id=self.shake.id,
expires_at=datetime.utcnow() +
timedelta(seconds=60 * 60 * 24 * 365))
self.promotion.save()
self.used_voucher = Voucher(offered_by_user_id=0,
promotion_id=self.promotion.id,
voucher_key="abc123")
# apply_to_user saves the voucher object (because it touches the
# claimed_by_user_id and dates) and also the user object (by
# updating the is_paid status)
self.used_voucher.apply_to_user(self.admin)
self.unused_voucher = Voucher(offered_by_user_id=0,
claimed_by_user_id=0,
promotion_id=self.promotion.id,
voucher_key="unclaimed")
self.unused_voucher.save()
tornado.httpclient.HTTPClient()
def test_create_account_has_code_field(self):
"""
Create account page should display the discount
code field for new user registration.
"""
self.sign_out()
response = self.fetch_url("/create-account")
self.assertEqual(200, response.code)
self.assertTrue(response.body.find("Discount code:") > -1)
def test_create_account_with_bad_voucher(self):
"""
Submitting an invalid discount code should result in an error.
"""
self.sign_out()
arguments = self._valid_arguments()
arguments["key"] = "ABCDEFGHIJKL"
response = self.post_url("/create-account", arguments=arguments)
self.assertEqual(200, response.code)
self.assertTrue(response.body.find("Invalid discount code") > -1)
def test_create_account_with_unrecognized_voucher(self):
"""
Submitting an invalid discount code should result in an error.
"""
self.sign_out()
arguments = self._valid_arguments()
arguments["key"] = "foobar"
response = self.post_url("/create-account", arguments=arguments)
self.assertEqual(200, response.code)
self.assertTrue(response.body.find("Invalid discount code") > -1)
def test_create_account_with_good_voucher(self):
"""
Submitting a valid discount code should create an account
with appropriate credit applied.
"""
self.sign_out()
arguments = self._valid_arguments()
arguments["key"] = "unclaimed"
# this will create the account, but the redirect doesn't
# carry forward the session info, so sign in separately
# then verify the confirm-account page displays with a greeting
self.post_url("/create-account", arguments=arguments)
self.sign_in(arguments["name"], arguments["password"])
response = self.fetch_url("/confirm-account")
self.assertTrue(
response.body.find("Hello, %s!" % arguments["name"]) > -1)
response = self.fetch_url("/account/settings")
self.assertTrue(response.body.find("5 Years") > -1)
def test_settings_page_with_credit(self):
"""
Validate that our test user account is showing a 5 year
credit on their account settings.
"""
response = self.fetch_url("/account/settings")
self.assertTrue(response.body.find("5 Years") > -1)
def test_redeem_page_with_pro_user(self):
"""
A pro member shouldn't have access to the redeem page.
"""
response = self.fetch_url("/account/redeem")
self.assertTrue(response.body.find("Redeem a Coupon") == -1)
def test_redeem_voucher_with_bad_voucher(self):
self.sign_out()
user = test.factories.user(name="free_user",
email="*****@*****.**",
is_paid=0)
self.sign_in(user.name, "password")
response = self.fetch_url("/account/settings")
# verify this account is currently free
self.assertTrue(
response.body.find("You are currently using a free account.") > -1)
arguments = {"key": "abc123"}
response = self.post_url("/account/redeem", arguments)
self.assertTrue(response.body.find("Invalid") > -1)
def test_redeem_voucher_with_good_voucher(self):
self.sign_out()
user = test.factories.user(name="free_user",
email="*****@*****.**")
user.is_paid = 0
user.save()
self.sign_in(user.name, "password")
response = self.fetch_url("/account/settings")
# verify this account is currently free
self.assertTrue(
response.body.find("You are currently using a free account.") > -1)
arguments = {"key": "unclaimed"}
# this will post and redirect to the settings page which should
# then reflect that we are a paid user with 5 years of credit
response = self.post_url("/account/redeem", arguments)
self.assertTrue(response.body.find("5 Years") > -1)
payments = PaymentLog.where("user_id=%s", user.id)
self.assertEquals(len(payments), 1)
self.assertEquals(payments[0].operation, "redeem")
self.assertEquals(payments[0].status, "credit")
self.assertEquals(payments[0].reference_id, str(self.promotion.id))
self.assertEquals(payments[0].transaction_id, arguments['key'])
self.assertEquals(payments[0].buyer_email, user.email)
self.assertEquals(payments[0].buyer_name, user.name)
# self.assertEquals(payments[0].next_transaction_date, )
voucher = Voucher.get("claimed_by_user_id=%s", user.id)
self.assertEquals(voucher.promotion_id, self.promotion.id)
self.assertEquals(voucher.claimed_by_user_id, user.id)
self.assertEquals(voucher.offered_by_user_id, self.admin.id)
def test_active_promotion_list(self):
promotions = Promotion.active()
self.assertEqual(len(promotions), 1)
self.assertEqual(promotions[0].id, self.promotion.id)
def _valid_arguments(self):
return {
"name": "valid_user",
"password": "******",
"password_again": "asdfasdf",
"email": "*****@*****.**",
"_skip_recaptcha_test_only": True,
}
