Example #1
0
    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')
Example #2
0
    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")
Example #3
0
 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')
Example #4
0
 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')
Example #5
0
 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")
Example #6
0
 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")
Example #7
0
 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'
Example #8
0
 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')
Example #9
0
 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')