Remove now unused methods that populated template usage redis cache

This commit is contained in:
Pea Tyczynska
2019-01-15 14:38:45 +00:00
parent 3ce0024eec
commit d36c4d8a78
5 changed files with 4 additions and 197 deletions

View File

@@ -1,73 +0,0 @@
from datetime import datetime
from freezegun import freeze_time
import pytest
from app.commands import populate_redis_template_usage
from tests.conftest import set_config
from tests.app.db import create_notification, create_template, create_service
def test_populate_redis_template_usage_does_nothing_if_redis_disabled(mocker, notify_api, sample_service):
mock_redis = mocker.patch('app.commands.redis_store')
with set_config(notify_api, 'REDIS_ENABLED', False):
with pytest.raises(SystemExit) as exit_signal:
populate_redis_template_usage.callback.__wrapped__(sample_service.id, datetime.utcnow())
assert mock_redis.mock_calls == []
# sys.exit with nonzero exit code
assert exit_signal.value.code != 0
def test_populate_redis_template_usage_does_nothing_if_no_data(mocker, notify_api, sample_service):
mock_redis = mocker.patch('app.commands.redis_store')
with set_config(notify_api, 'REDIS_ENABLED', True):
populate_redis_template_usage.callback.__wrapped__(sample_service.id, datetime.utcnow())
assert mock_redis.mock_calls == []
@freeze_time('2017-06-12')
def test_populate_redis_template_usage_only_populates_for_today(mocker, notify_api, sample_template):
mock_redis = mocker.patch('app.commands.redis_store')
# created at in utc
create_notification(sample_template, created_at=datetime(2017, 6, 9, 23, 0, 0))
create_notification(sample_template, created_at=datetime(2017, 6, 9, 23, 0, 0))
create_notification(sample_template, created_at=datetime(2017, 6, 10, 0, 0, 0))
create_notification(sample_template, created_at=datetime(2017, 6, 10, 23, 0, 0)) # actually on 11th BST
with set_config(notify_api, 'REDIS_ENABLED', True):
populate_redis_template_usage.callback.__wrapped__(sample_template.service_id, datetime(2017, 6, 10))
mock_redis.set_hash_and_expire.assert_called_once_with(
'service-{}-template-usage-2017-06-10'.format(sample_template.service_id),
{str(sample_template.id): 3},
notify_api.config['EXPIRE_CACHE_EIGHT_DAYS'],
raise_exception=True
)
@freeze_time('2017-06-12')
def test_populate_redis_template_usage_only_populates_for_given_service(mocker, notify_api, notify_db_session):
mock_redis = mocker.patch('app.commands.redis_store')
# created at in utc
s1 = create_service(service_name='a')
s2 = create_service(service_name='b')
t1 = create_template(s1)
t2 = create_template(s2)
create_notification(t1, created_at=datetime(2017, 6, 10))
create_notification(t1, created_at=datetime(2017, 6, 10))
create_notification(t2, created_at=datetime(2017, 6, 10))
with set_config(notify_api, 'REDIS_ENABLED', True):
populate_redis_template_usage.callback.__wrapped__(s1.id, datetime(2017, 6, 10))
mock_redis.set_hash_and_expire.assert_called_once_with(
'service-{}-template-usage-2017-06-10'.format(s1.id),
{str(t1.id): 2},
notify_api.config['EXPIRE_CACHE_EIGHT_DAYS'],
raise_exception=True
)

View File

@@ -213,7 +213,6 @@ def test_persist_notification_with_optionals(sample_job, sample_api_key, mocker)
@freeze_time("2016-01-01 11:09:00.061258")
def test_persist_notification_doesnt_touch_cache_for_old_keys_that_dont_exist(sample_template, sample_api_key, mocker):
mock_incr = mocker.patch('app.notifications.process_notifications.redis_store.incr')
mock_incr_hash_value = mocker.patch('app.notifications.process_notifications.redis_store.increment_hash_value')
mocker.patch('app.notifications.process_notifications.redis_store.get', return_value=None)
mocker.patch('app.notifications.process_notifications.redis_store.get_all_from_hash', return_value=None)
@@ -229,16 +228,11 @@ def test_persist_notification_doesnt_touch_cache_for_old_keys_that_dont_exist(sa
reference="ref"
)
mock_incr.assert_not_called()
mock_incr_hash_value.assert_called_once_with(
"service-{}-template-usage-2016-01-01".format(sample_template.service_id),
sample_template.id
)
@freeze_time("2016-01-01 11:09:00.061258")
def test_persist_notification_increments_cache_if_key_exists(sample_template, sample_api_key, mocker):
mock_incr = mocker.patch('app.notifications.process_notifications.redis_store.incr')
mock_incr_hash_value = mocker.patch('app.notifications.process_notifications.redis_store.increment_hash_value')
mocker.patch('app.notifications.process_notifications.redis_store.get', return_value=1)
mocker.patch('app.notifications.process_notifications.redis_store.get_all_from_hash',
return_value={sample_template.id, 1})
@@ -255,10 +249,6 @@ def test_persist_notification_increments_cache_if_key_exists(sample_template, sa
reference="ref2")
mock_incr.assert_called_once_with(str(sample_template.service_id) + "-2016-01-01-count", )
assert mock_incr_hash_value.mock_calls == [
call("{}-template-counter-limit-7-days".format(sample_template.service_id), sample_template.id),
call("service-{}-template-usage-2016-01-01".format(sample_template.service_id), sample_template.id),
]
@pytest.mark.parametrize((
@@ -516,44 +506,6 @@ def test_persist_letter_notification_finds_correct_postage(
assert persisted_notification.postage == expected_postage
@pytest.mark.parametrize('utc_time, day_in_key', [
('2016-01-01 23:00:00', '2016-01-01'),
('2016-06-01 22:59:00', '2016-06-01'),
('2016-06-01 23:00:00', '2016-06-02'),
])
def test_persist_notification_increments_and_expires_redis_template_usage(
utc_time,
day_in_key,
sample_template,
sample_api_key,
mocker
):
mock_incr_hash_value = mocker.patch('app.notifications.process_notifications.redis_store.increment_hash_value')
mock_expire = mocker.patch('app.notifications.process_notifications.redis_store.expire')
mocker.patch('app.notifications.process_notifications.redis_store.get', return_value=None)
mocker.patch('app.notifications.process_notifications.redis_store.get_all_from_hash', return_value=None)
with freeze_time(utc_time):
persist_notification(
template_id=sample_template.id,
template_version=sample_template.version,
recipient='+447111111122',
service=sample_template.service,
personalisation={},
notification_type='sms',
api_key_id=sample_api_key.id,
key_type=sample_api_key.key_type,
)
mock_incr_hash_value.assert_called_once_with(
'service-{}-template-usage-{}'.format(str(sample_template.service_id), day_in_key),
sample_template.id
)
mock_expire.assert_called_once_with(
'service-{}-template-usage-{}'.format(str(sample_template.service_id), day_in_key),
current_app.config['EXPIRE_CACHE_EIGHT_DAYS']
)
def test_persist_notification_with_billable_units_stores_correct_info(
sample_template,
):