def delete(self, *args, **kwargs):
'''Intercept the delete function to also remove this record from CloudSearch
if it is indexed'''
if self._cs_document_endpoint:
from boto.cloudsearch.document import DocumentServiceConnection
conn = DocumentServiceConnection(endpoint=self._cs_document_endpoint)
doc_id = self.id
doc_id = b32encode(doc_id).lower().replace('=', '_')
conn.delete(doc_id, int(time.time()))
conn.commit()
return Item.delete(self, *args, **kwargs)
def test_cloudsearch_delete_results(self):
"""
Check that the result of a single document deletion is parsed properly.
"""
document = DocumentServiceConnection(
endpoint="doc-demo-userdomain.us-east-1.cloudsearch.amazonaws.com")
document.delete("5", "10")
doc = document.commit()
self.assertEqual(doc.status, 'success')
self.assertEqual(doc.adds, 0)
self.assertEqual(doc.deletes, 1)
def test_cloudsearch_delete_results(self):
"""
Check that the result of a single document deletion is parsed properly.
"""
document = DocumentServiceConnection(
endpoint="doc-demo-userdomain.us-east-1.cloudsearch.amazonaws.com")
document.delete("5", "10")
doc = document.commit()
self.assertEqual(doc.status, 'success')
self.assertEqual(doc.adds, 0)
self.assertEqual(doc.deletes, 1)
def test_cloudsearch_delete(self):
"""
Test that the request for a single document deletion is done properly.
"""
document = DocumentServiceConnection(
endpoint="doc-demo-userdomain.us-east-1.cloudsearch.amazonaws.com")
document.delete("5", "10")
document.commit()
args = json.loads(HTTPretty.last_request.body)[0]
self.assertEqual(args['version'], '10')
self.assertEqual(args['type'], 'delete')
self.assertEqual(args['id'], '5')
def test_cloudsearch_delete(self):
"""
Test that the request for a single document deletion is done properly.
"""
document = DocumentServiceConnection(
endpoint="doc-demo-userdomain.us-east-1.cloudsearch.amazonaws.com")
document.delete("5", "10")
document.commit()
args = json.loads(HTTPretty.last_request.body.decode('utf-8'))[0]
self.assertEqual(args['version'], '10')
self.assertEqual(args['type'], 'delete')
self.assertEqual(args['id'], '5')
def test_cloudsearch_delete_multiples(self):
document = DocumentServiceConnection(
endpoint="doc-demo-userdomain.us-east-1.cloudsearch.amazonaws.com")
document.delete("5", "10")
document.delete("6", "11")
document.commit()
args = json.loads(HTTPretty.last_request.body)
self.assertEqual(len(args), 2)
for arg in args:
self.assertEqual(arg['type'], 'delete')
if arg['id'] == '5':
self.assertEqual(arg['version'], '10')
elif arg['id'] == '6':
self.assertEqual(arg['version'], '11')
else: # Unknown result out of AWS that shouldn't be there
self.assertTrue(False)
def test_cloudsearch_delete_multiples(self):
document = DocumentServiceConnection(
endpoint="doc-demo-userdomain.us-east-1.cloudsearch.amazonaws.com")
document.delete("5", "10")
document.delete("6", "11")
document.commit()
args = json.loads(HTTPretty.last_request.body.decode('utf-8'))
self.assertEqual(len(args), 2)
for arg in args:
self.assertEqual(arg['type'], 'delete')
if arg['id'] == '5':
self.assertEqual(arg['version'], '10')
elif arg['id'] == '6':
self.assertEqual(arg['version'], '11')
else: # Unknown result out of AWS that shouldn't be there
self.assertTrue(False)
