view tests/mockutil.py @ 51:b87217b90048

Cache Bluesky email before asking for password. It makes it easier to re-authenticate if the user entered the wrong password.
author Ludovic Chabant <ludovic@chabant.com>
date Sun, 08 Oct 2023 13:49:30 -0700
parents 27543b2e73b9
children b404445082f8
line wrap: on
line source


def mock_urllib(m):
    import urllib.request
    m.setattr(urllib.request, 'urlretrieve', _patched_urlretrieve)
    m.setattr(urllib.request, 'urlcleanup', _patched_urlcleanup)
    return m


def _patched_urlretrieve(url):
    return ('/retrieved/' + url.lstrip('/'), None)


def _patched_urlcleanup():
    pass