def test_raises_if_message_not_provided(self, monkeypatch): task = PushbulletTask() client = MagicMock() pushbullet = MagicMock(client=client) monkeypatch.setattr("pushbullet.Pushbullet", pushbullet) with set_temporary_config({"cloud.use_local_secrets": True}): with prefect.context(secrets=dict(PUSHBULLET_TOKEN=42)): with pytest.raises(ValueError, match="message"): task.run()
def test_token_pulled_from_secrets(self, monkeypatch): task = PushbulletTask(msg="test") client = MagicMock() pushbullet = MagicMock(client=client) monkeypatch.setattr("pushbullet.Pushbullet", pushbullet) with set_temporary_config({"cloud.use_local_secrets": True}): with prefect.context(secrets=dict(PUSHBULLET_TOKEN=42)): task.run() kwargs = pushbullet.call_args[0] assert kwargs == (42, )
def test_raises_if_secret_not_provided(self): task = PushbulletTask() with pytest.raises(ValueError, match="Secret"): task.run()
def test_kwargs_get_passed_to_task_init(self): t = PushbulletTask(msg="Test", tags=["foo"]) assert t.msg == "Test" assert t.tags == {"foo"}
def test_inits_with_no_args(self): t = PushbulletTask() assert t