def test_update_service_data_retention_does_not_update_if_row_does_not_exist( sample_service): updated_count = update_service_data_retention( service_data_retention_id=uuid.uuid4(), service_id=sample_service.id, days_of_retention=5) assert updated_count == 0 assert len(ServiceDataRetention.query.all()) == 0
def test_update_service_data_retention_does_not_update_row_if_data_retention_is_for_different_service( sample_service): data_retention = insert_service_data_retention( service_id=sample_service.id, notification_type='email', days_of_retention=3) updated_count = update_service_data_retention( service_data_retention_id=data_retention.id, service_id=uuid.uuid4(), days_of_retention=5) assert updated_count == 0
def modify_service_data_retention(service_id, data_retention_id): form = validate(request.get_json(), update_service_data_retention_request) update_count = update_service_data_retention( service_data_retention_id=data_retention_id, service_id=service_id, days_of_retention=form.get("days_of_retention")) if update_count == 0: raise InvalidRequest( message= "The service data retention for id: {} was not found for service: {}" .format(data_retention_id, service_id), status_code=404) return '', 204
def test_update_service_data_retention(sample_service): data_retention = insert_service_data_retention( service_id=sample_service.id, notification_type='sms', days_of_retention=3) updated_count = update_service_data_retention( service_data_retention_id=data_retention.id, service_id=sample_service.id, days_of_retention=5) assert updated_count == 1 results = ServiceDataRetention.query.all() assert len(results) == 1 assert results[0].id == data_retention.id assert results[0].service_id == sample_service.id assert results[0].notification_type == 'sms' assert results[0].days_of_retention == 5 assert results[0].created_at.date() == datetime.utcnow().date() assert results[0].updated_at.date() == datetime.utcnow().date()