def test_put_upload_segment_resource_bad_offset(self, mock_factory): """ Test the UploadSegmentResourceView with an invalid offset value (not an int) """ mock_upload_manager = mock.MagicMock() mock_factory.content_upload_manager.return_value = mock_upload_manager request = mock.MagicMock() upload_segment_resource = UploadSegmentResourceView() self.assertRaises(InvalidValue, upload_segment_resource.put, request, 'mock_id', 'invalid_offset')
def test_put_upload_segment_resource(self, mock_factory, mock_resp): """ Test the UploadSegmentResourceView under normal conditions """ mock_upload_manager = mock.MagicMock() mock_factory.content_upload_manager.return_value = mock_upload_manager request = mock.MagicMock() request.body = 'upload these bits' upload_segment_resource = UploadSegmentResourceView() response = upload_segment_resource.put(request, 'mock_id', 4) mock_upload_manager.save_data.assert_called_once_with('mock_id', 4, 'upload these bits') mock_resp.assert_called_once_with(None) self.assertTrue(response is mock_resp.return_value)