From e1d1cf76caa54846d818b41341b65eea26c7e263 Mon Sep 17 00:00:00 2001
From: Derek Brooks <derek@broox.com>
Date: Fri, 17 Nov 2017 23:12:36 -0600
Subject: [PATCH] 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
---
 .coveragerc                              |   1 -
 tests/components/notify/test_facebook.py | 129 +++++++++++++++++++++++
 2 files changed, 129 insertions(+), 1 deletion(-)
 create mode 100644 tests/components/notify/test_facebook.py

diff --git a/.coveragerc b/.coveragerc
index 01187b92d66..2b846ca5b09 100644
--- a/.coveragerc
+++ b/.coveragerc
@@ -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
diff --git a/tests/components/notify/test_facebook.py b/tests/components/notify/test_facebook.py
new file mode 100644
index 00000000000..7bc7a55869a
--- /dev/null
+++ b/tests/components/notify/test_facebook.py
@@ -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)