From 5c3f89cbc504a854bedf8728729d9890579ca21f Mon Sep 17 00:00:00 2001 From: Jan Bundesmann Date: Wed, 15 May 2024 11:33:53 +0200 Subject: [PATCH] Apply organizational changes in tests too --- tests/providers/test_foreman.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/providers/test_foreman.py b/tests/providers/test_foreman.py index f1ab0675..839210b4 100644 --- a/tests/providers/test_foreman.py +++ b/tests/providers/test_foreman.py @@ -3,18 +3,19 @@ from broker import Broker from broker.helpers import MockStub -from broker.providers.foreman import Foreman, ForemanAPI +from broker.providers.foreman import Foreman +from broker.binds.foreman import ForemanBind from broker.exceptions import ProviderError HOSTGROUP_VALID = "hg1" HOSTGROUP_INVALID = "hg7" -class ForemanApiStub(MockStub, ForemanAPI): +class ForemanApiStub(MockStub, ForemanBind): """Runtime to mock queries to Foreman.""" def __init__(self, **kwargs): MockStub.__init__(self, in_dict={}) - ForemanAPI.__init__(self) + ForemanBind.__init__(self) def _post(self, url, **kwargs): if "/api/job_invocations" in url: