def test_get_country_by_ip_default(self, geo_mock): """Geo failure should return default country.""" geo_mock.get.return_value = None self.assertEqual(geo.get_country_from_ip('1.1.1.1'), 'XX') geo_mock.get.assert_called_with('1.1.1.1') geo_mock.reset_mock() geo_mock.get.side_effect = ValueError self.assertEqual(geo.get_country_from_ip('1.1.1.1'), 'XX') geo_mock.get.assert_called_with('1.1.1.1')
def test_get_country_by_ip_default(self, geo_mock): """Geo failure should return default country.""" geo_mock.get.return_value = None self.assertEqual(geo.get_country_from_ip("1.1.1.1"), "XX") geo_mock.get.assert_called_with("1.1.1.1") geo_mock.reset_mock() geo_mock.get.side_effect = ValueError self.assertEqual(geo.get_country_from_ip("1.1.1.1"), "XX") geo_mock.get.assert_called_with("1.1.1.1")
def test_get_country_by_ip_bad_data(self, geo_mock): """Bad data from geo should return default country.""" geo_mock.get.return_value = {'fred': 'flintstone'} self.assertEqual(geo.get_country_from_ip('1.1.1.1'), 'XX') geo_mock.get.assert_called_with('1.1.1.1')
def test_get_country_by_ip(self, geo_mock): geo_mock.get.return_value = self.good_geo_data self.assertEqual(geo.get_country_from_ip('1.1.1.1'), 'US') geo_mock.get.assert_called_with('1.1.1.1')
def test_get_country_by_ip_bad_data(self, geo_mock): """Bad data from geo should return default country.""" geo_mock.get.return_value = {"fred": "flintstone"} self.assertEqual(geo.get_country_from_ip("1.1.1.1"), "XX") geo_mock.get.assert_called_with("1.1.1.1")
def test_get_country_by_ip(self, geo_mock): geo_mock.get.return_value = self.good_geo_data self.assertEqual(geo.get_country_from_ip("1.1.1.1"), "US") geo_mock.get.assert_called_with("1.1.1.1")
def test_get_country_by_ip_dev_mode(self, geo_mock): geo_mock.return_value = None assert geo.get_country_from_ip('1.1.1.1') == 'XX'
def test_get_country_by_ip_bad_data(self, geo_mock): """Bad data from geo should return None.""" geo_mock.return_value.get.return_value = {'fred': 'flintstone'} self.assertIsNone(geo.get_country_from_ip('1.1.1.1')) geo_mock.return_value.get.assert_called_with('1.1.1.1')