Add Facebook Notification tests (#10642)

* test the facebook notification component

* respond to hound feedback

* remove unnecessary line breaks

* parse_qs not needed with requests_mock

* remove facebook notifier from .coveragerc
This commit is contained in:
Derek Brooks 2017-11-17 23:12:36 -06:00 committed by Paulus Schoutsen
parent 1317297191
commit e1d1cf76ca
2 changed files with 129 additions and 1 deletions

View file

@ -426,7 +426,6 @@ omit =
homeassistant/components/notify/clicksend.py
homeassistant/components/notify/clicksend_tts.py
homeassistant/components/notify/discord.py
homeassistant/components/notify/facebook.py
homeassistant/components/notify/free_mobile.py
homeassistant/components/notify/gntp.py
homeassistant/components/notify/group.py

View file

@ -0,0 +1,129 @@
"""The test for the Facebook notify module."""
import unittest
import requests_mock
import homeassistant.components.notify.facebook as facebook
class TestFacebook(unittest.TestCase):
"""Tests for Facebook notifification service."""
def setUp(self):
"""Set up test variables."""
access_token = "page-access-token"
self.facebook = facebook.FacebookNotificationService(access_token)
@requests_mock.Mocker()
def test_send_simple_message(self, mock):
"""Test sending a simple message with success."""
mock.register_uri(
requests_mock.POST,
facebook.BASE_URL,
status_code=200
)
message = "This is just a test"
target = ["+15555551234"]
self.facebook.send_message(message=message, target=target)
self.assertTrue(mock.called)
self.assertEqual(mock.call_count, 1)
expected_body = {
"recipient": {"phone_number": target[0]},
"message": {"text": message}
}
self.assertEqual(mock.last_request.json(), expected_body)
expected_params = {"access_token": ["page-access-token"]}
self.assertEqual(mock.last_request.qs, expected_params)
@requests_mock.Mocker()
def test_sending_multiple_messages(self, mock):
"""Test sending a message to multiple targets."""
mock.register_uri(
requests_mock.POST,
facebook.BASE_URL,
status_code=200
)
message = "This is just a test"
targets = ["+15555551234", "+15555551235"]
self.facebook.send_message(message=message, target=targets)
self.assertTrue(mock.called)
self.assertEqual(mock.call_count, 2)
for idx, target in enumerate(targets):
request = mock.request_history[idx]
expected_body = {
"recipient": {"phone_number": target},
"message": {"text": message}
}
self.assertEqual(request.json(), expected_body)
expected_params = {"access_token": ["page-access-token"]}
self.assertEqual(request.qs, expected_params)
@requests_mock.Mocker()
def test_send_message_attachment(self, mock):
"""Test sending a message with a remote attachment."""
mock.register_uri(
requests_mock.POST,
facebook.BASE_URL,
status_code=200
)
message = "This will be thrown away."
data = {
"attachment": {
"type": "image",
"payload": {"url": "http://www.example.com/image.jpg"}
}
}
target = ["+15555551234"]
self.facebook.send_message(message=message, data=data, target=target)
self.assertTrue(mock.called)
self.assertEqual(mock.call_count, 1)
expected_body = {
"recipient": {"phone_number": target[0]},
"message": data
}
self.assertEqual(mock.last_request.json(), expected_body)
expected_params = {"access_token": ["page-access-token"]}
self.assertEqual(mock.last_request.qs, expected_params)
@requests_mock.Mocker()
def test_send_targetless_message(self, mock):
"""Test sending a message without a target."""
mock.register_uri(
requests_mock.POST,
facebook.BASE_URL,
status_code=200
)
self.facebook.send_message(message="goin nowhere")
self.assertFalse(mock.called)
@requests_mock.Mocker()
def test_send_message_with_400(self, mock):
"""Test sending a message with a 400 from Facebook."""
mock.register_uri(
requests_mock.POST,
facebook.BASE_URL,
status_code=400,
json={
"error": {
"message": "Invalid OAuth access token.",
"type": "OAuthException",
"code": 190,
"fbtrace_id": "G4Da2pFp2Dp"
}
}
)
self.facebook.send_message(message="nope!", target=["+15555551234"])
self.assertTrue(mock.called)
self.assertEqual(mock.call_count, 1)