def test_checkForPokemonGoErrorHandling(self, urlopen_mock): urlopen_mock.side_effect = urllib.error.URLError("Error retrieving webpage") with self.assertRaises(pokemongo.PokemonGoError): pokemongo.checkForPokemonGo("us")
def test_checkForPokemonGo(self, urlopen_mock): urlopen_mock.return_value = Mock(spec=http.client.HTTPResponse) urlopen_mock.return_value.read.return_value = "<html><meta content=\"Pokémon GO on the App Store\"></meta></html>".encode("utf-8") self.assertEqual(pokemongo.checkForPokemonGo("us"), True) # those bastards!