Files
zulip/zerver/webhooks/thinkst/tests.py
Anders Kaseorg 69730a78cc python: Use trailing commas consistently.
Automatically generated by the following script, based on the output
of lint with flake8-comma:

import re
import sys

last_filename = None
last_row = None
lines = []

for msg in sys.stdin:
    m = re.match(
        r"\x1b\[35mflake8    \|\x1b\[0m \x1b\[1;31m(.+):(\d+):(\d+): (\w+)", msg
    )
    if m:
        filename, row_str, col_str, err = m.groups()
        row, col = int(row_str), int(col_str)

        if filename == last_filename:
            assert last_row != row
        else:
            if last_filename is not None:
                with open(last_filename, "w") as f:
                    f.writelines(lines)

            with open(filename) as f:
                lines = f.readlines()
            last_filename = filename
        last_row = row

        line = lines[row - 1]
        if err in ["C812", "C815"]:
            lines[row - 1] = line[: col - 1] + "," + line[col - 1 :]
        elif err in ["C819"]:
            assert line[col - 2] == ","
            lines[row - 1] = line[: col - 2] + line[col - 1 :].lstrip(" ")

if last_filename is not None:
    with open(last_filename, "w") as f:
        f.writelines(lines)

Signed-off-by: Anders Kaseorg <anders@zulipchat.com>
2020-06-11 16:04:12 -07:00

103 lines
4.2 KiB
Python

from zerver.lib.test_classes import WebhookTestCase
class ThinkstHookTests(WebhookTestCase):
STREAM_NAME = 'travis'
URL_TEMPLATE = "/api/v1/external/thinkst?stream={stream}&api_key={api_key}"
FIXTURE_DIR_NAME = 'thinkst'
def test_canary_alert(self) -> None:
"""
Canary alerts are generated by Thinkst's hardware or virtual canaries.
"""
expected_message = ('**:alert: Canary has been triggered!**\n\n'
'On 2020-06-09 13:59:38 (UTC), `0000000testnode` was triggered '
'from `2.2.2.2` (`attacker-ip.local`):\n\n'
'> This is a dummy incident.')
self.send_and_test_stream_message(
'canary_alert',
'canary alert - 0000000testnode',
expected_message,
content_type="application/x-www-form-urlencoded",
)
def test_canary_alert_no_reverse_dns(self) -> None:
"""
Canary alerts are generated by Thinkst's hardware or virtual canaries.
"""
expected_message = ('**:alert: Canary has been triggered!**\n\n'
'On 2020-06-09 13:59:38 (UTC), `0000000testnode` was triggered '
'from `2.2.2.2`:\n\n'
'> This is a dummy incident.')
self.send_and_test_stream_message(
'canary_alert_no_reverse_dns',
'canary alert - 0000000testnode',
expected_message,
content_type="application/x-www-form-urlencoded",
)
def test_canary_alert_with_specific_topic(self) -> None:
"""
Canary alerts are generated by Thinkst's hardware or virtual canaries.
"""
self.url = self.build_webhook_url(topic='foo')
expected_message = ('**:alert: Canary `0000000testnode` has been triggered!**\n\n'
'On 2020-06-09 13:59:38 (UTC), `0000000testnode` was triggered '
'from `2.2.2.2` (`attacker-ip.local`):\n\n'
'> This is a dummy incident.')
self.send_and_test_stream_message(
'canary_alert',
'foo',
expected_message,
content_type="application/x-www-form-urlencoded",
)
def test_canarytoken_new(self) -> None:
"""
Thinkst Canarytokens are simple tripwires for detecting when resources have been accessed.
"""
expected_message = ('**:alert: Canarytoken has been triggered on 2020-06-09 14:04:39!**\n\n'
'> Congrats! The newly saved webhook works \n\n'
'[Manage this canarytoken](http://example.com/test/url/for/webhook)')
self.send_and_test_stream_message(
'canarytoken_new',
'canarytoken alert',
expected_message,
content_type="application/x-www-form-urlencoded",
)
def test_canarytoken_real(self) -> None:
"""
Thinkst Canarytokens are simple tripwires for detecting when resources have been accessed.
"""
expected_message = ('**:alert: Canarytoken has been triggered on 2020-06-09 14:04:47 (UTC)!**\n\n'
'> Canarytoken example \n\n'
'[Manage this canarytoken](https://canarytokens.org/manage?token=foo&auth=bar)')
self.send_and_test_stream_message(
'canarytoken_real',
'canarytoken alert',
expected_message,
content_type="application/x-www-form-urlencoded",
)
def test_canarytoken_with_specific_topic(self) -> None:
"""
Thinkst Canarytokens are simple tripwires for detecting when resources have been accessed.
"""
self.url = self.build_webhook_url(topic='foo')
expected_message = ('**:alert: Canarytoken has been triggered on 2020-06-09 14:04:47 (UTC)!**\n\n'
'> Canarytoken example \n\n'
'[Manage this canarytoken](https://canarytokens.org/manage?token=foo&auth=bar)')
self.send_and_test_stream_message(
'canarytoken_real',
'foo',
expected_message,
content_type="application/x-www-form-urlencoded",
)