Merge branch 'main' into aanand-sqlalchemy-version-update

This commit is contained in:
Aditi Anand
2024-03-29 15:51:11 -04:00
29 changed files with 630 additions and 451 deletions

View File

@@ -34,6 +34,7 @@ from app.serialised_models import (
SerialisedService,
SerialisedTemplate,
)
from app.service.utils import service_allowed_to_send_to
from app.utils import get_template_instance
from app.v2.errors import BadRequestError, RateLimitError, TotalRequestsError
from tests.app.db import (
@@ -498,7 +499,7 @@ def test_check_service_over_api_rate_limit_when_exceed_rate_limit_request_fails_
with pytest.raises(RateLimitError) as e:
check_service_over_api_rate_limit(serialised_service, serialised_api_key)
assert app.redis_store.exceeded_rate_limit.called_with(
app.redis_store.exceeded_rate_limit.assert_called_with(
f"{sample_service.id}-{api_key.key_type}",
sample_service.rate_limit,
60,
@@ -527,7 +528,7 @@ def test_check_service_over_api_rate_limit_when_rate_limit_has_not_exceeded_limi
)[0]
check_service_over_api_rate_limit(serialised_service, serialised_api_key)
assert app.redis_store.exceeded_rate_limit.called_with(
app.redis_store.exceeded_rate_limit.assert_called_with(
f"{sample_service.id}-{api_key.key_type}",
3000,
60,
@@ -802,3 +803,21 @@ def test_check_service_over_total_message_limit(mocker, sample_service):
sample_service,
)
assert service_stats == 0
def test_service_allowed_to_send_to_simulated_numbers():
trial_mode_service = create_service(service_name="trial mode", restricted=True)
can_send = service_allowed_to_send_to(
"+14254147755",
trial_mode_service,
KeyType.NORMAL,
allow_guest_list_recipients=True,
)
can_not_send = service_allowed_to_send_to(
"+15555555555",
trial_mode_service,
KeyType.NORMAL,
allow_guest_list_recipients=True,
)
assert can_send is True
assert can_not_send is False

View File

@@ -1,3 +1,4 @@
import os
import uuid
import pytest
@@ -36,6 +37,7 @@ def test_create_invited_org_user(
platform_admin,
expected_invited_by,
):
os.environ["LOGIN_DOT_GOV_REGISTRATION_URL"] = "http://foo.fake.gov"
mocked = mocker.patch("app.celery.provider_tasks.deliver_email.apply_async")
email_address = "invited_user@example.com"
sample_user.platform_admin = platform_admin
@@ -67,8 +69,8 @@ def test_create_invited_org_user(
assert len(notification.personalisation.keys()) == 3
assert notification.personalisation["organization_name"] == "sample organization"
assert notification.personalisation["user_name"] == expected_invited_by
assert notification.personalisation["url"].startswith(expected_start_of_invite_url)
assert len(notification.personalisation["url"]) > len(expected_start_of_invite_url)
# assert notification.personalisation["url"].startswith(expected_start_of_invite_url)
# assert len(notification.personalisation["url"]) > len(expected_start_of_invite_url)
mocked.assert_called_once_with(
[(str(notification.id))], queue="notify-internal-tasks"