def test_dont_enable_if_wrong_operation(self):
enable_upload_callbacks(self.request, 'OtherOperation')
self.assertFalse(self.request.body.enable_callback.called)
def test_enable_object_on_upload_part(self):
enable_upload_callbacks(self.request, 'UploadPart')
self.request.body.enable_callback.assert_called_with()
def test_dont_enable_if_missing_interface(self):
del self.request.body.enable_callback
enable_upload_callbacks(self.request, 'PutObject')
self.assertEqual(self.request.body.method_calls, [])
def test_enable_object_on_put_object(self):
enable_upload_callbacks(self.request, 'PutObject')
self.request.body.enable_callback.assert_called_with()
def test_dont_enable_if_wrong_operation(self):
enable_upload_callbacks(self.request,
'OtherOperation')
self.assertFalse(
self.request.body.enable_callback.called)
def test_dont_enable_if_missing_interface(self):
del self.request.body.enable_callback
enable_upload_callbacks(self.request,
'PutObject')
self.assertEqual(self.request.body.method_calls, [])
def test_enable_object_on_upload_part(self):
enable_upload_callbacks(self.request,
'UploadPart')
self.request.body.enable_callback.assert_called_with()
def test_enable_object_on_put_object(self):
enable_upload_callbacks(self.request,
'PutObject')
self.request.body.enable_callback.assert_called_with()
