77import com .intellij .openapi .actionSystem .AnAction ;
88import com .intellij .openapi .actionSystem .AnActionEvent ;
99import com .intellij .openapi .actionSystem .PlatformDataKeys ;
10+ import com .intellij .openapi .editor .Document ;
11+ import com .intellij .openapi .fileEditor .FileDocumentManager ;
1012import com .intellij .openapi .project .Project ;
1113import com .intellij .openapi .vfs .VirtualFile ;
1214import com .intellij .openapi .wm .ToolWindow ;
@@ -46,6 +48,14 @@ public void update(AnActionEvent event) {
4648 event .getPresentation ().setEnabled (true );
4749 }
4850
51+ protected void saveFiles (PhpClass currentTestClass , Project project ) {
52+ Document documentTestClass = FileDocumentManager .getInstance ().getDocument (currentTestClass .getContainingFile ().getVirtualFile ());
53+ Document documentTestedClass = FileDocumentManager .getInstance ().getDocument (Utils .locateTestedClass (project , currentTestClass ).getContainingFile ().getVirtualFile ());
54+ FileDocumentManager .getInstance ().saveDocument (documentTestClass );
55+ FileDocumentManager .getInstance ().saveDocument (documentTestedClass );
56+
57+ }
58+
4959 public void actionPerformed (final AnActionEvent e ) {
5060 PhpClass currentTestClass = getCurrentTestClass (e );
5161 if (currentTestClass == null ) {
@@ -57,6 +67,8 @@ public void actionPerformed(final AnActionEvent e) {
5767 ToolWindow toolWindow = getToolWindow (project );
5868 ConsoleView console = getConsole (toolWindow , project );
5969
70+ saveFiles (currentTestClass , project );
71+
6072 String output = runTest (currentTestClass , project );
6173
6274 toolWindow .setIcon (Icons .ATOUM );
0 commit comments