tests: Reduce db initialization fixtures to a minimum

This commit is contained in:
Antoine R. Dumont (@ardumont) 2020-10-29 17:37:45 +01:00
parent b35dff1266
commit b90ffa4bdd
No known key found for this signature in database
GPG key ID: 52E2E9840D10C3B8
3 changed files with 14 additions and 34 deletions

View file

@ -9,14 +9,12 @@ import os
from unittest.mock import patch
import pytest
from sqlalchemy.engine import create_engine
from swh.lister import get_lister
from swh.lister.core.models import initialize
@pytest.fixture
def lister_launchpad(datadir, lister_db_url, swh_scheduler):
def lister_launchpad(datadir, lister_db_url, engine, swh_scheduler):
class Collection:
entries = []
@ -42,11 +40,8 @@ def lister_launchpad(datadir, lister_db_url, swh_scheduler):
with patch("launchpadlib.launchpad.Launchpad.login_anonymously"):
lister = get_lister("launchpad", db_url=lister_db_url)
lister.scheduler = swh_scheduler # inject scheduler fixture
lister.launchpad.git_repositories.getRepositories.side_effect = [
mock_lp_response(i) for i in range(3)
]
initialize(create_engine(lister_db_url), drop_tables=True)
return lister