def test_generate_supervisor_conf_with_token_and_key(): # Covers deprecated token based auth colliding with key based auth agent = LocalAgent() with pytest.raises(ValueError, match="Given both a API token and API key"): agent.generate_supervisor_conf( token="token", key="key", labels=["label"], import_paths=["path"], env_vars={"TESTKEY": "TESTVAL"}, )
def test_generate_supervisor_conf(cloud_api): agent = LocalAgent() conf = agent.generate_supervisor_conf(token="token", labels=["label"], import_paths=["path"]) assert "-t token" in conf assert "-l label" in conf assert "-p path" in conf
def install(label, env, import_paths, **kwargs): """Generate a supervisord.conf file for a Local agent""" from prefect.agent.local import LocalAgent conf = LocalAgent.generate_supervisor_conf( labels=sorted(set(label)), env_vars=dict(e.split("=", 1) for e in env), import_paths=list(import_paths), **kwargs, ) click.echo(conf)
def test_generate_supervisor_conf(): agent = LocalAgent() conf = agent.generate_supervisor_conf( token="token", labels=["label"], import_paths=["path"], env_vars={"TESTKEY": "TESTVAL"}, ) assert "-t token" in conf assert "-l label" in conf assert "-p path" in conf assert "-e TESTKEY=TESTVAL" in conf
def test_generate_supervisor_conf_with_token(): # Covers deprecated token based auth agent = LocalAgent() conf = agent.generate_supervisor_conf( token="token", labels=["label"], import_paths=["path"], env_vars={"TESTKEY": "TESTVAL"}, ) assert "-t token" in conf assert "-l label" in conf assert "-p path" in conf assert "-e TESTKEY=TESTVAL" in conf
def test_generate_supervisor_conf_with_key(): agent = LocalAgent() conf = agent.generate_supervisor_conf( key="key", tenant_id="tenant", labels=["label"], import_paths=["path"], env_vars={"TESTKEY": "TESTVAL"}, ) assert "-k key" in conf assert "--tenant-id tenant" in conf assert "-l label" in conf assert "-p path" in conf assert "-e TESTKEY=TESTVAL" in conf