mirror of
https://github.com/GSA/notifications-api.git
synced 2026-01-30 14:31:57 -05:00
Add labels to query so that the named tuples can be referenced later.
Remove unnecessary function
This commit is contained in:
@@ -111,7 +111,7 @@ def assert_monthly_billing(monthly_billing, year, month, service_id, expected_le
|
||||
assert sorted(monthly_billing.monthly_totals[0]) == sorted(first_row)
|
||||
|
||||
|
||||
def test_get_service_id():
|
||||
def test_get_service_id(notify_db_session):
|
||||
service_1 = create_service(service_name="Service One")
|
||||
template_1 = create_template(service=service_1)
|
||||
service_2 = create_service(service_name="Service Two")
|
||||
|
||||
@@ -1,8 +1,8 @@
|
||||
import uuid
|
||||
from datetime import datetime
|
||||
from decimal import Decimal
|
||||
from app.dao.provider_rates_dao import create_provider_rates, create_sms_rate
|
||||
from app.models import ProviderRates, ProviderDetails, Rate
|
||||
from app.dao.provider_rates_dao import create_provider_rates
|
||||
from app.models import ProviderRates, ProviderDetails
|
||||
|
||||
|
||||
def test_create_provider_rates(notify_db, notify_db_session, mmg_provider):
|
||||
@@ -16,11 +16,3 @@ def test_create_provider_rates(notify_db, notify_db_session, mmg_provider):
|
||||
assert ProviderRates.query.first().rate == rate
|
||||
assert ProviderRates.query.first().valid_from == now
|
||||
assert ProviderRates.query.first().provider_id == provider.id
|
||||
|
||||
|
||||
def test_create_sms_rate():
|
||||
rate = Rate(id=uuid.uuid4(), valid_from=datetime.now(), rate=0.014, notification_type='sms')
|
||||
create_sms_rate(rate)
|
||||
rates = Rate.query.all()
|
||||
assert len(rates) == 1
|
||||
assert rates[0] == rate
|
||||
|
||||
@@ -1,8 +1,8 @@
|
||||
from datetime import datetime
|
||||
import uuid
|
||||
|
||||
from app import db
|
||||
from app.dao.jobs_dao import dao_create_job
|
||||
from app.dao.provider_rates_dao import create_sms_rate
|
||||
from app.dao.service_inbound_api_dao import save_service_inbound_api
|
||||
from app.models import (
|
||||
Service,
|
||||
@@ -244,5 +244,6 @@ def create_organisation(colour='blue', logo='test_x2.png', name='test_org_1'):
|
||||
|
||||
def create_rate(start_date, value, notification_type):
|
||||
rate = Rate(id=uuid.uuid4(), valid_from=start_date, rate=value, notification_type=notification_type)
|
||||
create_sms_rate(rate)
|
||||
db.session.add(rate)
|
||||
db.session.commit()
|
||||
return rate
|
||||
|
||||
Reference in New Issue
Block a user