summaryrefslogtreecommitdiff
path: root/tests/generate_test_db.py
blob: bd7f09a59a12fce06012e5f566cf1ec25f692672 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
#!/usr/bin/env python3
# flake8: noqa: E402
"""Generate a SQLite cache with some content for testing behavior during version upgrades"""
import sys
from os.path import abspath, join

sys.path.insert(0, abspath('.'))

from requests_cache import CachedSession, __version__
from tests.conftest import HTTPBIN_FORMATS, SAMPLE_DATA_DIR

DB_PATH = join(SAMPLE_DATA_DIR, f'sample.db.{__version__}')


def make_sample_db():
    session = CachedSession(DB_PATH)

    for format in HTTPBIN_FORMATS:
        session.get(f'https://httpbin.org/{format}')
    print(session.cache.urls())


if __name__ == '__main__':
    make_sample_db()