def test_dashboard(self):
app = create_testapp()
self.cpolicy.SetProbability(1.0)
# Pre-generate some coupons
ct = model.get_coupon_type("plane_long")
order = model.order_create("1", ct)
model.coupon_create(order)
model.coupon_spawn(model.get_coupon_type("training"), 3,
"*****@*****.**", "test")
resp = app.get("/admin")
self.assertEqual(resp.status, "200 OK")
self.assertIn("Statistika", resp)
def generate(self, data):
ct = model.get_coupon_type(data['coupon_type'])
coupons = model.coupon_spawn(ct, data['count'],
data['email'], data['notes'],
test=self.app.config['debug'])
for c in coupons:
send_confirmation_email(c)
def post(self, name):
ct = model.get_coupon_type(name)
assert ct.in_stock, "Cannot order this item"
order_id = model.order_gen_id()
order = model.order_create(order_id, ct, test=self.app.config['debug'])
data = self.prepare_webtopay_request(order, ct)
logging.info('Starting payment transaction for %s' % data)
url = webtopay.get_redirect_to_payment_url(data)
webapp2.redirect(url, abort=True)
def test_admin_check(self):
app = create_testapp()
self.cpolicy.SetProbability(1.0)
ct = model.get_coupon_type("plane_long")
order = model.order_create("1", ct)
coupons = model.coupon_create(order)
cid = coupons[0].coupon_id
resp = app.get("/admin/check/%s" % cid)
self.assertEqual(resp.status, "200 OK")
self.assertIn("Kvietimas galioja.", resp)
def test_qr(self):
app = create_testapp()
ct = model.get_coupon_type("plane_long")
order = model.order_create("1", ct)
coupons = model.coupon_create(order)
self.assertEquals(len(coupons), 1)
cid = coupons[0].coupon_id
resp = app.get("/qr/%s" % cid)
self.assertEqual(resp.status, "200 OK")
self.assertEqual(resp.headers['Content-Type'], 'image/png')
self.assertEqual(resp.body[:4], '\x89PNG')
def test_admin_check_use(self):
app = create_testapp()
self.cpolicy.SetProbability(1.0)
ct = model.get_coupon_type("plane_long")
order = model.order_create("1", ct)
coupons = model.coupon_create(order)
cid = coupons[0].coupon_id
resp = app.post("/admin/check/%s" % cid)
self.assertEqual(resp.status, "302 Moved Temporarily")
used = model.coupon_get(cid)
self.assertEqual(used.status, model.Coupon.ST_USED)
def format_coupon_type(value):
try:
ct = model.get_coupon_type(value)
return ct.title
except ValueError:
return '?'
def get(self, name):
ct = model.get_coupon_type(name)
self.show_form(ct)
