zulip/zerver/webhooks/wordpress/tests.py

# -*- coding: utf-8 -*-
from six import text_type
from zerver.lib.test_classes import WebhookTestCase

class WordPressHookTests(WebhookTestCase):
    STREAM_NAME = 'wordpress'
    URL_TEMPLATE = "/api/v1/external/wordpress?api_key={api_key}"
    FIXTURE_DIR_NAME = 'wordpress'

    def test_publish_post(self):
        # type: () -> None

        expected_topic = u"WordPress Post"
        expected_message = u"New post published.\n[New Blog Post](http://example.com\n)"

        self.send_and_test_stream_message('publish_post', expected_topic, expected_message,
                                          content_type="application/x-www-form-urlencoded")

    def test_publish_post_type_not_provided(self):
        # type: () -> None

        expected_topic = u"WordPress Post"
        expected_message = u"New post published.\n[New Blog Post](http://example.com\n)"

        self.send_and_test_stream_message('publish_post_type_not_provided',
                                          expected_topic, expected_message,
                                          content_type="application/x-www-form-urlencoded")

    def test_publish_post_no_data_provided(self):
        # type: () -> None

        # Note: the fixture includes 'hook=publish_post' because it's always added by HookPress
        expected_topic = u"WordPress Notification"
        expected_message = u"New post published.\n" + "[New WordPress Post](WordPress Post URL)"

        self.send_and_test_stream_message('publish_post_no_data_provided',
                                          expected_topic, expected_message,
                                          content_type="application/x-www-form-urlencoded")

    def test_publish_page(self):
        # type: () -> None

        expected_topic = u"WordPress Page"
        expected_message = u"New page published.\n" + "[New Blog Page](http://example.com\n)"

        self.send_and_test_stream_message('publish_page', expected_topic, expected_message,
                                          content_type="application/x-www-form-urlencoded")

    def test_user_register(self):
        # type: () -> None

        expected_topic = u"New Blog Users"
        expected_message = u"New blog user registered.\nName: test_user\nemail: test_user@example.com"

        self.send_and_test_stream_message('user_register', expected_topic, expected_message,
                                          content_type="application/x-www-form-urlencoded")

    def test_wp_login(self):
        # type: () -> None

        expected_topic = u"New Login"
        expected_message = u"User testuser logged in."

        self.send_and_test_stream_message('wp_login', expected_topic, expected_message,
                                          content_type="application/x-www-form-urlencoded")

    def test_unknown_action_no_data(self):
        # type: () -> None

        # Mimic send_and_test_stream_message() to manually execute a negative test.
        # Otherwise its call to send_json_payload() would assert on the non-success
        # we are testing. The value of result is the error message the webhook should
        # return if no params are sent. The fixture for this test is an empty file.

        # subscribe to the target stream
        self.subscribe_to_stream(self.TEST_USER_EMAIL, self.STREAM_NAME)

        # post to the webhook url
        post_params = {'stream_name': self.STREAM_NAME,
                       'content_type': 'application/x-www-form-urlencoded'}
        result = self.client_post(self.url, 'unknown_action', **post_params)

        # check that we got the expected error message
        self.assert_json_error(result, "Unknown WordPress webhook action: WordPress Action")

    def test_unknown_action_no_hook_provided(self):
        # type: () -> None

        # Similar to unknown_action_no_data, except the fixture contains valid blog post
        # params but without the hook parameter. This should also return an error.

        self.subscribe_to_stream(self.TEST_USER_EMAIL, self.STREAM_NAME)
        post_params = {'stream_name': self.STREAM_NAME,
                       'content_type': 'application/x-www-form-urlencoded'}
        result = self.client_post(self.url, 'unknown_action', **post_params)

        self.assert_json_error(result, "Unknown WordPress webhook action: WordPress Action")

    def get_body(self, fixture_name):
        # type: (text_type) -> text_type
        return self.fixture_data("wordpress", fixture_name, file_type="txt")