@@ -55,7 +55,8 @@ public TestExplorerViewModel(ISelectionService selectionService,
55
55
// ReSharper disable once UnusedParameter.Local - left in place because it will likely be needed for app wide font settings, etc.
56
56
IConfigurationService < Configuration > configService ,
57
57
ISettingsFormFactory settingsFormFactory ,
58
- IRewritingManager rewritingManager )
58
+ IRewritingManager rewritingManager ,
59
+ IAnnotationUpdater annotationUpdater )
59
60
{
60
61
_clipboard = clipboard ;
61
62
_settingsFormFactory = settingsFormFactory ;
@@ -74,6 +75,7 @@ public TestExplorerViewModel(ISelectionService selectionService,
74
75
UnignoreTestCommand = new DelegateCommand ( LogManager . GetCurrentClassLogger ( ) , ExecuteUnignoreTestCommand ) ;
75
76
76
77
RewritingManager = rewritingManager ;
78
+ AnnotationUpdater = annotationUpdater ;
77
79
78
80
Model = model ;
79
81
Model . TestCompleted += HandleTestCompletion ;
@@ -233,6 +235,7 @@ private void HandleTestCompletion(object sender, TestCompletedEventArgs e)
233
235
}
234
236
235
237
public IRewritingManager RewritingManager { get ; }
238
+ public IAnnotationUpdater AnnotationUpdater { get ; }
236
239
237
240
private TestMethod _mousedOverTestMethod => ( ( TestMethodViewModel ) SelectedItem ) . Method ;
238
241
public bool DisplayUnignoreTestLabel => SelectedItem != null && _mousedOverTestMethod . IsIgnored ;
@@ -371,8 +374,7 @@ private void ExecuteIgnoreTestCommand(object parameter)
371
374
{
372
375
var rewriteSession = RewritingManager . CheckOutCodePaneSession ( ) ;
373
376
374
- var annotationUpdater = new AnnotationUpdater ( ) ;
375
- annotationUpdater . AddAnnotation ( rewriteSession , _mousedOverTestMethod . Declaration , Parsing . Annotations . AnnotationType . IgnoreTest ) ;
377
+ AnnotationUpdater . AddAnnotation ( rewriteSession , _mousedOverTestMethod . Declaration , Parsing . Annotations . AnnotationType . IgnoreTest ) ;
376
378
377
379
rewriteSession . TryRewrite ( ) ;
378
380
}
@@ -383,13 +385,10 @@ private void ExecuteUnignoreTestCommand(object parameter)
383
385
var ignoreTestAnnotations = _mousedOverTestMethod . Declaration . Annotations
384
386
. Where ( iannotations => iannotations . AnnotationType == Parsing . Annotations . AnnotationType . IgnoreTest ) ;
385
387
386
- var annotationUpdater = new AnnotationUpdater ( ) ;
387
-
388
388
foreach ( var ignoreTestAnnotation in ignoreTestAnnotations )
389
389
{
390
- annotationUpdater . RemoveAnnotation ( rewriteSession , ignoreTestAnnotation ) ;
390
+ AnnotationUpdater . RemoveAnnotation ( rewriteSession , ignoreTestAnnotation ) ;
391
391
}
392
-
393
392
394
393
rewriteSession . TryRewrite ( ) ;
395
394
}
0 commit comments