def test_get_from_v3_postal_code(self):
postal_code = "48867"
Ziptastic.api_key = "abc123"
result = Ziptastic.get_from_postal_code(postal_code)
self.assertIn("postal_code", result[0])
self.assertEqual("https://zip.getziptastic.com/v3/US/48867", self.urllib_mock.call_args[0][0].get_full_url())
def test_get_from_v2_postal_code(self):
self.setUpV3()
postal_code = "48867"
result = Ziptastic.get_from_postal_code(postal_code)
self.assertIn("postal_code", result)
self.assertEqual("https://zip.getziptastic.com/v2/US/48867", self.urllib_mock.call_args[0][0].get_full_url())
def test_get_from_v2_postal_code(self, m):
url = 'https://zip.getziptastic.com/v2/US/48867'
m.get(url, text=compare_v2_json)
postal_code = '48867'
ziptastic = Ziptastic('')
result = ziptastic.get_from_postal_code(postal_code)
req = m.request_history[0]
eq_(url, req.url)
eq_(loads(compare_v2_json), result)
def test_object_is_returned(self):
result = Ziptastic.get_from_postal_code("48867")
self.assertTrue(type(result), "object")
