mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-08 16:53:23 +00:00
4a8e80ee3e
* Drop support for legacy Python 2.7 * Upgrade Python syntax with pyupgrade --py3-plus * Trim testing matrix to remove python2 * re-enable python3.8 in travis as tests that are not allowed to fail * Remove some six * The future is now * Remove Python 2 imports * Add back example, but change py27 to py36 * Remove redundant compat.py * Blacken * Credit hugovk in changelog WIP Updating Sphinx Docs and AutoDoc * Fix AutoDoc and update Sphinx theme to python_doc_theme * Fix #420, autodoc even undocumented (docstring-less) method signatures * Doc theme 'nature'. Add global TOC to doc sidebar * Comment last reference to package six * Changelog is now a consistent format * Yet another documentation fix for links and title hierarchy * Start work on new SVG logo test SVG in README trying to test new SVG logo in README Apply centering Apply readme logo centering Trying to align image Trying random shit trying align right add emoji Large logo has higher priority Change title hierarchy Actually use a H1 Try again try and organise badges revert link back to point at master * updated new take on VCR logo as SVG code * Testing modern logo in docs * Add sanitize for rendering SVG * Switch to alabaster theme * Update vcrpy logo (#503) * Add credit for V4 logo changes. * Add rewind and play animation * Add svg into ReadTheDocs static assets so that it can be hosted so the animations work. * Need to embedd the SVG for ReadTheDocs somewhere so I can get the link to later embed in the README Co-authored-by: Hugo van Kemenade <hugovk@users.noreply.github.com> Co-authored-by: Sean Bailey <sean@seanbailey.io>
56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Tests for cassettes with custom persistence"""
|
|
|
|
# External imports
|
|
import os
|
|
from urllib.request import urlopen
|
|
|
|
# Internal imports
|
|
import vcr
|
|
from vcr.persisters.filesystem import FilesystemPersister
|
|
|
|
|
|
class CustomFilesystemPersister(object):
|
|
"""Behaves just like default FilesystemPersister but adds .test extension
|
|
to the cassette file"""
|
|
|
|
@staticmethod
|
|
def load_cassette(cassette_path, serializer):
|
|
cassette_path += ".test"
|
|
return FilesystemPersister.load_cassette(cassette_path, serializer)
|
|
|
|
@staticmethod
|
|
def save_cassette(cassette_path, cassette_dict, serializer):
|
|
cassette_path += ".test"
|
|
FilesystemPersister.save_cassette(cassette_path, cassette_dict, serializer)
|
|
|
|
|
|
def test_save_cassette_with_custom_persister(tmpdir, httpbin):
|
|
"""Ensure you can save a cassette using custom persister"""
|
|
my_vcr = vcr.VCR()
|
|
my_vcr.register_persister(CustomFilesystemPersister)
|
|
|
|
# Check to make sure directory doesnt exist
|
|
assert not os.path.exists(str(tmpdir.join("nonexistent")))
|
|
|
|
# Run VCR to create dir and cassette file using new save_cassette callback
|
|
with my_vcr.use_cassette(str(tmpdir.join("nonexistent", "cassette.yml"))):
|
|
urlopen(httpbin.url).read()
|
|
|
|
# Callback should have made the file and the directory
|
|
assert os.path.exists(str(tmpdir.join("nonexistent", "cassette.yml.test")))
|
|
|
|
|
|
def test_load_cassette_with_custom_persister(tmpdir, httpbin):
|
|
"""
|
|
Ensure you can load a cassette using custom persister
|
|
"""
|
|
my_vcr = vcr.VCR()
|
|
my_vcr.register_persister(CustomFilesystemPersister)
|
|
|
|
test_fixture = str(tmpdir.join("synopsis.json.test"))
|
|
|
|
with my_vcr.use_cassette(test_fixture, serializer="json"):
|
|
response = urlopen(httpbin.url).read()
|
|
assert b"difficult sometimes" in response
|