From c25d8a5d34567eb5de45163af8815d824d28007d Mon Sep 17 00:00:00 2001 From: Jeandre Le Roux Date: Fri, 23 Mar 2018 08:17:20 +0200 Subject: [PATCH] Fix rocket.chat notification test flake8 Signed-off-by: Jeandre Le Roux --- awx/main/tests/unit/notifications/test_rocketchat.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/awx/main/tests/unit/notifications/test_rocketchat.py b/awx/main/tests/unit/notifications/test_rocketchat.py index 7cd68b4734..d621f74ba2 100644 --- a/awx/main/tests/unit/notifications/test_rocketchat.py +++ b/awx/main/tests/unit/notifications/test_rocketchat.py @@ -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)