def setup_data(opensearch_with_collector):
"""Sets up data for the tests."""
investor_company = CompanyFactory(name='large abcdef')
argentina_investor_company = CompanyFactory(
name='argentina plc',
address_country_id=CountryConstant.argentina.value.id,
)
with freeze_time('2010-02-01'):
frozen_created_on_profile = LargeCapitalInvestorProfileFactory(
investor_company=CompanyFactory(
name='Frozen limited',
),
investor_description='frozen in 2010',
construction_risks=[
ConstructionRiskConstant.greenfield.value.id,
],
desired_deal_roles=[
DesiredDealRoleConstant.lead_manager.value.id,
],
minimum_equity_percentage_id=EquityPercentageConstant.zero_percent.value.id,
investable_capital=0,
global_assets_under_management=10,
uk_region_locations=[
UKRegionConstant.north_west.value.id,
UKRegionConstant.east_of_england.value.id,
],
)
with freeze_time('2018-01-01 10:00:00'):
south_project = LargeCapitalInvestorProfileFactory(
investor_company=CompanyFactory(
name='South',
),
investor_description='South Project',
investment_types=[
InvestmentTypesConstant.direct_investment_in_project_equity.value.id,
],
global_assets_under_management=60,
)
with freeze_time('2018-01-01 11:00:00'):
north_project = LargeCapitalInvestorProfileFactory(
investable_capital=20,
investor_company=CompanyFactory(
name='North',
),
investor_description='North Project',
uk_region_locations=[
UKRegionConstant.north_west.value.id,
UKRegionConstant.north_east.value.id,
],
other_countries_being_considered=[
CountryConstant.ireland.value.id,
CountryConstant.canada.value.id,
],
global_assets_under_management=70,
)
with freeze_time('2019-01-01'):
investor_profiles = [
LargeCapitalInvestorProfileFactory(
investor_description='Operational construction',
investor_company=investor_company,
investable_capital=950,
construction_risks=[
ConstructionRiskConstant.operational.value.id,
],
minimum_return_rate_id=ReturnRateConstant.up_to_five_percent.value.id,
time_horizons=[
TimeHorizonConstant.up_to_five_years.value.id,
TimeHorizonConstant.five_to_nine_years.value.id,
],
global_assets_under_management=20,
),
LargeCapitalInvestorProfileFactory(
investor_description='Argentina project',
investor_company=argentina_investor_company,
investable_capital=1490,
construction_risks=[
ConstructionRiskConstant.brownfield.value.id,
],
time_horizons=[
TimeHorizonConstant.up_to_five_years.value.id,
],
restrictions=[
RestrictionConstant.inflation_adjustment.value.id,
],
global_assets_under_management=30,
),
frozen_created_on_profile,
LargeCapitalInvestorProfileFactory(
investor_company=CompanyFactory(
address_country_id=CountryConstant.argentina.value.id,
name='2 constructions ltd',
),
investor_description='2 construction risks',
construction_risks=[
ConstructionRiskConstant.brownfield.value.id,
ConstructionRiskConstant.greenfield.value.id,
],
investable_capital=3000,
asset_classes_of_interest=[
AssetClassInterestConstant.biomass.value.id,
],
restrictions=[
RestrictionConstant.inflation_adjustment.value.id,
],
global_assets_under_management=40,
),
LargeCapitalInvestorProfileFactory(
investor_company=CompanyFactory(
name='Deal up ltd',
),
investable_capital=10,
investor_description='Deal up',
deal_ticket_sizes=[
DealTicketSizeConstant.up_to_forty_nine_million.value.id,
],
asset_classes_of_interest=[
AssetClassInterestConstant.biofuel.value.id,
],
time_horizons=[
TimeHorizonConstant.five_to_nine_years.value.id,
],
restrictions=[
RestrictionConstant.liquidity.value.id,
],
minimum_equity_percentage_id=EquityPercentageConstant.zero_percent.value.id,
global_assets_under_management=50,
),
north_project,
south_project,
]
opensearch_with_collector.flush_and_refresh()
yield investor_profiles