def setup_config_file(tmp_path, config_brokers, config_topic, config_params, config_file):
cwd = Path.cwd().absolute()
try:
os.chdir(str(tmp_path))
create_config_file(config_brokers, config_topic, config_params, config=config_file)
yield
finally:
os.chdir(str(cwd))
def setup_bad_env_config(tmp_path, config_brokers, config_topic):
cwd = Path.cwd().absolute()
try:
os.chdir(str(tmp_path))
config = tmp_path / 'ss-config.yml'
create_config_file(brokers=config_brokers, topic=config_topic)
os.environ['SINETSTREAM_CONFIG_URL'] = config.as_uri()
yield
finally:
del (os.environ['SINETSTREAM_CONFIG_URL'])
os.chdir(str(cwd))
def create_topic():
cwd = Path.cwd().absolute()
with TemporaryDirectory() as work_dir:
try:
os.chdir(str(work_dir))
create_config_file(brokers=[BROKER])
with MessageWriter(SERVICE, TOPIC) as f:
logger.debug(f"create topic: {TOPIC}")
f.publish(b"message 000")
finally:
os.chdir(str(cwd))
def setup_home_config(tmp_path, config_brokers, config_topic):
cwd = Path.cwd().absolute()
config = Path('~/.config/sinetstream/config.yml').expanduser()
backup = None
try:
os.chdir(str(tmp_path))
if config.exists():
backup = tmp_path / 'config.yml'
copyfile(config, backup)
else:
config.parent.mkdir(parents=True, exist_ok=True)
create_config_file(brokers=config_brokers,
topic=config_topic,
config=config)
yield
finally:
os.chdir(str(cwd))
if backup:
copyfile(backup, config)
else:
config.unlink()