def salvar(handler,nome, sigla, ddd,id=None): ddd=long(ddd) if id: estado=Estado(id=long(id),nome=nome,sigla=sigla,ddd=ddd) else: estado=Estado(nome=nome,sigla=sigla,ddd=ddd) estado.put() handler.redirect(router.to_path(listar))
def test_editar(self): def write_tmpl(tmpl_name, dct): tmpl.render(tmpl_name, dct) salvo=Estado(nome="1",sigla="2",ddd=1) salvo.put() handler = HandlerMock() id=salvo.key.id() estado.salvar(handler, "Sao Paulo", "SP", "11",str(id)) editado=salvo.key.get() self.assertEqual("Sao Paulo", editado.nome) self.assertEqual("SP", editado.sigla) self.assertEqual(11, editado.ddd)