def add_records(self, layer, records):
features = {
'type': 'FeatureCollection',
'features': [record.to_dict() for record in records],
}
endpoint = self._endpoint('add_records', layer=layer)
self._request(endpoint, "POST", json.dumps(features))
def test_dont_Recordify_results(self):
""" _request() is required to return the exact string that the HTTP
server sent to it -- no transforming it, such as by json-decoding and
then constructing a Record. """
EXAMPLE_RECORD_JSONSTR=json.dumps({ 'geometry' : { 'type' : 'Point', 'coordinates' : [D('10.0'), D('11.0')] }, 'id' : 'my_id', 'type' : 'Feature', 'properties' : { 'key' : 'value' , 'type' : 'object' } })
mockhttp = mock.Mock()
mockhttp.request.return_value = ({'status': '200', 'content-type': 'application/json', }, EXAMPLE_RECORD_JSONSTR)
self.client.http = mockhttp
res = self.client._request("http://thing", 'POST')[1]
self.failUnlessEqual(res, EXAMPLE_RECORD_JSONSTR)
def test_annotate(self):
mockhttp = mock.Mock()
headers = {'status': '200', 'content-type': 'application/json'}
mockhttp.request.return_value = (headers, json.dumps(EXAMPLE_ANNOTATE_RESPONSE))
self.client.http = mockhttp
res = self.client.annotate(self.handle, EXAMPLE_ANNOTATIONS, True)
self.assertEqual(mockhttp.method_calls[0][0], 'request')
self.assertEqual(mockhttp.method_calls[0][1][0], 'http://api.simplegeo.com:80/%s/features/%s/annotations.json' % (API_VERSION, self.handle))
self.assertEqual(mockhttp.method_calls[0][1][1], 'POST')
# Make sure client returns a dict.
self.failUnless(isinstance(res, dict))
def annotate(self, simplegeohandle, annotations, private):
if not isinstance(annotations, dict):
raise TypeError("annotations must be of type dict")
if not len(annotations.keys()):
raise ValueError("annotations dict is empty")
for annotation_type in annotations.keys():
if not len(annotations[annotation_type].keys()):
raise ValueError('annotation type "%s" is empty' % annotation_type)
if not isinstance(private, bool):
raise TypeError("private must be of type bool")
data = {"annotations": annotations, "private": private}
endpoint = self._endpoint("annotations", simplegeohandle=simplegeohandle)
return json_decode(self._request(endpoint, "POST", data=json.dumps(data))[1])
def to_json(self):
return json.dumps(self.to_dict())
def mockrequest(*args, **kwargs):
self.assertEqual(args[0], 'http://api.simplegeo.com:80/%s/places' % (API_VERSION,))
self.assertEqual(args[1], 'POST')
bodyobj = json.loads(kwargs['body'])
self.failUnless(not hasattr(bodyobj, 'id'))
methods_called.append(('request', args, kwargs))
mockhttp.request = mockrequest2
return ({'status': '202', 'content-type': 'application/json', 'location': newloc}, json.dumps({'id': handle}))
def test_no_terms_search_by_address(self):
rec1 = Feature((D('11.03'), D('10.04')), simplegeohandle='SG_abcdefghijkmlnopqrstuv', properties={'name': "Bob's House Of Monkeys", 'category': "monkey dealership"})
rec2 = Feature((D('11.03'), D('10.05')), simplegeohandle='SG_abcdefghijkmlnopqrstuv', properties={'name': "Monkey Food 'R' Us", 'category': "pet food store"})
mockhttp = mock.Mock()
mockhttp.request.return_value = ({'status': '200', 'content-type': 'application/json', }, json.dumps({'type': "FeatureColllection", 'features': [rec1.to_dict(), rec2.to_dict()]}))
self.client.places.http = mockhttp
lat = D('11.03')
lon = D('10.04')
self.failUnlessRaises(ValueError, self.client.places.search_by_address, lat, lon) # Someone accidentally passed a lat,lon search_by_address().
addr = '41 Decatur St, San Francisco, CA'
res = self.client.places.search_by_address(addr)
self.failUnless(isinstance(res, (list, tuple)), (repr(res), type(res)))
self.failUnlessEqual(len(res), 2)
self.failUnless(all(isinstance(f, Feature) for f in res))
self.assertEqual(mockhttp.method_calls[0][0], 'request')
self.assertEqual(mockhttp.method_calls[0][1][0], 'http://api.simplegeo.com:80/%s/places/address.json?address=%s' % (API_VERSION, urllib.quote_plus(addr)))
self.assertEqual(mockhttp.method_calls[0][1][1], 'GET')
def test_no_terms_search_by_ip(self):
rec1 = Feature((D('11.03'), D('10.04')), simplegeohandle='SG_abcdefghijkmlnopqrstuv', properties={'name': "Bob's House Of Monkeys", 'category': "monkey dealership"})
rec2 = Feature((D('11.03'), D('10.05')), simplegeohandle='SG_abcdefghijkmlnopqrstuv', properties={'name': "Monkey Food 'R' Us", 'category': "pet food store"})
mockhttp = mock.Mock()
mockhttp.request.return_value = ({'status': '200', 'content-type': 'application/json', }, json.dumps({'type': "FeatureColllection", 'features': [rec1.to_dict(), rec2.to_dict()]}))
self.client.places.http = mockhttp
ipaddr = '192.0.32.10'
res = self.client.places.search_by_ip(ipaddr)
self.failUnless(isinstance(res, (list, tuple)), (repr(res), type(res)))
self.failUnlessEqual(len(res), 2)
self.failUnless(all(isinstance(f, Feature) for f in res))
self.assertEqual(mockhttp.method_calls[0][0], 'request')
self.assertEqual(mockhttp.method_calls[0][1][0], 'http://api.simplegeo.com:80/%s/places/%s.json' % (API_VERSION, ipaddr))
self.assertEqual(mockhttp.method_calls[0][1][1], 'GET')
def test_radius_search_by_address(self):
mockhttp = mock.Mock()
mockhttp.request.return_value = ({'status': '200', 'content-type': 'application/json', }, json.dumps({'type': "FeatureColllection", 'features': []}))
self.client.places.http = mockhttp
lat = D('11.03')
lon = D('10.04')
radius = D('0.01')
self.failUnlessRaises((AssertionError, TypeError), self.client.places.search_by_address, lat, lon, radius=radius) # Someone accidentally passed a lat,lon to search_by_address().
addr = '41 Decatur St, San Francisco, CA'
res = self.client.places.search_by_address(addr, radius=radius)
self.failUnless(isinstance(res, (list, tuple)), (repr(res), type(res)))
self.failUnlessEqual(len(res), 0)
self.assertEqual(mockhttp.method_calls[0][0], 'request')
self.assertEqual(mockhttp.method_calls[0][1][0], 'http://api.simplegeo.com:80/%s/places/address.json?radius=%s&address=%s' % (API_VERSION, radius, urllib.quote_plus(addr)))
self.assertEqual(mockhttp.method_calls[0][1][1], 'GET')
def test_radius_search_by_my_ip(self):
mockhttp = mock.Mock()
mockhttp.request.return_value = ({'status': '200', 'content-type': 'application/json', }, json.dumps({'type': "FeatureColllection", 'features': []}))
self.client.places.http = mockhttp
ipaddr = '192.0.32.10'
radius = D('0.01')
self.failUnlessRaises((AssertionError, TypeError), self.client.places.search_by_my_ip, ipaddr, radius=radius) # Someone accidentally passed an ip addr to search_by_my_ip().
res = self.client.places.search_by_my_ip(radius=radius)
self.failUnless(isinstance(res, (list, tuple)), (repr(res), type(res)))
self.failUnlessEqual(len(res), 0)
self.assertEqual(mockhttp.method_calls[0][0], 'request')
self.assertEqual(mockhttp.method_calls[0][1][0], 'http://api.simplegeo.com:80/%s/places/ip.json?radius=%s' % (API_VERSION, radius))
self.assertEqual(mockhttp.method_calls[0][1][1], 'GET')
def test_radius_search(self):
mockhttp = mock.Mock()
mockhttp.request.return_value = ({'status': '200', 'content-type': 'application/json', }, json.dumps({'type': "FeatureColllection", 'features': []}))
self.client.places.http = mockhttp
lat = D('11.03')
lon = D('10.04')
radius = D('0.01')
res = self.client.places.search(lat, lon, radius=radius)
self.failUnless(isinstance(res, (list, tuple)), (repr(res), type(res)))
self.failUnlessEqual(len(res), 0)
self.assertEqual(mockhttp.method_calls[0][0], 'request')
self.assertEqual(mockhttp.method_calls[0][1][0], 'http://api.simplegeo.com:80/%s/places/%s,%s.json?radius=%s' % (API_VERSION, lat, lon, radius))
self.assertEqual(mockhttp.method_calls[0][1][1], 'GET')
def test_search_by_address_nonascii(self):
rec1 = Feature((D('11.03'), D('10.04')), simplegeohandle='SG_abcdefghijkmlnopqrstuv', properties={'name': "Bob's House Of Monkeys", 'category': "monkey dealership"})
rec2 = Feature((D('11.03'), D('10.05')), simplegeohandle='SG_abcdefghijkmlnopqrstuv', properties={'name': "Monkey Food 'R' Us", 'category': "pet food store"})
mockhttp = mock.Mock()
mockhttp.request.return_value = ({'status': '200', 'content-type': 'application/json', }, json.dumps({'type': "FeatureColllection", 'features': [rec1.to_dict(), rec2.to_dict()]}))
self.client.places.http = mockhttp
lat = D('11.03')
lon = D('10.04')
self.failUnlessRaises(ValueError, self.client.places.search_by_address, lat, lon) # Someone accidentally passed a lat,lon to search_by_address().
addr = u'41 Decatur St, San Francisc❦, CA'
res = self.client.places.search_by_address(addr, query='monkeys', category='animal')
self.failUnless(isinstance(res, (list, tuple)), (repr(res), type(res)))
self.failUnlessEqual(len(res), 2)
self.failUnless(all(isinstance(f, Feature) for f in res))
self.assertEqual(mockhttp.method_calls[-1][0], 'request')
urlused = mockhttp.method_calls[-1][1][0]
cod = urllib.quote_plus(addr.encode('utf-8'))
self.assertEqual(urlused, 'http://api.simplegeo.com:80/%s/places/address.json?q=monkeys&category=animal&address=%s' % (API_VERSION, cod,))
self.assertEqual(mockhttp.method_calls[0][1][1], 'GET')
res = self.client.places.search_by_address(addr, query=u'monke❦s', category=u'ani❦al')
self.failUnless(isinstance(res, (list, tuple)), (repr(res), type(res)))
self.failUnlessEqual(len(res), 2)
self.failUnless(all(isinstance(f, Feature) for f in res))
self.assertEqual(mockhttp.method_calls[-1][0], 'request')
urlused = mockhttp.method_calls[-1][1][0]
quargs = {'q': u'monke❦s', 'category': u'ani❦al', 'address': addr}
equargs = dict( (k, v.encode('utf-8')) for k, v in quargs.iteritems() )
s2quargs = urllib.urlencode(equargs)
self.assertEqual(urlused, 'http://api.simplegeo.com:80/%s/places/address.json?%s' % (API_VERSION, s2quargs))
self.assertEqual(mockhttp.method_calls[0][1][1], 'GET')
def test_search_by_my_ip_nonascii(self):
rec1 = Feature((D('11.03'), D('10.04')), simplegeohandle='SG_abcdefghijkmlnopqrstuv', properties={'name': "Bob's House Of Monkeys", 'category': "monkey dealership"})
rec2 = Feature((D('11.03'), D('10.05')), simplegeohandle='SG_abcdefghijkmlnopqrstuv', properties={'name': "Monkey Food 'R' Us", 'category': "pet food store"})
mockhttp = mock.Mock()
mockhttp.request.return_value = ({'status': '200', 'content-type': 'application/json', }, json.dumps({'type': "FeatureColllection", 'features': [rec1.to_dict(), rec2.to_dict()]}))
self.client.places.http = mockhttp
ipaddr = '192.0.32.10'
self.failUnlessRaises(ValueError, self.client.places.search_by_my_ip, ipaddr) # Someone accidentally passed an ip addr to search_by_my_ip().
res = self.client.places.search_by_my_ip(query='monk❥y', category='animal')
self.failUnless(isinstance(res, (list, tuple)), (repr(res), type(res)))
self.failUnlessEqual(len(res), 2)
self.failUnless(all(isinstance(f, Feature) for f in res))
self.assertEqual(mockhttp.method_calls[-1][0], 'request')
urlused = mockhttp.method_calls[-1][1][0]
urlused = urllib.unquote(urlused).decode('utf-8')
self.assertEqual(urlused, u'http://api.simplegeo.com:80/%s/places/ip.json?q=monk❥y&category=animal' % (API_VERSION,))
self.assertEqual(mockhttp.method_calls[0][1][1], 'GET')
res = self.client.places.search_by_my_ip(query='monk❥y', category='anim❥l')
self.failUnless(isinstance(res, (list, tuple)), (repr(res), type(res)))
self.failUnlessEqual(len(res), 2)
self.failUnless(all(isinstance(f, Feature) for f in res))
self.assertEqual(mockhttp.method_calls[-1][0], 'request')
urlused = mockhttp.method_calls[-1][1][0]
urlused = urllib.unquote(urlused).decode('utf-8')
self.assertEqual(urlused, u'http://api.simplegeo.com:80/%s/places/ip.json?q=monk❥y&category=anim❥l' % (API_VERSION,))
self.assertEqual(mockhttp.method_calls[0][1][1], 'GET')
def test_search(self):
rec1 = Feature((D('11.03'), D('10.04')), simplegeohandle='SG_abcdefghijkmlnopqrstuv', properties={'name': "Bob's House Of Monkeys", 'category': "monkey dealership"})
rec2 = Feature((D('11.03'), D('10.05')), simplegeohandle='SG_abcdefghijkmlnopqrstuv', properties={'name': "Monkey Food 'R' Us", 'category': "pet food store"})
mockhttp = mock.Mock()
mockhttp.request.return_value = ({'status': '200', 'content-type': 'application/json', }, json.dumps({'type': "FeatureColllection", 'features': [rec1.to_dict(), rec2.to_dict()]}))
self.client.places.http = mockhttp
self.failUnlessRaises(ValueError, self.client.places.search, -91, 100)
self.failUnlessRaises(ValueError, self.client.places.search, -81, 361)
lat = D('11.03')
lon = D('10.04')
res = self.client.places.search(lat, lon, query='monkeys', category='animal')
self.failUnless(isinstance(res, (list, tuple)), (repr(res), type(res)))
self.failUnlessEqual(len(res), 2)
self.failUnless(all(isinstance(f, Feature) for f in res))
self.assertEqual(mockhttp.method_calls[0][0], 'request')
self.assertEqual(mockhttp.method_calls[0][1][0], 'http://api.simplegeo.com:80/%s/places/%s,%s.json?q=monkeys&category=animal' % (API_VERSION, lat, lon))
self.assertEqual(mockhttp.method_calls[0][1][1], 'GET')
