def test_jira_notifier_pulls_creds_from_secret(monkeypatch):
client = MagicMock()
jira = MagicMock(client=client)
monkeypatch.setattr("jira.JIRA", jira)
state = Failed(message="1", result=0)
with set_temporary_config({"cloud.use_local_secrets": True}):
with prefect.context(
secrets=dict(
JIRASECRETS={
"JIRAUSER": "******",
"JIRATOKEN": "",
"JIRASERVER": "https://foo/bar",
}
)
):
jira_notifier(
Task(),
"",
state,
options={"project": "TEST", "issuetype": {"name": "Task"}},
)
with pytest.raises(ValueError, match="JIRASECRETS"):
jira_notifier(
Task(),
"",
state,
options={"project": "TEST", "issuetype": {"name": "Task"}},
)
kwargs = jira.call_args[1]
assert kwargs == {
"basic_auth": ("Bob", ""),
"options": {"server": "https://foo/bar"},
}
def test_jira_notifier_is_curried_and_uses_only_states(monkeypatch, state):
state = state()
client = MagicMock()
jiraMock = MagicMock(client=client)
monkeypatch.setattr(
"prefect.utilities.notifications.jira_notification.JIRA", jiraMock)
handler = jira_notifier(only_states=[TriggerFailed])
with set_temporary_config({"cloud.use_local_secrets": True}):
with prefect.context(secrets=dict(
JIRASECRETS={
"JIRAUSER": "******",
"JIRATOKEN": "",
"JIRASERVER": ""
})):
returned = handler(
Task(),
"",
state,
options={
"project": "TEST",
"issuetype": {
"name": "Task"
}
},
)
assert returned is state
assert jiraMock.called is isinstance(state, TriggerFailed)
def test_jira_notifier_ignores_ignore_states(monkeypatch):
all_states = [
Running,
Pending,
Finished,
Failed,
TriggerFailed,
Cached,
Scheduled,
Retrying,
Success,
Skipped,
]
client = MagicMock()
jiraMock = MagicMock(client=client)
monkeypatch.setattr("jira.JIRA", jiraMock)
for state in all_states:
s = state()
with set_temporary_config({"cloud.use_local_secrets": True}):
with prefect.context(
secrets=dict(
JIRASECRETS={"JIRAUSER": "******", "JIRATOKEN": "", "JIRASERVER": ""}
)
):
returned = jira_notifier(
Task(),
"",
s,
ignore_states=[State],
options={"project": "TEST", "issuetype": {"name": "Task"}},
)
assert returned is s
assert jiraMock.called is False
def test_jira_notifier_returns_new_state_and_old_state_is_ignored(monkeypatch):
client = MagicMock()
jira = MagicMock(client=client)
monkeypatch.setattr("jira.JIRA", jira)
new_state = Failed(message="1", result=0)
with set_temporary_config({"cloud.use_local_secrets": True}):
with prefect.context(
secrets=dict(
JIRASECRETS={"JIRAUSER": "******", "JIRATOKEN": "123", "JIRASERVER": ""}
)
):
assert (
jira_notifier(
Task(),
"",
new_state,
options={"project": "TEST", "issuetype": {"name": "Task"}},
)
is new_state
)
def test_jira_notifier_is_curried_and_ignores_ignore_states(monkeypatch, state):
state = state()
client = MagicMock()
jiraMock = MagicMock(client=client)
monkeypatch.setattr("jira.JIRA", jiraMock)
handler = jira_notifier(ignore_states=[Finished])
with set_temporary_config({"cloud.use_local_secrets": True}):
with prefect.context(
secrets=dict(
JIRASECRETS={"JIRAUSER": "******", "JIRATOKEN": "", "JIRASERVER": ""}
)
):
returned = handler(
Task(),
"",
state,
options={"project": "TEST", "issuetype": {"name": "Task"}},
)
assert returned is state
assert jiraMock.called is not state.is_finished()