Wire in the new API method that calculates the total cost and total billable units.

- Used on dashboard to calculate free tier/cost
- Update tests to mock new method
- Two new tests to check output on dashboard page
This commit is contained in:
Martyn Inglis
2017-05-24 10:35:52 +01:00
parent 1c1c5bd758
commit 52326539d6
6 changed files with 84 additions and 2 deletions

View File

@@ -1332,6 +1332,15 @@ def mock_get_usage(mocker, service_one, fake_uuid):
'app.service_api_client.get_service_usage', side_effect=_get_usage)
@pytest.fixture(scope='function')
def mock_get_yearly_sms_unit_count_and_cost(mocker, service_one, fake_uuid):
def _get_usage(service_id, year=None):
return {"billable_sms_units": 100, "total_cost": 200.0}
return mocker.patch(
'app.service_api_client.get_yearly_sms_unit_count_and_cost', side_effect=_get_usage)
@pytest.fixture(scope='function')
def mock_get_billable_units(mocker):
def _get_usage(service_id, year):