def test_log_visit(self): #due to the not so great nature of adding false entries to , #I'm grouping a few tests here less = model.db.query(func.count(Visit.id)).first() original = model.url_info(self.code) model.log_visit(self.code) greater = model.db.query(func.count(Visit.id)).first() new = model.url_info(self.code) self.assertGreater(greater, less) self.assertEqual(original.url, new.url)
def redirect_on_code(code): try: model.db_connect() model.log_visit(code) url = model.url_info(code) model.db.close() return redirect(url.url) except: model.db.close() return render_template('404.html'), 404
def test_url_info_visits(self): url = model.url_info(self.code) self.assertGreater(url.visits, 4)
def test_url_info_code(self): url = model.url_info(self.code) self.assertEqual(url.code, self.code)
def test_url_info_url(self): url = model.url_info(self.code) self.assertEqual(url.url, "https://www.heroku.com/")
def url_visits(code): model.db_connect() url=model.url_info(code) model.db.close() return render_template("visits.html", url=url)