2019-10-20 02:12:00 +02:00
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
from mock import MagicMock, patch
|
|
|
|
|
|
|
|
|
|
from zerver.lib.test_classes import WebhookTestCase
|
|
|
|
|
|
2020-01-14 22:06:24 +01:00
|
|
|
|
2019-10-20 02:12:00 +02:00
|
|
|
class HarborHookTests(WebhookTestCase):
|
|
|
|
|
STREAM_NAME = "harbor"
|
|
|
|
|
URL_TEMPLATE = u"/api/v1/external/harbor?api_key={api_key}&stream={stream}"
|
|
|
|
|
|
|
|
|
|
def test_push_image(self) -> None:
|
|
|
|
|
expected_topic = "example/test"
|
|
|
|
|
expected_message = """**admin** pushed image `example/test:latest`"""
|
|
|
|
|
self.send_and_test_stream_message(
|
|
|
|
|
"push_image", expected_topic, expected_message)
|
|
|
|
|
|
|
|
|
|
@patch('zerver.lib.webhooks.common.check_send_webhook_message')
|
|
|
|
|
def test_delete_image_ignored(
|
|
|
|
|
self, check_send_webhook_message_mock: MagicMock) -> None:
|
|
|
|
|
self.url = self.build_webhook_url()
|
|
|
|
|
payload = self.get_body('delete_image')
|
|
|
|
|
result = self.client_post(self.url, payload, content_type="application/json")
|
|
|
|
|
self.assertFalse(check_send_webhook_message_mock.called)
|
|
|
|
|
self.assert_json_success(result)
|
|
|
|
|
|
|
|
|
|
def test_scanning_completed(self) -> None:
|
|
|
|
|
expected_topic = "example/test"
|
|
|
|
|
|
|
|
|
|
expected_message = """
|
|
|
|
|
Image scan completed for `example/test:latest`. Vulnerabilities by severity:
|
|
|
|
|
|
|
|
|
|
* High: **12**
|
|
|
|
|
* Medium: **16**
|
|
|
|
|
* Low: **7**
|
|
|
|
|
* Unknown: **2**
|
|
|
|
|
* None: **131**
|
|
|
|
|
""".strip()
|
|
|
|
|
|
|
|
|
|
self.send_and_test_stream_message(
|
|
|
|
|
"scanning_completed", expected_topic, expected_message)
|
|
|
|
|
|
|
|
|
|
def get_body(self, fixture_name: str) -> str:
|
|
|
|
|
return self.webhook_fixture_data("harbor", fixture_name, file_type="json")
|
