Skip to content

change_producer

faststream.nats.testing.change_producer #

change_producer(
    config: ConfigComposition[NatsBrokerConfig],
    producer: NatsFastProducer,
) -> Generator[None, None, None]
Source code in faststream/nats/testing.py 
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
@contextmanager
def change_producer(
    config: "ConfigComposition[NatsBrokerConfig]",
    producer: "NatsFastProducer",
) -> Generator[None, None, None]:
    old_producer, config.broker_config.producer = (
        config.broker_config.producer,
        producer,
    )
    old_js_producer, config.broker_config.js_producer = (
        config.broker_config.js_producer,
        producer,
    )
    yield
    config.broker_config.producer = old_producer
    config.broker_config.js_producer = old_js_producer