conftest.py
No OneTemporary
Actions

Size

1 KB

Subscribers

None

conftest.py
View Options

	# Copyright (C) 2020 The Software Heritage developers
	# See the AUTHORS file at the top-level directory of this distribution
	# License: GNU General Public License version 3, or any later version
	# See top-level LICENSE file for more information

	from datetime import datetime
	import json
	import os
	from unittest.mock import patch

	import pytest

	from swh.lister import get_lister


	@pytest.fixture
	def lister_launchpad(datadir, lister_db_url, engine, swh_scheduler):
	class Collection:
	entries = []

	def __init__(self, file):
	self.entries = [Repo(r) for r in file]

	def __getitem__(self, key):
	return self.entries[key]

	class Repo:
	def __init__(self, d: dict):
	for key in d.keys():
	if key == "date_last_modified":
	setattr(self, key, datetime.fromisoformat(d[key]))
	else:
	setattr(self, key, d[key])

	def mock_lp_response(page) -> Collection:
	response_filepath = os.path.join(datadir, f"response{page}.json")
	with open(response_filepath, "r", encoding="utf-8") as f:
	return Collection(json.load(f))

	with patch("swh.lister.launchpad.lister.Launchpad") as m:
	m.login_anonymously.return_value = m
	m.git_repositories.getRepositories.side_effect = [
	mock_lp_response(i) for i in range(3)
	]
	lister = get_lister("launchpad", db_url=lister_db_url)

	lister.scheduler.create_task_type(
	{
	"type": "load-git",
	"description": "Load git repository",
	"backend_name": "swh.loader.git.tasks.UpdateGitRepository",
	"default_interval": "1 day",
	}
	)

	return lister