def test_decode_encoded_from_json(self):
self.app.conf.elasticsearch_save_meta_as_text, prev = False, self.app.conf.elasticsearch_save_meta_as_text
try:
x = ElasticsearchBackend(app=self.app)
result_meta = x._get_result_meta({'solution': 42}, states.SUCCESS, None, None)
assert x.decode(x.encode(result_meta)) == result_meta
finally:
self.app.conf.elasticsearch_save_meta_as_text = prev
def test_decode_not_dict(self, kv_decode_mock):
self.app.conf.elasticsearch_save_meta_as_text, prev = False, self.app.conf.elasticsearch_save_meta_as_text
try:
kv_decode_mock.return_value = sentinel.decoded
x = ElasticsearchBackend(app=self.app)
assert x.decode(sentinel.encoded) == sentinel.decoded
kv_decode_mock.assert_called_once()
finally:
self.app.conf.elasticsearch_save_meta_as_text = prev
def test_decode_none_from_json(self):
self.app.conf.elasticsearch_save_meta_as_text, prev = False, self.app.conf.elasticsearch_save_meta_as_text
try:
x = ElasticsearchBackend(app=self.app)
result_meta = x._get_result_meta(None, states.SUCCESS, None, None)
# result_meta['result'] = x._encode(result_meta['result'])[2]
assert x.decode(result_meta) == result_meta
finally:
self.app.conf.elasticsearch_save_meta_as_text = prev
def test_decode_encoded_exception_as_json(self):
self.app.conf.elasticsearch_save_meta_as_text, prev = False, self.app.conf.elasticsearch_save_meta_as_text
try:
x = ElasticsearchBackend(app=self.app)
try:
raise Exception("failed")
except Exception as exc:
einfo = ExceptionInfo()
result_meta = x._get_result_meta(
x.encode_result(exc, states.FAILURE), states.FAILURE,
einfo.traceback, None)
assert x.decode(x.encode(result_meta)) == result_meta
finally:
self.app.conf.elasticsearch_save_meta_as_text = prev
