forked from goodboy/tractor
				
			Add some bcaster ref sanity asserts around subscriptions
							parent
							
								
									0d70e3081a
								
							
						
					
					
						commit
						63ec740e27
					
				| 
						 | 
				
			
			@ -9,7 +9,7 @@ from dataclasses import dataclass
 | 
			
		|||
from typing import (
 | 
			
		||||
    Any, Iterator, Optional, Callable,
 | 
			
		||||
    AsyncGenerator, Dict,
 | 
			
		||||
    AsyncIterator, Awaitable
 | 
			
		||||
    AsyncIterator
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
import warnings
 | 
			
		||||
| 
						 | 
				
			
			@ -264,7 +264,7 @@ class ReceiveMsgStream(trio.abc.ReceiveChannel):
 | 
			
		|||
                self,
 | 
			
		||||
                # use memory channel size by default
 | 
			
		||||
                self._rx_chan._state.max_buffer_size,  # type: ignore
 | 
			
		||||
               receive_afunc=self.receive,
 | 
			
		||||
                receive_afunc=self.receive,
 | 
			
		||||
            )
 | 
			
		||||
 | 
			
		||||
            # NOTE: we override the original stream instance's receive
 | 
			
		||||
| 
						 | 
				
			
			@ -277,6 +277,8 @@ class ReceiveMsgStream(trio.abc.ReceiveChannel):
 | 
			
		|||
            # https://github.com/python/mypy/issues/708
 | 
			
		||||
 | 
			
		||||
        async with self._broadcaster.subscribe() as bstream:
 | 
			
		||||
            assert bstream.key != self._broadcaster.key
 | 
			
		||||
            assert bstream._recv == self._broadcaster._recv
 | 
			
		||||
            yield bstream
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue