def tearDown(self):
super().tearDown()
data_file_path = get_datafile_filename(self)
self.assertTrue(os.path.exists(data_file_path))
# use just class and test name instead of full ID
self.assertIn(".".join(self.id().split(".")[-2:]), data_file_path.name)
self.assertIn("test_data", str(data_file_path))
shutil.rmtree(os.path.dirname(get_datafile_filename(self)))
def remove_storage_file_after(request):
storage_file = get_datafile_filename(request.node)
yield storage_file
# TODO: found why this were necessary before change of singletons
# PersistentObjectStorage().cassette.dump()
if storage_file.is_file():
storage_file.unlink()
def setUp(self):
super().setUp()
self.token = os.environ.get("GITHUB_TOKEN")
if not Path(
get_datafile_filename(obj=self)).exists() and not self.token:
raise EnvironmentError(
"You are in Requre write mode, please set proper GITHUB_TOKEN env variables"
)
self._service = None
self._ogr_project = None
def setUp(self):
self.github_app_id = os.environ.get("GITHUB_APP_ID")
self.github_app_private_key_path = os.environ.get(
"GITHUB_APP_PRIVATE_KEY_PATH")
if not get_datafile_filename(obj=self) and (
not self.github_app_id
or not self.github_app_private_key_path):
raise EnvironmentError(
"You are in Requre write mode, please set "
"GITHUB_APP_ID GITHUB_APP_PRIVATE_KEY_PATH env variables")
def setUp(self):
super().setUp()
self.token = os.environ.get("PAGURE_OGR_TEST_TOKEN", "")
if not get_datafile_filename(obj=self) and (not self.token):
raise EnvironmentError("please set PAGURE_OGR_TEST_TOKEN env variables")
self._service = None
self._user = None
self._ogr_project = None
self._ogr_fork = None
def change_storage_file(cassette: Cassette,
func,
args,
storage_file: Optional[str] = None):
"""
Internal function that try to construct persistent data file based on various
possibilities.
:param cassette: Cassette instance to pass inside object to work with
"""
if storage_file:
cassette.storage_file = storage_file
else:
if len(args):
try:
cassette.storage_file = get_datafile_filename(args[0])
except NameError:
cassette.storage_file = get_datafile_filename(func)
else:
cassette.storage_file = get_datafile_filename(func)
original_storage_file = cassette.storage_file
return original_storage_file
def record_requests_fixture(request):
storage_file = get_datafile_filename(request.node)
with recording_requests(storage_file=storage_file) as cassette:
mode_description = {
StorageMode.read: "replaying",
StorageMode.write: "recording",
StorageMode.append: "appending",
StorageMode.default: "in default mode",
}[cassette.mode]
logger.debug(
f"Start requre {mode_description} with storage file: {storage_file}"
)
yield cassette
cassette.dump()
logger.debug(
f"End requre {mode_description} with storage file: {storage_file}")
def setUp(self):
self._github_service = None
self._pagure_service = None
self._gitlab_service = None
self.github_token = os.environ.get("GITHUB_TOKEN")
self.pagure_token = os.environ.get("PAGURE_TOKEN")
self.gitlab_token = os.environ.get("GITLAB_TOKEN") or "some_token"
if not Path(get_datafile_filename(obj=self)).exists():
if (
not self.github_token
or not self.pagure_token
or not os.environ.get("GITLAB_TOKEN")
):
raise EnvironmentError(
"You are in requre write mode, please set GITHUB_TOKEN PAGURE_TOKEN"
" GITLAB_TOKEN env variables"
)
def setUp(self) -> None:
super().setUp()
self.cassette.storage_file = get_datafile_filename(self.id)
TempFile.set_cassette(self.cassette)
def remove_storage_file(request):
storage_file = get_datafile_filename(request.node)
if storage_file.is_file():
storage_file.unlink()
return storage_file
def setUp(self):
super().setUp()
response_file = get_datafile_filename(self)
os.makedirs(os.path.dirname(response_file), mode=0o777, exist_ok=True)
self.cassette = PersistentObjectStorage().cassette
self.cassette.storage_file = response_file