3
3
4
4
from .call_handler import CallHandler
5
5
from .spy import SpyCreator
6
- from .spy_events import WhenRehearsal , PropAccessType , SpyEvent , SpyPropAccess
6
+ from .spy_events import WhenRehearsal , PropAccessType , SpyEvent , SpyInfo , SpyPropAccess
7
7
from .spy_log import SpyLog
8
8
from .stub_store import StubBehavior , StubStore
9
9
from .types import ContextValueT , ReturnT
@@ -65,20 +65,15 @@ def verify(
65
65
count = len (_rehearsals ),
66
66
ignore_extra_args = ignore_extra_args ,
67
67
)
68
- calls = self ._spy_log .get_calls_to_verify ([r .spy_id for r in rehearsals ])
68
+ calls = self ._spy_log .get_calls_to_verify ([r .spy . id for r in rehearsals ])
69
69
70
70
self ._verifier .verify (rehearsals = rehearsals , calls = calls , times = times )
71
71
72
72
def prop (self , _rehearsal : ReturnT ) -> "PropCore" :
73
73
"""Get a property setter/deleter rehearser."""
74
- spy_id , spy_name , payload = self ._spy_log .consume_prop_rehearsal ()
74
+ spy , payload = self ._spy_log .consume_prop_rehearsal ()
75
75
76
- return PropCore (
77
- spy_id = spy_id ,
78
- spy_name = spy_name ,
79
- prop_name = payload .prop_name ,
80
- spy_log = self ._spy_log ,
81
- )
76
+ return PropCore (spy = spy , prop_name = payload .prop_name , spy_log = self ._spy_log )
82
77
83
78
def reset (self ) -> None :
84
79
"""Reset and remove all stored spies and stubs."""
@@ -134,21 +129,18 @@ class PropCore:
134
129
135
130
def __init__ (
136
131
self ,
137
- spy_id : int ,
138
- spy_name : str ,
132
+ spy : SpyInfo ,
139
133
prop_name : str ,
140
134
spy_log : SpyLog ,
141
135
) -> None :
142
- self ._spy_id = spy_id
143
- self ._spy_name = spy_name
136
+ self ._spy = spy
144
137
self ._prop_name = prop_name
145
138
self ._spy_log = spy_log
146
139
147
140
def set (self , value : Any ) -> None :
148
141
"""Create a property setter rehearsal."""
149
142
event = SpyEvent (
150
- spy_id = self ._spy_id ,
151
- spy_name = self ._spy_name ,
143
+ spy = self ._spy ,
152
144
payload = SpyPropAccess (
153
145
prop_name = self ._prop_name ,
154
146
access_type = PropAccessType .SET ,
@@ -160,8 +152,7 @@ def set(self, value: Any) -> None:
160
152
def delete (self ) -> None :
161
153
"""Create a property deleter rehearsal."""
162
154
event = SpyEvent (
163
- spy_id = self ._spy_id ,
164
- spy_name = self ._spy_name ,
155
+ spy = self ._spy ,
165
156
payload = SpyPropAccess (
166
157
prop_name = self ._prop_name ,
167
158
access_type = PropAccessType .DELETE ,
0 commit comments