1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94
|
# (C) Copyright 2005-2023 Enthought, Inc., Austin, TX
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in LICENSE.txt and may be redistributed only under
# the conditions described in the aforementioned license. The license
# is also available online at http://www.enthought.com/licenses/BSD.txt
#
# Thanks for using Enthought open source!
import os
from pathlib import Path
import unittest
from traits.api import File, HasTraits, TraitError
from traits.testing.optional_dependencies import requires_traitsui
class ExampleModel(HasTraits):
file_name = File(exists=True)
new_file_name = File(exists=False)
class FastExampleModel(HasTraits):
file_name = File()
class FileTestCase(unittest.TestCase):
def test_valid_file(self):
example_model = ExampleModel(file_name=__file__)
example_model.file_name = os.path.__file__
def test_valid_pathlike_file(self):
ExampleModel(file_name=Path(__file__))
def test_invalid_file(self):
example_model = ExampleModel(file_name=__file__)
with self.assertRaises(TraitError):
example_model.file_name = "not_valid_path!#!#!#"
def test_invalid_pathlike_file(self):
example_model = ExampleModel(file_name=__file__)
with self.assertRaises(TraitError):
example_model.file_name = Path("not_valid_path!#!#!#")
def test_directory(self):
example_model = ExampleModel(file_name=__file__)
with self.assertRaises(TraitError):
example_model.file_name = os.path.dirname(__file__)
def test_pathlike_directory(self):
example_model = ExampleModel(file_name=__file__)
with self.assertRaises(TraitError):
example_model.file_name = Path(os.path.dirname(__file__))
def test_invalid_type(self):
example_model = ExampleModel(file_name=__file__)
with self.assertRaises(TraitError):
example_model.file_name = 11
def test_fast(self):
example_model = FastExampleModel(file_name=__file__)
example_model.path = "."
def test_info_text(self):
example_model = ExampleModel()
with self.assertRaises(TraitError) as exc_cm:
example_model.file_name = 47
self.assertIn("a string or os.PathLike object", str(exc_cm.exception))
self.assertIn("referring to an existing file", str(exc_cm.exception))
with self.assertRaises(TraitError) as exc_cm:
example_model.new_file_name = 47
self.assertIn("a string or os.PathLike object", str(exc_cm.exception))
self.assertNotIn("exist", str(exc_cm.exception))
class TestCreateEditor(unittest.TestCase):
@requires_traitsui
def test_exists_controls_editor_dialog_style(self):
x = File(exists=True)
editor = x.create_editor()
self.assertEqual(editor.dialog_style, "open")
x = File(exists=False)
editor = x.create_editor()
self.assertEqual(editor.dialog_style, "save")
|