mirror of
https://github.com/PostHog/posthog.git
synced 2024-11-25 02:31:30 +01:00
85 lines
4.1 KiB
Python
85 lines
4.1 KiB
Python
from unittest.mock import MagicMock, patch
|
|
|
|
from freezegun import freeze_time
|
|
|
|
from ee.tasks.subscriptions.email_subscriptions import send_email_subscription_report
|
|
from ee.tasks.test.subscriptions.subscriptions_test_factory import create_subscription
|
|
from posthog.models.dashboard import Dashboard
|
|
from posthog.models.exported_asset import ExportedAsset
|
|
from posthog.models.insight import Insight
|
|
from posthog.models.instance_setting import set_instance_setting
|
|
from posthog.models.subscription import Subscription
|
|
from posthog.tasks.test.utils_email_tests import mock_email_messages
|
|
from posthog.test.base import APIBaseTest
|
|
|
|
|
|
@patch("ee.tasks.subscriptions.email_subscriptions.EmailMessage")
|
|
@freeze_time("2022-02-02T08:55:00.000Z")
|
|
class TestEmailSubscriptionsTasks(APIBaseTest):
|
|
subscription: Subscription
|
|
dashboard: Dashboard
|
|
insight: Insight
|
|
asset: ExportedAsset
|
|
|
|
def setUp(self) -> None:
|
|
self.dashboard = Dashboard.objects.create(team=self.team, name="private dashboard", created_by=self.user)
|
|
self.insight = Insight.objects.create(team=self.team, short_id="123456", name="My Test subscription")
|
|
|
|
set_instance_setting("EMAIL_HOST", "fake_host")
|
|
set_instance_setting("EMAIL_ENABLED", True)
|
|
|
|
self.asset = ExportedAsset.objects.create(team=self.team, insight_id=self.insight.id, export_format="image/png")
|
|
self.subscription = create_subscription(team=self.team, insight=self.insight, created_by=self.user)
|
|
|
|
def test_subscription_delivery(self, MockEmailMessage: MagicMock) -> None:
|
|
mocked_email_messages = mock_email_messages(MockEmailMessage)
|
|
|
|
send_email_subscription_report("test1@posthog.com", self.subscription, [self.asset])
|
|
|
|
assert len(mocked_email_messages) == 1
|
|
assert mocked_email_messages[0].send.call_count == 1
|
|
assert "is ready!" in mocked_email_messages[0].html_body
|
|
assert f"/exporter/export-my-test-subscription.png?token=ey" in mocked_email_messages[0].html_body
|
|
|
|
def test_new_subscription_delivery(self, MockEmailMessage: MagicMock) -> None:
|
|
mocked_email_messages = mock_email_messages(MockEmailMessage)
|
|
|
|
send_email_subscription_report(
|
|
"test1@posthog.com", self.subscription, [self.asset], invite_message="My invite message"
|
|
)
|
|
|
|
assert len(mocked_email_messages) == 1
|
|
assert mocked_email_messages[0].send.call_count == 1
|
|
|
|
assert f"has subscribed you" in mocked_email_messages[0].html_body
|
|
assert "Someone subscribed you to a PostHog Insight" == mocked_email_messages[0].subject
|
|
assert "This subscription is sent every day. The next subscription will be sent on Wednesday February 02, 2022"
|
|
assert "My invite message" in mocked_email_messages[0].html_body
|
|
|
|
def test_should_have_different_text_for_self(self, MockEmailMessage: MagicMock) -> None:
|
|
mocked_email_messages = mock_email_messages(MockEmailMessage)
|
|
|
|
send_email_subscription_report(
|
|
self.user.email, self.subscription, [self.asset], invite_message="My invite message"
|
|
)
|
|
|
|
assert len(mocked_email_messages) == 1
|
|
assert mocked_email_messages[0].send.call_count == 1
|
|
assert "You have been subscribed" in mocked_email_messages[0].html_body
|
|
assert "You have been subscribed to a PostHog Insight" == mocked_email_messages[0].subject
|
|
|
|
def test_sends_dashboard_subscription(self, MockEmailMessage: MagicMock) -> None:
|
|
mocked_email_messages = mock_email_messages(MockEmailMessage)
|
|
|
|
subscription = create_subscription(team=self.team, dashboard=self.dashboard, created_by=self.user)
|
|
|
|
send_email_subscription_report(
|
|
self.user.email, subscription, [self.asset], invite_message="My invite message", total_asset_count=10
|
|
)
|
|
|
|
assert len(mocked_email_messages) == 1
|
|
assert mocked_email_messages[0].send.call_count == 1
|
|
assert "You have been subscribed" in mocked_email_messages[0].html_body
|
|
assert "You have been subscribed to a PostHog Dashboard" == mocked_email_messages[0].subject
|
|
assert f"SHOWING 1 OF 10 DASHBOARD INSIGHTS" in mocked_email_messages[0].html_body
|