3 d@s~ddlmZddlmZddlmZddlmZejZ ddZ ddZ d d Z Gd d d Z Gd ddZGdddZdS))mock)add_or_remove_notifiers) ObserverGraph)NotifierNotFoundcCs ||dS)N)handlereventrr=/tmp/pip-build-7vycvbft/traits/traits/observation/_testing.py dispatch_samesr cGsD|d}t|d}x,|dddddD]}t||gd}q*W|S)z Create an ObserverGraph with the given nodes joined one after another. Parameters ---------- *nodes : hashable Items to be attached as nodes Returns ------- ObserverGraph )nodeN)r childrenrr)r)Znodesr graphrrr create_graphs  rcKs:ttjtddtjttdd}|j|tf|dS)z Convenience function for calling add_or_remove_notifiers with default values. Parameters ---------- **kwargs New argument values to use instead. N)r F)objectrrtarget dispatcherremove)dictrMockr_DEFAULT_TARGETr updater)kwargsvaluesrrr call_add_or_remove_notifiers-s  rc@s eZdZdZddZddZdS)DummyObservablez< A dummy implementation of IObservable for testing purposes.cCs g|_dS)N) notifiers)selfrrr __init__EszDummyObservable.__init__cCs|jS)N)r)rZ force_createrrr _notifiersHszDummyObservable._notifiersN)__name__ __module__ __qualname____doc__rr rrrr rBsrc@s eZdZdZddZddZdS) DummyNotifierz: A dummy implementation of INotifier for testing purposes.cCs|jdj|dS)NT)r append)r observablerrr add_toOszDummyNotifier.add_toc Cs:|jd}y|j|Wntk r4tdYnXdS)NTzNotifier not found.)r r ValueErrorr)rr'rrrr remove_fromRs  zDummyNotifier.remove_fromN)r!r"r#r$r(r*rrrr r%Lsr%c@s^eZdZdZdffddffddZddZdd Zd d Zd d ZddZ ddZ ddZ dS) DummyObserveraC A dummy implementation of IObserver for testing purposes. Parameters ---------- notify : boolean, optional The mocked return value from IObserver.notify observables : iterable of IObservable, optional The mocked yielded values from IObserver.iter_observables next_objects : iterable of object, optional The mocked yielded values from IObserver.iter_objects notifier : INotifier, optional The mocked returned value from IObserver.get_notifier If not provided, a dummy notifier is created. maintainer : INotifier, optional The mocked returned value from IObserver.get_maintainer if not provided, a dummy notifier is created. extra_graphs : iterable of ObserverGraph, optional The mocked yielded values from IObserver.iter_extra_graphs TNcCsD|dkrt}|dkrt}||_||_||_||_||_||_dS)N)r%notify observables next_objectsnotifier maintainer extra_graphs)rr,r-r.r/r0r1rrr ros zDummyObserver.__init__cCs||kS)Nr)rotherrrr __eq__szDummyObserver.__eq__cCsdS)Nr r)rrrr __hash__szDummyObserver.__hash__ccs|jEdHdS)N)r-)rrrrr iter_observablesszDummyObserver.iter_observablesccs|jEdHdS)N)r.)rrrrr iter_objectsszDummyObserver.iter_objectscCs|jS)N)r/)rrrrrrr get_notifierszDummyObserver.get_notifiercCs|jS)N)r0)rrrrrrrr get_maintainerszDummyObserver.get_maintainerccs|jEdHdS)N)r1)rrrrr iter_extra_graphsszDummyObserver.iter_extra_graphs) r!r"r#r$rr3r4r5r6r7r8r9rrrr r+Zs r+N)ZunittestrZtraits.observation._observerZ"traits.observation._observer_graphrZtraits.observation.exceptionsrrrr rrrr%r+rrrr  s