def __open_or_create_test_file(self, style):
if self.__cur_po_class:
folder = self.GetTopLevelParent().get_root_folder()
if not folder:
folder = os.path.dirname(self.__cur_po_class.file_path)
elif RootFolder.TESTS_FOLDER in os.listdir(folder):
folder = os.path.join(folder, RootFolder.TESTS_FOLDER)
dialog = FileDialog(self,
defaultDir=folder,
style=style,
wildcard='*.py')
if dialog.ShowModal() == ID_OK:
test_file = dialog.GetPath()
load_file = style == FD_OPEN
filename = os.path.basename(test_file)
if StringUtils.is_test_file_name_correct(test_file):
test_file_ui = TestFileUI(self.tabs, test_file,
self.__cur_po_class, load_file)
self.tabs.AddPage(test_file_ui, filename)
self.tabs.SetSelection(self.tabs.GetPageCount() - 1)
if style == FD_SAVE:
test_file_ui.set_file_was_changed()
else:
show_dialog_bad_name(self, filename, 'my_first_test.py')
else:
show_dialog(self, u'Please select class file.',
u'Class file was not opened')
def __open_or_create_test_file(self, style):
if self.__cur_po_class:
folder = self.GetTopLevelParent().get_root_folder()
if not folder:
folder = os.path.dirname(self.__cur_po_class.file_path)
elif RootFolder.TESTS_FOLDER in os.listdir(folder):
folder = os.path.join(folder, RootFolder.TESTS_FOLDER)
dialog = FileDialog(self,
defaultDir=folder,
style=style,
wildcard='*.py')
if dialog.ShowModal() == ID_OK:
test_file = dialog.GetPath()
load_file = style == FD_OPEN
filename = os.path.basename(test_file)
if StringUtils.is_test_file_name_correct(test_file):
test_file_ui = TestFileUI(self.tabs, test_file,
self.__cur_po_class, load_file)
self.tabs.AddPage(test_file_ui, filename)
self.tabs.SetSelection(self.tabs.GetPageCount() - 1)
if style == FD_SAVE:
test_file_ui.set_file_was_changed()
else:
show_dialog_bad_name(self, filename, 'my_first_test.py')
else:
show_dialog(self,
u'Please select class file.',
u'Class file was not opened')
def test_is_test_file_name_correct(self):
self.assertTrue(StringUtils.is_test_file_name_correct('my_new_test.py'))
self.assertTrue(StringUtils.is_test_file_name_correct('2my_new_test.py'))
self.assertTrue(StringUtils.is_test_file_name_correct('my_2new_test.py'))
self.assertFalse(StringUtils.is_test_file_name_correct('test.py'))
self.assertFalse(StringUtils.is_test_file_name_correct('_test.py'))
self.assertFalse(StringUtils.is_test_file_name_correct('A_test.py'))
self.assertFalse(StringUtils.is_test_file_name_correct('9B_test.py'))
self.assertFalse(StringUtils.is_test_file_name_correct('B9_test.py'))
def test_is_test_file_name_correct(self):
self.assertTrue(
StringUtils.is_test_file_name_correct('my_new_test.py'))
self.assertTrue(
StringUtils.is_test_file_name_correct('2my_new_test.py'))
self.assertTrue(
StringUtils.is_test_file_name_correct('my_2new_test.py'))
self.assertFalse(StringUtils.is_test_file_name_correct('test.py'))
self.assertFalse(StringUtils.is_test_file_name_correct('_test.py'))
self.assertFalse(StringUtils.is_test_file_name_correct('A_test.py'))
self.assertFalse(StringUtils.is_test_file_name_correct('9B_test.py'))
self.assertFalse(StringUtils.is_test_file_name_correct('B9_test.py'))
