mirror of
https://github.com/ansible/awx.git
synced 2026-06-25 08:28:03 -02:30
fccb6744f9
* Address even more pytest warnings, co-authored with Opus 4.6 * Upgrade pyparsing * Attempt to update smart inventory logic * Move smart inventory tests here * Fix some failing dev env tests Assisted-by: claude * Use shared fixture for teardown * Fix test goof assisted-by: claude Opus 4.6
219 lines
8.6 KiB
Python
219 lines
8.6 KiB
Python
from unittest import mock
|
|
import datetime as dt
|
|
from django.core.mail.message import EmailMessage
|
|
import pytest
|
|
|
|
import awx.main.notifications.grafana_backend as grafana_backend
|
|
|
|
|
|
def test_send_messages():
|
|
with mock.patch('awx.main.notifications.grafana_backend.requests') as requests_mock:
|
|
requests_mock.post.return_value.status_code = 200
|
|
m = {}
|
|
m['started'] = dt.datetime.fromtimestamp(60, tz=dt.timezone.utc).isoformat()
|
|
m['finished'] = dt.datetime.fromtimestamp(120, tz=dt.timezone.utc).isoformat()
|
|
m['subject'] = "test subject"
|
|
backend = grafana_backend.GrafanaBackend("testapikey", dashboardId='', panelId='')
|
|
message = EmailMessage(
|
|
m['subject'],
|
|
{"started": m['started'], "finished": m['finished']},
|
|
[],
|
|
[
|
|
'https://example.com',
|
|
],
|
|
)
|
|
sent_messages = backend.send_messages(
|
|
[
|
|
message,
|
|
]
|
|
)
|
|
requests_mock.post.assert_called_once_with(
|
|
'https://example.com/api/annotations',
|
|
headers={'Content-Type': 'application/json', 'Authorization': 'Bearer testapikey'},
|
|
json={'text': 'test subject', 'isRegion': True, 'timeEnd': 120000, 'time': 60000},
|
|
verify=True,
|
|
)
|
|
assert sent_messages == 1
|
|
|
|
|
|
def test_send_messages_with_no_verify_ssl():
|
|
with mock.patch('awx.main.notifications.grafana_backend.requests') as requests_mock:
|
|
requests_mock.post.return_value.status_code = 200
|
|
m = {}
|
|
m['started'] = dt.datetime.fromtimestamp(60, tz=dt.timezone.utc).isoformat()
|
|
m['finished'] = dt.datetime.fromtimestamp(120, tz=dt.timezone.utc).isoformat()
|
|
m['subject'] = "test subject"
|
|
backend = grafana_backend.GrafanaBackend("testapikey", dashboardId='', panelId='', grafana_no_verify_ssl=True)
|
|
message = EmailMessage(
|
|
m['subject'],
|
|
{"started": m['started'], "finished": m['finished']},
|
|
[],
|
|
[
|
|
'https://example.com',
|
|
],
|
|
)
|
|
sent_messages = backend.send_messages(
|
|
[
|
|
message,
|
|
]
|
|
)
|
|
requests_mock.post.assert_called_once_with(
|
|
'https://example.com/api/annotations',
|
|
headers={'Content-Type': 'application/json', 'Authorization': 'Bearer testapikey'},
|
|
json={'text': 'test subject', 'isRegion': True, 'timeEnd': 120000, 'time': 60000},
|
|
verify=False,
|
|
)
|
|
assert sent_messages == 1
|
|
|
|
|
|
@pytest.mark.parametrize("dashboardId", [42, 0])
|
|
def test_send_messages_with_dashboardid(dashboardId):
|
|
with mock.patch('awx.main.notifications.grafana_backend.requests') as requests_mock:
|
|
requests_mock.post.return_value.status_code = 200
|
|
m = {}
|
|
m['started'] = dt.datetime.fromtimestamp(60, tz=dt.timezone.utc).isoformat()
|
|
m['finished'] = dt.datetime.fromtimestamp(120, tz=dt.timezone.utc).isoformat()
|
|
m['subject'] = "test subject"
|
|
backend = grafana_backend.GrafanaBackend("testapikey", dashboardId=dashboardId, panelId='')
|
|
message = EmailMessage(
|
|
m['subject'],
|
|
{"started": m['started'], "finished": m['finished']},
|
|
[],
|
|
[
|
|
'https://example.com',
|
|
],
|
|
)
|
|
sent_messages = backend.send_messages(
|
|
[
|
|
message,
|
|
]
|
|
)
|
|
requests_mock.post.assert_called_once_with(
|
|
'https://example.com/api/annotations',
|
|
headers={'Content-Type': 'application/json', 'Authorization': 'Bearer testapikey'},
|
|
json={'text': 'test subject', 'isRegion': True, 'timeEnd': 120000, 'time': 60000, 'dashboardId': dashboardId},
|
|
verify=True,
|
|
)
|
|
assert sent_messages == 1
|
|
|
|
|
|
@pytest.mark.parametrize("panelId", ['42', '0'])
|
|
def test_send_messages_with_panelid(panelId):
|
|
with mock.patch('awx.main.notifications.grafana_backend.requests') as requests_mock:
|
|
requests_mock.post.return_value.status_code = 200
|
|
m = {}
|
|
m['started'] = dt.datetime.fromtimestamp(60, tz=dt.timezone.utc).isoformat()
|
|
m['finished'] = dt.datetime.fromtimestamp(120, tz=dt.timezone.utc).isoformat()
|
|
m['subject'] = "test subject"
|
|
backend = grafana_backend.GrafanaBackend("testapikey", dashboardId='', panelId=panelId)
|
|
message = EmailMessage(
|
|
m['subject'],
|
|
{"started": m['started'], "finished": m['finished']},
|
|
[],
|
|
[
|
|
'https://example.com',
|
|
],
|
|
)
|
|
sent_messages = backend.send_messages(
|
|
[
|
|
message,
|
|
]
|
|
)
|
|
requests_mock.post.assert_called_once_with(
|
|
'https://example.com/api/annotations',
|
|
headers={'Content-Type': 'application/json', 'Authorization': 'Bearer testapikey'},
|
|
json={'text': 'test subject', 'isRegion': True, 'timeEnd': 120000, 'panelId': int(panelId), 'time': 60000},
|
|
verify=True,
|
|
)
|
|
assert sent_messages == 1
|
|
|
|
|
|
def test_send_messages_with_bothids():
|
|
with mock.patch('awx.main.notifications.grafana_backend.requests') as requests_mock:
|
|
requests_mock.post.return_value.status_code = 200
|
|
m = {}
|
|
m['started'] = dt.datetime.fromtimestamp(60, tz=dt.timezone.utc).isoformat()
|
|
m['finished'] = dt.datetime.fromtimestamp(120, tz=dt.timezone.utc).isoformat()
|
|
m['subject'] = "test subject"
|
|
backend = grafana_backend.GrafanaBackend("testapikey", dashboardId='42', panelId='42')
|
|
message = EmailMessage(
|
|
m['subject'],
|
|
{"started": m['started'], "finished": m['finished']},
|
|
[],
|
|
[
|
|
'https://example.com',
|
|
],
|
|
)
|
|
sent_messages = backend.send_messages(
|
|
[
|
|
message,
|
|
]
|
|
)
|
|
requests_mock.post.assert_called_once_with(
|
|
'https://example.com/api/annotations',
|
|
headers={'Content-Type': 'application/json', 'Authorization': 'Bearer testapikey'},
|
|
json={'text': 'test subject', 'isRegion': True, 'timeEnd': 120000, 'panelId': 42, 'time': 60000, 'dashboardId': 42},
|
|
verify=True,
|
|
)
|
|
assert sent_messages == 1
|
|
|
|
|
|
def test_send_messages_with_emptyids():
|
|
with mock.patch('awx.main.notifications.grafana_backend.requests') as requests_mock:
|
|
requests_mock.post.return_value.status_code = 200
|
|
m = {}
|
|
m['started'] = dt.datetime.fromtimestamp(60, tz=dt.timezone.utc).isoformat()
|
|
m['finished'] = dt.datetime.fromtimestamp(120, tz=dt.timezone.utc).isoformat()
|
|
m['subject'] = "test subject"
|
|
backend = grafana_backend.GrafanaBackend("testapikey", dashboardId='', panelId='')
|
|
message = EmailMessage(
|
|
m['subject'],
|
|
{"started": m['started'], "finished": m['finished']},
|
|
[],
|
|
[
|
|
'https://example.com',
|
|
],
|
|
)
|
|
sent_messages = backend.send_messages(
|
|
[
|
|
message,
|
|
]
|
|
)
|
|
requests_mock.post.assert_called_once_with(
|
|
'https://example.com/api/annotations',
|
|
headers={'Content-Type': 'application/json', 'Authorization': 'Bearer testapikey'},
|
|
json={'text': 'test subject', 'isRegion': True, 'timeEnd': 120000, 'time': 60000},
|
|
verify=True,
|
|
)
|
|
assert sent_messages == 1
|
|
|
|
|
|
def test_send_messages_with_tags():
|
|
with mock.patch('awx.main.notifications.grafana_backend.requests') as requests_mock:
|
|
requests_mock.post.return_value.status_code = 200
|
|
m = {}
|
|
m['started'] = dt.datetime.fromtimestamp(60, tz=dt.timezone.utc).isoformat()
|
|
m['finished'] = dt.datetime.fromtimestamp(120, tz=dt.timezone.utc).isoformat()
|
|
m['subject'] = "test subject"
|
|
backend = grafana_backend.GrafanaBackend("testapikey", dashboardId='', panelId='', annotation_tags=["ansible"])
|
|
message = EmailMessage(
|
|
m['subject'],
|
|
{"started": m['started'], "finished": m['finished']},
|
|
[],
|
|
[
|
|
'https://example.com',
|
|
],
|
|
)
|
|
sent_messages = backend.send_messages(
|
|
[
|
|
message,
|
|
]
|
|
)
|
|
requests_mock.post.assert_called_once_with(
|
|
'https://example.com/api/annotations',
|
|
headers={'Content-Type': 'application/json', 'Authorization': 'Bearer testapikey'},
|
|
json={'tags': ['ansible'], 'text': 'test subject', 'isRegion': True, 'timeEnd': 120000, 'time': 60000},
|
|
verify=True,
|
|
)
|
|
assert sent_messages == 1
|