def test_failed_upload(self):
"""
Test the upload class method failed upload
"""
obj_hash = {
'_id': 127,
u'foo': u'bar'
}
url = "http://127.0.0.1:9200/pacifica/ElasticAPI/127"
obj = ElasticAPI()
obj.id = 127
setattr(obj, 'to_hash', lambda: obj_hash)
response_body = {
"status": "failed!"
}
httpretty.register_uri(httpretty.PUT, url,
body=dumps(response_body),
content_type="application/json",
status=500)
#pylint: disable=broad-except
try:
ElasticAPI.elastic_upload(obj)
except Exception, ex:
self.assertEqual(httpretty.last_request().method, "PUT")
self.assertEqual(str(ex), "upload_obj: 500\n")
def test_upload(self):
"""
Test the upload class method
"""
obj_hash = {
'_id': 127,
u'foo': u'bar'
}
url = "http://127.0.0.1:9200/pacifica/ElasticAPI/127"
obj = ElasticAPI()
obj.id = 127
setattr(obj, 'to_hash', lambda: obj_hash)
response_body = {
"status": "uploaded!"
}
httpretty.register_uri(httpretty.PUT, url,
body=dumps(response_body),
content_type="application/json")
ElasticAPI.elastic_upload(obj)
self.assertEqual(httpretty.last_request().method, "PUT")
sent_body = httpretty.last_request().parsed_body
self.assertTrue(isinstance(sent_body, unicode))
sent_body = loads(sent_body)
self.assertTrue('_id' not in sent_body)
for key, value in sent_body.iteritems():
self.assertEqual(value, obj_hash[key])
def test_delete(self):
"""
Test the delete class method
"""
url = "http://127.0.0.1:9200/pacifica/ElasticAPI/127"
obj = ElasticAPI()
obj.id = 127
response_body = {
"status": "deleted!"
}
httpretty.register_uri(httpretty.DELETE, url,
body=dumps(response_body),
content_type="application/json")
ElasticAPI.elastic_delete(obj)
self.assertEqual(httpretty.last_request().method, "DELETE")
def test_elastic_mapping(self):
"""
Test the elastic_mapping class method
"""
url = "http://127.0.0.1:9200/pacifica/_mapping/ElasticAPI"
obj = ElasticAPI()
obj.id = 127
response_body = {
"status": "mapped!"
}
httpretty.register_uri(httpretty.PUT, url,
body=dumps(response_body),
content_type="application/json")
ElasticAPI.create_elastic_mapping()
self.assertEqual(httpretty.last_request().method, "PUT")
def test_failed_delete(self):
"""
Test the delete class method failed upload
"""
url = "http://127.0.0.1:9200/pacifica/ElasticAPI/127"
obj = ElasticAPI()
obj.id = 127
response_body = {
"status": "error!"
}
httpretty.register_uri(httpretty.DELETE, url,
body=dumps(response_body),
content_type="application/json",
status=500)
#pylint: disable=broad-except
try:
ElasticAPI.elastic_delete(obj)
except Exception, ex:
self.assertEqual(httpretty.last_request().method, "DELETE")
self.assertEqual(str(ex), "elastic_delete_obj: 500\n")
def test_failed_elastic_mapping(self):
"""
Test the elastic mapping class method failed upload
"""
url = "http://127.0.0.1:9200/pacifica/_mapping/ElasticAPI"
obj = ElasticAPI()
obj.id = 127
response_body = {
"status": "error!"
}
httpretty.register_uri(httpretty.PUT, url,
body=dumps(response_body),
content_type="application/json",
status=500)
#pylint: disable=broad-except
try:
ElasticAPI.create_elastic_mapping()
except Exception, ex:
self.assertEqual(httpretty.last_request().method, "PUT")
self.assertEqual(str(ex), "create_elastic_mapping: 500\n")
