Move RequiredEventNotFoundError class to Processor class

This commit is contained in:
Christophe Bedard 2019-12-31 19:34:35 -05:00
parent d0229a0bed
commit f176cea4e3
2 changed files with 7 additions and 7 deletions

View file

@ -19,7 +19,6 @@ import unittest
from tracetools_analysis.processor import EventHandler from tracetools_analysis.processor import EventHandler
from tracetools_analysis.processor import EventMetadata from tracetools_analysis.processor import EventMetadata
from tracetools_analysis.processor import Processor from tracetools_analysis.processor import Processor
from tracetools_analysis.processor import RequiredEventNotFoundError
class StubHandler1(EventHandler): class StubHandler1(EventHandler):
@ -149,7 +148,7 @@ class TestProcessor(unittest.TestCase):
'cpu_id': 0, 'cpu_id': 0,
} }
# Fails check # Fails check
with self.assertRaises(RequiredEventNotFoundError): with self.assertRaises(Processor.RequiredEventNotFoundError):
Processor(EventHandlerWithRequiredEvent()).process([mock_event]) Processor(EventHandlerWithRequiredEvent()).process([mock_event])
required_mock_event = { required_mock_event = {

View file

@ -253,13 +253,14 @@ class DependencySolver():
visited.add(dep_type) visited.add(dep_type)
class RequiredEventNotFoundError(RuntimeError):
pass
class Processor(): class Processor():
"""Processor class, which dispatches events to event handlers.""" """Processor class, which dispatches events to event handlers."""
class RequiredEventNotFoundError(RuntimeError):
"""When a trace does not contain one event required by an EventHandler."""
pass
def __init__( def __init__(
self, self,
*handlers: EventHandler, *handlers: EventHandler,
@ -339,7 +340,7 @@ class Processor():
for handler in self._expanded_handlers: for handler in self._expanded_handlers:
for name in handler.required_events(): for name in handler.required_events():
if name not in event_names: if name not in event_names:
raise RequiredEventNotFoundError( raise self.RequiredEventNotFoundError(
f'missing event {name} for {handler.__class__.__name__}' f'missing event {name} for {handler.__class__.__name__}'
) )