diff options
Diffstat (limited to 'webapp/polls/tests/poll_tests.py')
-rw-r--r-- | webapp/polls/tests/poll_tests.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/webapp/polls/tests/poll_tests.py b/webapp/polls/tests/poll_tests.py index 5c173bc..4f58787 100644 --- a/webapp/polls/tests/poll_tests.py +++ b/webapp/polls/tests/poll_tests.py @@ -418,7 +418,8 @@ class PollResultFilesTest(MongoTestCase): ) uploaded_file_path = tmp_path - poll.add_result_files([uploaded_file_path]) + name = 'result.poll_result' + poll.add_result_files([(uploaded_file_path, name)]) results_files = poll.get_result_files() self.assertEqual(1, len(results_files)) self.assertTrue(hasattr(results_files[0], 'get_numero_escuela')) @@ -438,10 +439,10 @@ class AddResultFilesTest(MongoTestCase): uploaded_file = tempfile.NamedTemporaryFile(dir=temp_dir, suffix='.poll_result') uploaded_file_path = uploaded_file.name - uploaded_filename = os.path.basename(uploaded_file.name) + uploaded_filename = 'other_name.poll_result' poll = self.poll - poll.add_result_files([uploaded_file_path]) + poll.add_result_files([(uploaded_file_path, uploaded_filename)]) stub = lambda f: os.path.basename(f.name) results = poll.get_result_files(as_instance_of=stub) result_filename = results[0] |