def test_import_request_with_no_file_returns_none(self, mock_request): mock_request.method = 'POST' mock_request.files = dict(somekey='somevalue') form = BulkTaskLocalCSVImportForm(**self.form_data) return_value = form.get_import_data() assert return_value['type'] is 'localCSV' and return_value[ 'csv_filename'] is None
def test_import_request_with_no_file_returns_none(self, mock_request): from flask import flash mock_request.method = 'POST' mock_request.files = dict(somekey='somevalue') form = BulkTaskLocalCSVImportForm(**self.form_data) return_value = form.get_import_data() assert return_value['type'] is 'localCSV' and return_value['csv_filename'] is None
def test_import_invalid_local_csv_file_ext_returns_none(self, mock_request): mock_request.method = 'POST' mock_file = MagicMock() mock_file.filename = 'sample.txt' mock_request.files = dict(file=mock_file) form = BulkTaskLocalCSVImportForm(**self.form_data) return_value = form.get_import_data() assert return_value['type'] is 'localCSV' and return_value['csv_filename'] is None
def test_import_blank_local_csv_file_returns_none(self, mock_request): mock_request.method = 'POST' mock_file = MagicMock() mock_file.filename = '' mock_request.files = dict(file=mock_file) form = BulkTaskLocalCSVImportForm(**self.form_data) return_value = form.get_import_data() assert return_value['type'] is 'localCSV' and return_value[ 'csv_filename'] is None
def test_import_upload_path_works(self, mock_user, mock_request): mock_user.id = 1 mock_request.method = 'POST' mock_file = MagicMock() mock_file.filename = 'sample.csv' mock_request.files = dict(file=mock_file) form = BulkTaskLocalCSVImportForm(**self.form_data) return_value = form.get_import_data() assert return_value['type'] is 'localCSV', return_value assert 'user_1/sample.csv' in return_value['csv_filename'], return_value
def test_import_upload_path_works(self, mock_user, mock_request): mock_user.id = 1 mock_request.method = 'POST' mock_file = MagicMock() mock_file.filename = 'sample.csv' mock_request.files = dict(file=mock_file) form = BulkTaskLocalCSVImportForm(**self.form_data) return_value = form.get_import_data() assert return_value['type'] is 'localCSV', return_value assert 'user_1/sample.csv' in return_value[ 'csv_filename'], return_value
def test_import_upload_path_works(self, mock_user, mock_request, mock_upload): url = 'https://s3.amazonaws.com/bucket/hello.csv' patch_dict = {'S3_IMPORT_BUCKET': 'bucket'} with patch.dict(self.flask_app.config, patch_dict): mock_upload.return_value = url mock_user.id = 1 mock_request.method = 'POST' mock_file = MagicMock() mock_file.filename = 'sample.csv' mock_request.files = dict(file=mock_file) form = BulkTaskLocalCSVImportForm(**self.form_data) return_value = form.get_import_data() assert return_value['type'] is 'localCSV', return_value assert return_value['csv_filename'] == url, return_value
def test_import_upload_path_ioerror(self, mock_user, mock_request, mock_uploader): mock_user.id = 1 mock_request.method = 'POST' mock_file = MagicMock() mock_file.filename = 'sample.csv' mock_request.files = dict(file=mock_file) form = BulkTaskLocalCSVImportForm(**self.form_data) assert_raises(IOError, form.get_import_data)