@@ -41,12 +41,12 @@ async def update_tracker_on_receive(receiver_id: int, chan: Receiver[int]) -> No
4141
4242 receivers = []
4343 for ctr in range (num_receivers ):
44- receivers .append (update_tracker_on_receive (ctr , bcast .get_receiver ()))
44+ receivers .append (update_tracker_on_receive (ctr , bcast .new_receiver ()))
4545
4646 receivers_runs = asyncio .gather (* receivers )
4747 senders = []
4848 for ctr in range (num_senders ):
49- senders .append (send_msg (bcast .get_sender ()))
49+ senders .append (send_msg (bcast .new_sender ()))
5050
5151 await asyncio .gather (* senders )
5252 await bcast .close ()
@@ -64,8 +64,8 @@ async def test_broadcast_after_close() -> None:
6464 """Ensure closed channels can't get new messages."""
6565 bcast : Broadcast [int ] = Broadcast ("meter_5" )
6666
67- receiver = bcast .get_receiver ()
68- sender = bcast .get_sender ()
67+ receiver = bcast .new_receiver ()
68+ sender = bcast .new_sender ()
6969
7070 await bcast .close ()
7171
@@ -80,10 +80,10 @@ async def test_broadcast_overflow() -> None:
8080
8181 big_recv_size = 10
8282 small_recv_size = int (big_recv_size / 2 )
83- sender = bcast .get_sender ()
83+ sender = bcast .new_sender ()
8484
85- big_receiver = bcast .get_receiver ("named-recv" , big_recv_size )
86- small_receiver = bcast .get_receiver (None , small_recv_size )
85+ big_receiver = bcast .new_receiver ("named-recv" , big_recv_size )
86+ small_receiver = bcast .new_receiver (None , small_recv_size )
8787
8888 async def drain_receivers () -> Tuple [int , int ]:
8989 big_sum = 0
@@ -129,11 +129,11 @@ async def test_broadcast_resend_latest() -> None:
129129 """Check if new receivers get the latest value when resend_latest is set."""
130130 bcast : Broadcast [int ] = Broadcast ("new_recv_test" , resend_latest = True )
131131
132- sender = bcast .get_sender ()
133- old_recv = bcast .get_receiver ()
132+ sender = bcast .new_sender ()
133+ old_recv = bcast .new_receiver ()
134134 for val in range (0 , 10 ):
135135 await sender .send (val )
136- new_recv = bcast .get_receiver ()
136+ new_recv = bcast .new_receiver ()
137137
138138 await sender .send (100 )
139139
@@ -146,11 +146,11 @@ async def test_broadcast_no_resend_latest() -> None:
146146 """Ensure new receivers don't get the latest value when resend_latest isn't set."""
147147 bcast : Broadcast [int ] = Broadcast ("new_recv_test" , resend_latest = False )
148148
149- sender = bcast .get_sender ()
150- old_recv = bcast .get_receiver ()
149+ sender = bcast .new_sender ()
150+ old_recv = bcast .new_receiver ()
151151 for val in range (0 , 10 ):
152152 await sender .send (val )
153- new_recv = bcast .get_receiver ()
153+ new_recv = bcast .new_receiver ()
154154
155155 await sender .send (100 )
156156
@@ -161,9 +161,9 @@ async def test_broadcast_no_resend_latest() -> None:
161161async def test_broadcast_peek () -> None :
162162 """Ensure we are able to peek into broadcast channels."""
163163 bcast : Broadcast [int ] = Broadcast ("peek-test" )
164- receiver = bcast .get_receiver ()
164+ receiver = bcast .new_receiver ()
165165 peekable = receiver .into_peekable ()
166- sender = bcast .get_sender ()
166+ sender = bcast .new_sender ()
167167
168168 with pytest .raises (EOFError ):
169169 await receiver .receive ()
@@ -188,8 +188,8 @@ async def test_broadcast_async_iterator() -> None:
188188 """Check that the broadcast receiver works as an async iterator."""
189189 bcast : Broadcast [int ] = Broadcast ("iter_test" )
190190
191- sender = bcast .get_sender ()
192- receiver = bcast .get_receiver ()
191+ sender = bcast .new_sender ()
192+ receiver = bcast .new_receiver ()
193193
194194 async def send_values () -> None :
195195 for val in range (0 , 10 ):
@@ -210,10 +210,10 @@ async def send_values() -> None:
210210async def test_broadcast_map () -> None :
211211 """Ensure map runs on all incoming messages."""
212212 chan = Broadcast [int ]("input-chan" )
213- sender = chan .get_sender ()
213+ sender = chan .new_sender ()
214214
215215 # transform int receiver into bool receiver.
216- receiver : Receiver [bool ] = chan .get_receiver ().map (lambda num : num > 10 )
216+ receiver : Receiver [bool ] = chan .new_receiver ().map (lambda num : num > 10 )
217217
218218 await sender .send (8 )
219219 await sender .send (12 )
@@ -225,10 +225,10 @@ async def test_broadcast_map() -> None:
225225async def test_broadcast_receiver_drop () -> None :
226226 """Ensure deleted receivers get cleaned up."""
227227 chan = Broadcast [int ]("input-chan" )
228- sender = chan .get_sender ()
228+ sender = chan .new_sender ()
229229
230- receiver1 = chan .get_receiver ()
231- receiver2 = chan .get_receiver ()
230+ receiver1 = chan .new_receiver ()
231+ receiver2 = chan .new_receiver ()
232232
233233 await sender .send (10 )
234234
0 commit comments