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()
        message = EmailMessage('test subject', 'test body', [], ['http://example.com', ])
        sent_messages = backend.send_messages([message, ])
        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')
        message = EmailMessage('test subject', 'test body', [], ['http://example.com', ])
        sent_messages = backend.send_messages([message, ])
        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')
        message = EmailMessage('test subject', 'test body', [], ['http://example.com', ])
        sent_messages = backend.send_messages([message, ])
        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)
        message = EmailMessage('test subject', 'test body', [], ['http://example.com', ])
        sent_messages = backend.send_messages([message, ])
        requests_mock.post.assert_called_once_with('http://example.com', data='{"text": "test subject"}', verify=False)
        assert sent_messages == 1

Add unit test for rocket.chat notifications