def testToDictRaisesExceptionWhenNotValid(self):
envelope = Envelope()
from mock import Mock
envelope._is_valid = Mock(return_value=False)
with self.assertRaises(InvalidEnvelopeException):
envelope.to_dict()
def testToDictRaisesExceptionWhenNotValid(self):
envelope = Envelope()
from mock import Mock
envelope._is_valid = Mock(return_value=False)
with self.assertRaises(InvalidEnvelopeException):
envelope.to_dict()
def testToDictReturnProperEnvelope(self):
envelope = Envelope()
envelope.set_code(111)
envelope.set_message("test_message")
envelope.set_results(["foo", "bar"])
from mock import Mock
envelope._is_valid = Mock(return_value=True)
expected_envelope_result = {"results": ["foo", "bar"],
"message": "test_message",
"code": "111",
"errors": []}
self.assertEquals(expected_envelope_result, envelope.to_dict())
def testToDictReturnProperEnvelope(self):
envelope = Envelope()
envelope.set_code(111)
envelope.set_message("test_message")
envelope.set_results(["foo", "bar"])
from mock import Mock
envelope._is_valid = Mock(return_value=True)
expected_envelope_result = {
"results": ["foo", "bar"],
"message": "test_message",
"code": "111",
"errors": []
}
self.assertEquals(expected_envelope_result, envelope.to_dict())