def test_get_quota_definitions_when_has_quotas( mock_execute_sql, has_quotas, called_execute_sql, ): mock_execute_sql.return_value = [] application = mock.MagicMock(country_name='spain', execute_sql=mock_execute_sql) document = Document(application) document.has_quotas = has_quotas document.q = [] document.quota_order_number_list = [] document.get_quota_definitions() assert mock_execute_sql.called is called_execute_sql
def test_get_quota_balances( mock_first_come_first_serve, mock_licensed, has_quotas, called_add_first_come_first_serve, called_add_licensed, ): mock_first_come_first_serve.return_value = None mock_licensed.return_value = None application = mock.MagicMock(country_name='spain') document = Document(application) document.has_quotas = has_quotas document.get_quota_balances() assert mock_licensed.called is called_add_licensed assert mock_first_come_first_serve.called is called_add_first_come_first_serve
def test_get_quota_definitions(quota_order_number_id, add_balance, expected_quota_order_number_sid, expected_definition): geographical_area_id = '1011' country_profile = 'spain' quota_order_number = QuotaOrderNumberFactory( id=quota_order_number_id, quota_order_number_sid=str(quota_order_number_id)) if not quota_order_number_id.startswith('094'): QuotaDefinitionFactory( quota_order_number_id=quota_order_number.id, quota_order_number_sid=quota_order_number.quota_order_number_sid, initial_volume=2000, volume=3000, measurement_unit_code='') AgreementFactory(country_name=country_profile.capitalize(), slug=country_profile, country_codes=[geographical_area_id]) application = Application(country_profile=country_profile) document = Document(application) document.has_quotas = True if add_balance: qb = get_quota_balance( quota_order_number_id=quota_order_number.quota_order_number_id, y1_balance=6000, ) document.balance_dict[quota_order_number.quota_order_number_id] = qb document.q = [quota_order_number.quota_order_number_id] qon = QuotaOrderNumber(quota_order_number.quota_order_number_id) assert document.quota_definition_list == [] document.quota_order_number_list = [qon] document.get_quota_definitions() assert len(qon.quota_definition_list) == 1 assert qon.quota_definition_list[0].quota_order_number_id == str( quota_order_number_id) assert len(document.quota_definition_list) == 1 assert_object(document.quota_definition_list[0], expected_definition)