def test_reads_with_new_name(self): task = PrefectSecret("foo") result = SecretResult(task) with prefect.context(secrets=dict(x=99, foo="bar")): res1 = result.read("x") res2 = result.read("foo") assert res1.value == 99 assert res1.location == "x" assert res2.value == "bar" assert res2.location == "foo"
def test_reads_by_rerunning_task(self): task = PrefectSecret("foo") task.run = lambda *args, **kwargs: 42 result = SecretResult(task) result.location == "foo" new_result = result.read("foo") assert new_result.value == 42 new_result.location == "foo"