diff --git a/op_robot_tests/tests_files/initial_data.py b/op_robot_tests/tests_files/initial_data.py index e7a9eb1..7d7ea91 100644 --- a/op_robot_tests/tests_files/initial_data.py +++ b/op_robot_tests/tests_files/initial_data.py @@ -12,7 +12,8 @@ fake_en = Factory.create() def create_fake_doc(): content = fake.text() - tf = NamedTemporaryFile(delete=False) + suffix = fake.random_element(('.txt', '.doc', '.docx', '.pdf')) + tf = NamedTemporaryFile(delete=False, suffix=suffix) tf.write(content) tf.close() return tf.name