def test_top_n_countries_returns_all_by_default(self, geoip_mock,
                                                 current_app):
     current_app.config = {'GEO': True}
     self.create_mock_ip_addresses(geoip_mock)
     d = cache.get_top_n_countries()
     assert d['countries'] == ['England', 'France']
 def test_get_top_countries_returns_empty_dataset_when_no_data(self):
     d = cache.get_top_n_countries()
     assert d == {'countries': [], 'n_task_runs': []}, d
 def test_top_n_countries_returned(self, geoip_mock, current_app):
     current_app.config = {'GEO': True}
     self.create_mock_ip_addresses(geoip_mock)
     d = cache.get_top_n_countries(1)
     assert len(d['countries']) == 1
     assert d['countries'][0] in ['England', 'France']