def test_simple_location__fall_back_to_trading_block(barrier_data_5):
"""
When there's no country for whatever reason, fall back to trading bloc
"""
barrier_data_5["country"] = {}
barrier = Barrier(barrier_data_5)
assert "European Union" == barrier.simple_location
def barrier_details(self, version="latest", id=None, headers=None):
headers = headers or {}
uri = self.versioned_data_uri(version)
filters = {
"id": id,
}
barriers = self.get(uri, filters, headers=headers) or ()
try:
return Barrier(barriers[0])
except (IndexError, TypeError):
raise Http404("Barrier does not exist")
def barriers_list(self, version="latest", filters=None, sort_by=None, headers=None):
headers = headers or {}
uri = self.versioned_data_uri(version)
barriers = self.get(uri, filters, headers=headers) or ()
count = len(barriers)
barriers = [Barrier(d) for d in barriers]
if sort_by == "location":
barriers = self.sort_by_location(barriers)
elif sort_by == "sectors":
barriers = self.sort_by_sectors(barriers)
data = {
"all": barriers,
"count": count,
}
return data
"trading_bloc": None,
},
{
"id": 3,
"title": "Barrier 3",
"country": CountryFixtures.nepal,
"location": "Nepal",
"sectors": [SectorFixtures.railways],
"trading_bloc": None,
},
{
"id": 4,
"title": "Barrier 4",
"country": CountryFixtures.brazil,
"location": "Brazil",
"sectors": [SectorFixtures.energy],
"trading_bloc": None,
},
{
"id": 5,
"title": "Barrier 5",
"country": CountryFixtures.spain,
"location": "Spain",
"sectors": [SectorFixtures.energy],
"trading_bloc": TradingBlocFixtures.eu,
},
]
dummy_barrier_list_results = [Barrier(item) for item in dummy_barrier_list_results_raw]
def test_is_resolved_text__yes_without_date(barrier_data_1):
data = barrier_data_1.copy()
data.pop("status_date")
barrier = Barrier(data)
assert "Yes" == barrier.is_resolved_text
def test_categories_list(barrier_data_1):
barrier = Barrier(barrier_data_1)
assert ["Wibble", "Wobble"] == barrier.categories_list
def test_reported_on(barrier_data_1):
barrier = Barrier(barrier_data_1)
assert "2020-12-01 00:00:00" == str(barrier.reported_on)
def test_last_published_on(barrier_data_1):
barrier = Barrier(barrier_data_1)
assert "2020-12-12 00:00:00" == str(barrier.last_published_on)
def test_simple_location__returns_country(barrier_data_5):
barrier = Barrier(barrier_data_5)
assert "Spain" == barrier.simple_location
def test_location(barrier_data_1):
barrier = Barrier(barrier_data_1)
assert "Afghanistan" == barrier.location
def test_is_resolved_text__yes_with_date(barrier_data_1):
barrier = Barrier(barrier_data_1)
assert "Yes - December 2020" == barrier.is_resolved_text
def test_country(barrier_data_1):
barrier = Barrier(barrier_data_1)
assert "Afghanistan" == barrier.country
def test_is_resolved_text__no(barrier_data_1):
data = barrier_data_1.copy()
data["is_resolved"] = False
barrier = Barrier(data)
assert "No" == barrier.is_resolved_text
def test_status_date(barrier_data_1):
barrier = Barrier(barrier_data_1)
assert "2020-12-12 00:00:00" == str(barrier.status_date)