Coverage for src/robotide/editor/editorcreator.py: 100%

1# Copyright 2008-2015 Nokia Networks 

2# Copyright 2016- Robot Framework Foundation 

3# 

4# Licensed under the Apache License, Version 2.0 (the "License"); 

5# you may not use this file except in compliance with the License. 

6# You may obtain a copy of the License at 

7# 

8# http://www.apache.org/licenses/LICENSE-2.0 

9# 

10# Unless required by applicable law or agreed to in writing, software 

11# distributed under the License is distributed on an "AS IS" BASIS, 

12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

13# See the License for the specific language governing permissions and 

14# limitations under the License. 

15 

16 

17from .. import robotapi 2 ctx1ab

18from .. import controller 2 ctx1ab

19from ..controller.dataloader import TestDataDirectoryWithExcludes 2 ctx1ab

20from ..controller import filecontrollers 2 ctx1ab

21from ..controller.settingcontrollers import VariableController 2 ctx1ab

22from .editors import (InitFileEditor, TestCaseFileEditor, WelcomePage, ResourceFileEditor) 2 ctx1ab

23from .macroeditors import TestCaseEditor, UserKeywordEditor 2 ctx1ab

24 

25 

26def variable_editor_chooser(plugin, parent, _controller, tree): 2 ctx1ab

27 _controller = _controller.datafile_controller 2 ctx1dc

28 editor_class = plugin.get_editor(_controller.data.__class__) 2 ctx1dc

29 return editor_class(plugin, parent, _controller, tree) 2 ctx1dc

30 

31 

32class EditorCreator(object): 2 ctx1ab

33 # DEBUG: Should not use robot.model classes here 

34 _EDITORS = ((robotapi.TestDataDirectory, InitFileEditor), 2 ctx1ab

35 (robotapi.ResourceFile, ResourceFileEditor), 

36 (robotapi.TestCase, TestCaseEditor), 

37 (robotapi.TestCaseFile, TestCaseFileEditor), 

38 (robotapi.UserKeyword, UserKeywordEditor), 

39 (robotapi.Variable, variable_editor_chooser), 

40 (TestDataDirectoryWithExcludes, InitFileEditor)) 

41 

42 def __init__(self, editor_registerer): 2 ctx1ab

43 self._editor_registerer = editor_registerer (empty)

44 self._editor = None (empty)

45 

46 def register_editors(self): 2 ctx1ab

47 for item, editorclass in self._EDITORS: (empty)

48 self._editor_registerer(item, editorclass) (empty)

49 

50 def editor_for(self, plugin, editor_panel, tree): 2 ctx1ab

51 self._editor = self._create_editor(editor_panel, plugin, tree) 10 ctx1agdiefchjk

52 return self._editor 10 ctx1agdiefchjk

53 

54 def _create_editor(self, editor_panel, plugin, tree): 2 ctx1ab

55 _controller = plugin.get_selected_item() 10 ctx1agdiefchjk

56 if self._invalid(_controller): 10 ctx1agdiefchjk

57 # http://code.google.com/p/robotframework-ride/issues/detail?id=1092 

58 if self._editor and tree and (not tree.datafile_nodes or 5 ctx1aihjk

59 self._only_resource_files(tree)): 

60 self._editor.w_destroy() (empty)

61 self._editor = None (empty)

62 return None (empty)

63 if self._editor: 5 ctx1aihjk

64 return self._editor 1 ctx1h

65 return WelcomePage(editor_panel) 5 ctx1aihjk

66 if self._should_use_old_editor(_controller): 5 ctx1gdefc

67 return self._editor 1 ctx1c

68 return self._create_new_editor(_controller, editor_panel, plugin, tree) 5 ctx1gdefc

69 

70 @staticmethod 2 ctx1ab

71 def _invalid(_controller): 2 ctx1ab

72 return not _controller or _controller.data is None or \ 10 ctx1agdiefchjk

73 isinstance(_controller, controller.Project) or \ 

74 isinstance(_controller, filecontrollers.ExcludedDirectoryController) 

75 

76 def _should_use_old_editor(self, _controller): 2 ctx1ab

77 return self._editor and \ 5 ctx1gdefc

78 isinstance(_controller, VariableController) and \ 

79 _controller.datafile_controller is self._editor.controller 

80 

81 def _create_new_editor(self, _controller, editor_panel, plugin, tree): 2 ctx1ab

82 editor_class = plugin.get_editor(_controller.data.__class__) 5 ctx1gdefc

83 if self._editor: 5 ctx1gdefc

84 self._editor.w_destroy() 2 ctx1ef

85 editor_panel.Show(False) 5 ctx1gdefc

86 return editor_class(plugin, editor_panel, _controller, tree) 5 ctx1gdefc

87 

88 @staticmethod 2 ctx1ab

89 def _only_resource_files(tree): 2 ctx1ab

90 return all([tree.node_is_resource_file(node) 1 ctx1l

91 for node in tree.datafile_nodes])