Fix rocket.chat notification test flake8

Signed-off-by: Jeandre Le Roux <theblazehen@theblazehen.com>
This commit is contained in:
Jeandre Le Roux 2018-03-23 08:17:20 +02:00
parent 0525df595e
commit c25d8a5d34

View File

@ -1,9 +1,9 @@
import mock
import pytest
from django.core.mail.message import EmailMessage
import awx.main.notifications.rocketchat_backend as rocketchat_backend
def test_send_messages():
with mock.patch('awx.main.notifications.rocketchat_backend.requests') as requests_mock:
backend = rocketchat_backend.RocketChatBackend()
@ -12,6 +12,7 @@ def test_send_messages():
requests_mock.post.assert_called_once_with('http://example.com', data='{"text": "test subject"}', verify=True)
assert sent_messages == 1
def test_send_messages_with_username():
with mock.patch('awx.main.notifications.rocketchat_backend.requests') as requests_mock:
backend = rocketchat_backend.RocketChatBackend(rocketchat_username='testuser')
@ -20,6 +21,7 @@ def test_send_messages_with_username():
requests_mock.post.assert_called_once_with('http://example.com', data='{"username": "testuser", "text": "test subject"}', verify=True)
assert sent_messages == 1
def test_send_messages_with_icon_url():
with mock.patch('awx.main.notifications.rocketchat_backend.requests') as requests_mock:
backend = rocketchat_backend.RocketChatBackend(rocketchat_icon_url='http://example.com')
@ -28,6 +30,7 @@ def test_send_messages_with_icon_url():
requests_mock.post.assert_called_once_with('http://example.com', data='{"text": "test subject", "icon_url": "http://example.com"}', verify=True)
assert sent_messages == 1
def test_send_messages_with_no_verify_ssl():
with mock.patch('awx.main.notifications.rocketchat_backend.requests') as requests_mock:
backend = rocketchat_backend.RocketChatBackend(rocketchat_no_verify_ssl=True)