def test_sucesso(self):
usuario_logado = Usuario()
usuario_logado.put()
handler = Mock()
crud.salvar(handler, usuario_logado, 'INFORMATICA', 'BASICA')
cursos = Curso.query().fetch()
self.assertEqual(1, len(cursos))
curso = cursos[0]
self.assertEqual('INFORMATICA', curso.nome)
self.assertEqual('BASICA', curso.descricao)
handler.redirect.assert_called_once_with('/curso')
def execute(next_process, handler, dependencies, **kwargs):
user = users.get_current_user()
if user:
google_id = user.user_id()
query = Usuario.query_by_google(google_id)
usuario_logado = query.get()
if not usuario_logado:
usuario_logado = Usuario(nome=user.nickname(),
email=user.email(),
google_id=google_id)
usuario_logado.put()
logout_url = users.create_logout_url('/')
dependencies['usuario_logado'] = usuario_logado
dependencies['logout_url'] = logout_url
else:
dependencies['usuario_logado'] = False
dependencies['login_url'] = users.create_login_url('/')
next_process(dependencies, **kwargs)
def registrar_colegio():
if request.method == "POST":
data = request.json
usuario = Usuario.query.filter_by(
correo=data.get("colegio").get("correo")).first()
if usuario:
return "Correo ya registrado en el sistema."
else:
hashed_pass = generate_password_hash(
data.get("colegio").get("password"))
usuario = Usuario(correo=data.get("colegio").get("correo"),
password=hashed_pass,
id_tipo_usuario=2)
db.session.add(usuario)
db.session.flush()
colegio = Colegio(nombre=data.get("colegio").get("nombre"),
id_usuario=usuario.id)
db.session.add(colegio)
db.session.commit()
return "ok"
def registro_aspirante():
error = None
mensaje = None
colegios = db.session.query(Colegio, Usuario).filter(Colegio.id_usuario == Usuario.id, Usuario.live == True,
Usuario.id_tipo_usuario == 2).all()
if request.method == "POST":
form = request.form
usuario = Usuario.query.filter_by(correo=form["correo"]).first()
if usuario:
error = "Usuario ya registrado."
else:
hashed_pass = generate_password_hash(form["password"])
usuario = Usuario(
correo=form["correo"],
password=hashed_pass,
id_tipo_usuario=1
)
db.session.add(usuario)
db.session.flush()
aspirante = Aspirante(
nombres=form["nombres"],
apellidos=form["apellidos"],
id_usuario=usuario.id
)
db.session.add(aspirante)
db.session.flush()
if form["colegio"] != '0':
colegio = Colegio.query.filter_by(id=form["colegio"]).first()
aspirante_colegio = AspiranteColegio(
id_colegio=colegio.id,
id_aspirante=aspirante.id
)
db.session.add(aspirante_colegio)
db.session.flush()
mensaje = "Datos guardados en el sistema."
db.session.commit()
return render_template('aspirante/registro_aspirante.html', mensaje=mensaje, error=error, colegios=colegios)
def setUp(self):
prepare_to_test()
Usuario(nome="liu", email="*****@*****.**").put()