Skip to content

Commit dea0f60

Browse files
committed
Merge pull request #1092 from autoboosh/fixtests
fix tests
2 parents edd43ce + 9595640 commit dea0f60

File tree

2 files changed

+7
-4
lines changed

2 files changed

+7
-4
lines changed

RetailCoder.VBE/Inspections/ImplicitPublicMemberInspectionResult.cs

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ public ImplicitPublicMemberInspectionResult(IInspection inspection, QualifiedCon
1717
{
1818
_quickFixes = new CodeInspectionQuickFix[]
1919
{
20-
new SpecifyExplicitPublicModifierQuickFix(Context, QualifiedSelection),
20+
new SpecifyExplicitPublicModifierQuickFix(qualifiedContext.Context, QualifiedSelection),
2121
new IgnoreOnceQuickFix(qualifiedContext.Context, QualifiedSelection, Inspection.AnnotationName),
2222
};
2323
}
@@ -45,14 +45,13 @@ public override void Fix()
4545
var oldContent = Context.GetText();
4646
var newContent = Tokens.Public + ' ' + oldContent;
4747

48-
var selection = Selection.Selection;
48+
var selection = Context.GetSelection();
4949

5050
var module = Selection.QualifiedName.Component.CodeModule;
5151
var lines = module.get_Lines(selection.StartLine, selection.LineCount);
5252

53-
var result = lines.Replace(oldContent, newContent);
5453
module.DeleteLines(selection.StartLine, selection.LineCount);
55-
module.InsertLines(selection.StartLine, result);
54+
module.InsertLines(selection.StartLine, newContent);
5655
}
5756
}
5857
}

RubberduckTests/Inspections/ImplicitPublicMemberInspectionTests.cs

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -145,10 +145,14 @@ public void ImplicitPublicMember_QuickFixWorks()
145145
{
146146
const string inputCode =
147147
@"Sub Foo(ByVal arg1 as Integer)
148+
149+
148150
End Sub";
149151

150152
const string expectedCode =
151153
@"Public Sub Foo(ByVal arg1 as Integer)
154+
155+
152156
End Sub";
153157

154158
//Arrange

0 commit comments

Comments
 (0)