[Tests] Fix error in TestJSONRequestFailed test

Fix test generating following error

    'TestJSONRequestFailed' object has no attribute 'core'
This commit is contained in:
Calum Lind 2022-02-20 14:29:03 +00:00
parent 75b27485e1
commit 7b1a0ef89c
No known key found for this signature in database
GPG Key ID: 90597A687B836BA3
1 changed files with 6 additions and 5 deletions

View File

@ -11,6 +11,7 @@ from unittest.mock import MagicMock
import pytest import pytest
import pytest_twisted import pytest_twisted
from twisted.internet.defer import Deferred
from twisted.web import server from twisted.web import server
from twisted.web.http import Request from twisted.web.http import Request
@ -195,7 +196,6 @@ class TestJSONRequestFailed(WebServerMockBase):
test = TestClass() test = TestClass()
daemon.rpcserver.register_object(test) daemon.rpcserver.register_object(test)
""" """
from twisted.internet.defer import Deferred
extra_callback = { extra_callback = {
'deferred': Deferred(), 'deferred': Deferred(),
@ -210,18 +210,19 @@ class TestJSONRequestFailed(WebServerMockBase):
} }
def on_test_raise(*args): def on_test_raise(*args):
assert 'Unhandled error in Deferred:' in self.core.stderr_out assert 'Unhandled error in Deferred:' in daemon.stderr_out
assert 'in test_raise_error' in self.core.stderr_out assert 'in test_raise_error' in daemon.stderr_out
extra_callback['deferred'].addCallback(on_test_raise)
d, daemon = common.start_core( d, daemon = common.start_core(
custom_script=custom_script, custom_script=custom_script,
print_stdout=False, print_stdout=True,
print_stderr=False, print_stderr=False,
timeout=5, timeout=5,
extra_callbacks=[extra_callback], extra_callbacks=[extra_callback],
config_directory=config_dir, config_directory=config_dir,
) )
extra_callback['deferred'].addCallback(on_test_raise, daemon)
await d await d
yield yield
await daemon.kill() await daemon.kill()