|
11 | 11 |
|
12 | 12 | from dje.models import Dataspace |
13 | 13 | from dje.tests import create_superuser |
14 | | -from notification.models import Webhook |
| 14 | +from notification.models import WebhookSubscription |
15 | 15 |
|
16 | 16 |
|
17 | | -class NotificationModelsTestCase(TestCase): |
| 17 | +class WebhookSubscriptionModelTestCase(TestCase): |
18 | 18 | def setUp(self): |
19 | 19 | self.nexb_dataspace = Dataspace.objects.create(name="nexB") |
20 | 20 | self.nexb_user = create_superuser("nexb_user", self.nexb_dataspace) |
21 | 21 |
|
22 | | - self.webhook1 = Webhook.objects.create( |
| 22 | + self.webhook = WebhookSubscription.objects.create( |
23 | 23 | dataspace=self.nexb_dataspace, |
24 | | - target="http://1.2.3.4/", |
25 | | - user=self.nexb_user, |
| 24 | + target_url="http://1.2.3.4/", |
26 | 25 | event="request.added", |
27 | 26 | ) |
28 | 27 |
|
29 | | - def test_notification_webhook_model_str(self): |
30 | | - self.assertEqual("request.added => http://1.2.3.4/", str(self.webhook1)) |
| 28 | + def test_webhook_subscription_str(self): |
| 29 | + self.assertEqual("request.added => http://1.2.3.4/", str(self.webhook)) |
31 | 30 |
|
32 | | - def test_notification_webhook_model_dict(self): |
| 31 | + def test_webhook_subscription_dict(self): |
33 | 32 | expected = { |
34 | | - "uuid": str(self.webhook1.uuid), |
35 | | - "event": self.webhook1.event, |
36 | | - "target": self.webhook1.target, |
| 33 | + "uuid": str(self.webhook.uuid), |
| 34 | + "event": self.webhook.event, |
| 35 | + "target": self.webhook.target_url, |
37 | 36 | } |
38 | | - self.assertEqual(expected, self.webhook1.dict()) |
| 37 | + self.assertEqual(expected, self.webhook.dict()) |
39 | 38 |
|
40 | | - def test_notification_webhook_model_get_extra_headers(self): |
41 | | - self.webhook1.extra_headers = {"Header": "{{ENV_VALUE}}"} |
42 | | - self.webhook1.save() |
| 39 | + def test_webhook_subscription_get_extra_headers(self): |
| 40 | + self.webhook.extra_headers = {"Header": "{{ENV_VALUE}}"} |
| 41 | + self.webhook.save() |
43 | 42 |
|
44 | 43 | expected = {"Header": "{{ENV_VALUE}}"} |
45 | | - self.assertEqual(expected, self.webhook1.get_extra_headers()) |
| 44 | + self.assertEqual(expected, self.webhook.get_extra_headers()) |
46 | 45 |
|
47 | 46 | expected = {"Header": "some_value"} |
48 | 47 | with override_settings(HOOK_ENV={"ENV_VALUE": "some_value"}): |
49 | | - self.assertEqual(expected, self.webhook1.get_extra_headers()) |
| 48 | + self.assertEqual(expected, self.webhook.get_extra_headers()) |
0 commit comments