def test_submit__with_photos_and_user(self, mock_notifier, mock_debug,
mock_do_post,
mock_response):
# Mock typically uses side_effect() to specify multiple return
# value; clunky API but works fine.
do_post_return_values = [
(200, '''{
"user": "******",
"photo_post_url": "/photos/",
"rack_url": "/racks/1"
}'''),
(200, 'OK')]
def side_effect(*args, **kw):
return do_post_return_values.pop(0)
mock_do_post.side_effect = side_effect
notifier = mock_notifier()
http = FixcityHttp(notifier)
# Mock photo needs to be just file-like enough.
mock_photo_file = mock.Mock()
mock_photo_file.name = 'foo.jpg'
mock_photo_file.fileno.side_effect = AttributeError()
mock_photo_file.tell.return_value = 12345
mock_photo_file.read.return_value = ''
self.assertEqual(http.submit({'photos': {'photo': mock_photo_file}}),
None)
self.assertEqual(notifier.on_submit_success.call_count, 1)
vars = notifier.on_submit_success.call_args[0][0]
self.assert_(vars.has_key('rack_url'))
self.assert_(vars.has_key('rack_user'))
def test_submit__successful_empty_result(self, mock_notifier, mock_debug,
mock_request, mock_response):
response = mock_response()
notifier = mock_notifier()
response.status = 200
mock_request.return_value = (response, '{}')
http = FixcityHttp(notifier)
self.assertEqual(http.submit({}), None)
self.assertEqual(notifier.on_submit_success.call_count, 0)
def test_do_post__socket_error(self, mock_notifier, mock_debug,
mock_request):
import socket
notifier = mock_notifier()
mock_request.side_effect = socket.error("kaboom")
http = FixcityHttp(notifier)
status, content = http.do_post('http://example.com', 'test body')
self.assertEqual(status, None)
self.assertEqual(content, None)
self.assertEqual(notifier.on_server_temp_failure.call_count, 1)
def test_submit__server_error(self, mock_debug,
mock_request, mock_response, mock_notifier):
mock_response.status = 500
mock_request.return_value = (mock_response, 'blah')
http = FixcityHttp(mock_notifier)
data = {}
http.submit(data)
self.assertEqual(mock_notifier.on_server_error.call_count, 1)
args = mock_notifier.on_server_error.call_args[0]
self.assertEqual(args, ('blah',))
def test_do_post__success(self, mock_notifier, mock_debug, mock_request,
mock_response):
response = mock_response()
notifier = mock_notifier()
response.status = 200
mock_request.return_value = (response, 'hello POST world')
http = FixcityHttp(mock_notifier())
status, content = http.do_post('http://example.com', 'test body')
self.assertEqual(content, 'hello POST world')
self.assertEqual(status, 200)
self.failIf(notifier.bounce.call_count)
def test_do_post_json(self, mock_notifier, mock_debug, mock_request,
mock_response):
response = mock_response()
response.status = 200
notifier = mock_notifier()
mock_request.return_value = (response, '{"foo": "bar"}')
http = FixcityHttp(notifier)
content = http.do_post_json('http://example.com',
"{'some key': 'some value'}")
self.assertEqual(content, {'foo': 'bar'})
self.failIf(notifier.bounce.call_count)
def test_do_post_json__parse_error(self, mock_notifier, mock_log,
mock_request, mock_response):
response = mock_response()
response.status = 200
notifier = mock_notifier()
mock_request.return_value = (response, 'this is not my beautiful JSON')
http = FixcityHttp(notifier)
content = http.do_post_json('http://example.com',
"{'some key': 'some value'}")
self.assertEqual(content, None)
self.assertEqual(notifier.on_server_error.call_count, 1)
self.assertEqual(mock_log.call_count, 2)
def test_do_post__500_error(self, mock_notifier, mock_debug, mock_request,
mock_response):
response = mock_response()
notifier = mock_notifier()
response.status = 500
mock_request.return_value = (response, 'hello POST world')
http = FixcityHttp(notifier)
status, content = http.do_post('http://example.com', 'test body')
self.assertEqual(status, 500)
self.assertEqual(content, 'hello POST world')
self.assertEqual(notifier.on_server_error.call_count, 1)
self.assertEqual(notifier.on_server_error.call_args[0][0], content)
def test_do_post_json__validation_errors(self, mock_notifier, mock_debug,
mock_request, mock_response):
response = mock_response()
notifier = mock_notifier()
response.status = 200
from django.utils import simplejson as json
error_body = json.dumps(
{'errors': {'title': ['This field is required.']}})
mock_request.return_value = (response, error_body)
http = FixcityHttp(notifier)
content = http.do_post_json('http://example.com',
{'user': '******', 'some key': 'some value'})
self.assertEqual(content, json.loads(error_body))
self.assertEqual(notifier.on_user_error.call_count, 1)
def test_submit__user_errors(self, mock_do_post, mock_info,
mock_shorten, mock_notifier):
http = FixcityHttp(mock_notifier)
mock_do_post.return_value = (200, '{"errors": {"any": "thing at all"}}')
mock_shorten.return_value = 'http://short_url/'
data = {'title': 'TITLE', 'address': 'ADDRESS', 'twitter_user': '******',
'date': 'DATE', 'twitter_id': 123}
http.submit(data)
self.assertEqual(mock_do_post.call_count, 1)
# We notified the user of failure.
self.assertEqual(mock_notifier.on_user_error.call_count, 1)
notify_args, notify_kwargs = mock_notifier.on_user_error.call_args
self.assertEqual(notify_args[0], data)
self.assertEqual(notify_args[1], {"any": "thing at all"})