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)