def create_new(): app.logger.debug("Form keys: " + ", ".join(request.form.keys())) if request.method == "GET": return render_template("new.html", user=slugify(request.args.get("user", "")), key=slugify(request.args.get("key", "")), code=request.args.get("code", "")) else: user = slugify(request.form.get("user", "Anonymous")) try: key = slugify(request.form.get("key", modelo.gen_key(user))) except KeyError: app.logger.error("Too many retries to generate a key") abort(500) code = request.form.get("code", "").strip() if (code is None or len(code) == 0): flash("No code to submit?") return redirect(url_for("create_new", user=user, key=key)) elif modelo.is_used_key(user, key): flash("Select another key, that one has already been taken!") return redirect(url_for("create_new", user=user, key=key, code=code)) else: modelo.add_pasta(user, key, code) return redirect(url_for("get_pasta", user=user, key=key))
def test_gen_key_error(): with patch('modelo.pastas.find_one') as p: p.return_value=1 something_random = "somethin_random" modelo.gen_key("me", lambda:something_random)
def test_gen_key(): with patch('modelo.pastas.find_one') as p: p.return_value=None something_random = "something_random" eq_(modelo.gen_key("me", lambda:something_random), something_random)