class ClientTest(unittest.TestCase):
def setUp(self):
self.client = Client(MY_OAUTH_KEY, MY_OAUTH_SECRET, API_VERSION, API_HOST, API_PORT)
self.query_lat = D('37.8016')
self.query_lon = D('-122.4783')
def test_wrong_endpoint(self):
self.assertRaises(Exception, self.client._endpoint, 'wrongwrong')
def test_missing_argument(self):
self.assertRaises(Exception, self.client._endpoint, 'feature')
def test_get_point_feature(self):
mockhttp = mock.Mock()
mockhttp.request.return_value = ({'status': '200', 'content-type': 'application/json', 'thingie': "just to see if you're listening"}, EXAMPLE_POINT_BODY)
self.client.http = mockhttp
res = self.client.get_feature("SG_4bgzicKFmP89tQFGLGZYy0_34.714646_-86.584970")
self.assertEqual(mockhttp.method_calls[0][0], 'request')
self.assertEqual(mockhttp.method_calls[0][1][0], 'http://api.simplegeo.com:80/%s/features/%s.json' % (API_VERSION, "SG_4bgzicKFmP89tQFGLGZYy0_34.714646_-86.584970"))
self.assertEqual(mockhttp.method_calls[0][1][1], 'GET')
# the code under test is required to have json-decoded this before handing it back
self.failUnless(isinstance(res, Feature), (repr(res), type(res)))
self.failUnless(self.client.get_most_recent_http_headers(), {'status': '200', 'content-type': 'application/json', 'thingie': "just to see if you're listening"})
def test_get_polygon_feature(self):
mockhttp = mock.Mock()
mockhttp.request.return_value = ({'status': '200', 'content-type': 'application/json', }, EXAMPLE_BODY)
self.client.http = mockhttp
res = self.client.get_feature("SG_4bgzicKFmP89tQFGLGZYy0_34.714646_-86.584970")
self.assertEqual(mockhttp.method_calls[0][0], 'request')
self.assertEqual(mockhttp.method_calls[0][1][0], 'http://api.simplegeo.com:80/%s/features/%s.json' % (API_VERSION, "SG_4bgzicKFmP89tQFGLGZYy0_34.714646_-86.584970"))
self.assertEqual(mockhttp.method_calls[0][1][1], 'GET')
# the code under test is required to have json-decoded this before handing it back
self.failUnless(isinstance(res, Feature), (repr(res), type(res)))
def test_type_check_request(self):
self.failUnlessRaises(TypeError, self.client._request, 'whatever', 'POST', {'bogus': "non string"})
def test_get_feature_bad_json(self):
mockhttp = mock.Mock()
mockhttp.request.return_value = ({'status': '200', 'content-type': 'application/json', }, EXAMPLE_BODY + 'some crap')
self.client.http = mockhttp
try:
self.client.get_feature("SG_4bgzicKFmP89tQFGLGZYy0_34.714646_-86.584970")
except DecodeError, e:
self.failUnlessEqual(e.code,None,repr(e.code))
self.failUnless("Could not decode JSON" in e.msg, repr(e.msg))
erepr = repr(e)
self.failUnless('JSONDecodeError' in erepr, erepr)
self.assertEqual(mockhttp.method_calls[0][0], 'request')
self.assertEqual(mockhttp.method_calls[0][1][0], 'http://api.simplegeo.com:80/%s/features/%s.json' % (API_VERSION, "SG_4bgzicKFmP89tQFGLGZYy0_34.714646_-86.584970"))
self.assertEqual(mockhttp.method_calls[0][1][1], 'GET')
def test_get_feature_useful_validation_error_message(self):
c = Client('whatever', 'whatever')
try:
c.get_feature('wrong thing')
except TypeError, e:
self.failUnless(str(e).startswith('simplegeohandle is required to match '), str(e))
