Python Session.get_totp_secret Examples

Example #1

File: test_sessions.py Project: ofek/warehouse

    def test_get_totp_secret_empty(self, monkeypatch):
        generate_totp_secret = pretend.call_recorder(lambda: b"foobar")
        monkeypatch.setattr(otp, "generate_totp_secret", generate_totp_secret)

        session = Session()
        assert session.get_totp_secret() == b"foobar"
        assert session._totp_secret_key in session

Example #2

File: test_sessions.py Project: ofek/warehouse

    def test_get_totp_secret(self, monkeypatch):
        session = Session()
        session[session._totp_secret_key] = b"foobar"

        assert session.get_totp_secret() == b"foobar"