Pergunta

Eu tenho um teste de integração do Flask apoiado por um mongodb de 1 nó que falha aleatoriamente:

pytest/test_webapi.py:59: in test_register_test
>           assert res.status_code == 302
E           assert <Response streamed [404 NOT FOUND]>.status_code == 302

A taxa de falha é de aproximadamente 50%.

Teste em test_webapi.py assim:

def test_register_user(self):
    res = self.client.get("/logout")

    class MySMTPServer(smtpd.SMTPServer):
        mails = []
        def process_message(self, peer, mailfrom, rcpttos, data):
            self.mails.append((rcpttos[0], data))

    server = MySMTPServer(('localhost', 12345), None)

    t = threading.Thread(target=asyncore.loop, args=(1,))
    t.start()
    time.sleep(.1)

    try:
        res = self.client.post("/register", data=self.registration)
        assert res.status_code == 200

        mail, hash = server.mails[0]

        self.conn.fsync()
        time.sleep(.1)

        res = self.client.get('/activate/' + hash)
        assert res.status_code == 302
    finally:
        server.close()

Os métodos Flask relevantes de webapi.py:

@app.route("/register", methods=["POST"])
def register_user():
    mail = flask.request.form['mail']
    user = flask.request.form["user"]
    pw = flask.request.form["pass"]

    hash = users.register(user, pw, mail=mail)
    return flask.jsonify({'_id': None}) # XXX

@app.route('/activate/<hash>', methods=['GET'])
def activate_user(hash):
    key = users.activate(hash=hash)
    if not key:
        flask.abort(404)
    return flask.redirect("/")

...são apoiados por métodos de ação:

make_key = lambda : base64.encodestring(os.urandom(32)).strip()

def register(self, user, pw, **kw):
    hash = self.make_key()
    user = self.new(user, pw, activation=hash, **kw)
    self._send_mail(**user)
    return hash

def activate(self, hash):
    user = self.users.find_one({'activation': hash})
    if not user:
        return None
    key = self.make_key()
    activation = {
        '$unset': {'activation': 1},
        '$set': {'status': 'active', 'key': key} }
    self.users.update({'_id': user['_id']}, activation)
    return user

...onde self.users é uma coleção mongodb.

self.new() persiste a entidade usando safe=True.

Curiosamente, vários outros testes que fazem coisas semelhantes parecem nunca encontrar esse problema.

Eu pensei que isso seria suficiente para garantir que o objeto persistido seria visível para outros threads no pool de conexões do pymongo.Que parte da documentação do mongodb/pymongo devo ler com mais atenção?Ou há alguma interação estranha com o assíncrono?

Foi útil?

Solução

(portado do tópico de comentários)

No contexto dos testes, é o data valoriza apenas a string codificada em base64 da chave de ativação?base64 contém letras e dígitos, mas também "+" e "/", ambos os quais serão mal interpretados pelos analisadores de URL (particularmente "/").

Licenciado em: CC-BY-SA com atribuição
Não afiliado a StackOverflow
scroll top