Skip to content

Commit 7d0b34d

Browse files
committed
Added MockProvider to test template
1 parent 45e69f5 commit 7d0b34d

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

Rubberduck.UnitTesting/CodeGeneration/TestCodeGeneratorStatics.cs

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -81,11 +81,13 @@ End If
8181

8282
private static string LateBindingDeclarations =>
8383
@" Private Assert As Object
84-
Private Fakes As Object";
84+
Private Fakes As Object
85+
Private Mocks As Object";
8586

8687
private static string EarlyBindingDeclarations =>
8788
@" Private Assert As Rubberduck.{0}
88-
Private Fakes As Rubberduck.FakesProvider";
89+
Private Fakes As Rubberduck.FakesProvider
90+
Private Mocks As Rubberduck.MockProvider";
8991

9092
private static string DualBindingDeclarations =>
9193
$@"#Const {LateBindConstName} = {LateBindDirectiveName}
@@ -98,11 +100,13 @@ End If
98100

99101
private static string LateBindingInitialization =>
100102
@" Set Assert = CreateObject(""Rubberduck.{0}"")
101-
Set Fakes = CreateObject(""Rubberduck.FakesProvider"")";
103+
Set Fakes = CreateObject(""Rubberduck.FakesProvider"")
104+
Set Mocks = CreateObject(""Rubberduck.MockProvider"")";
102105

103106
private static string EarlyBindingInitialization =>
104107
@" Set Assert = New Rubberduck.{0}
105-
Set Fakes = New Rubberduck.FakesProvider";
108+
Set Fakes = New Rubberduck.FakesProvider
109+
Set Mocks = New Rubberduck.MockProvider";
106110

107111
private static string DualBindingInitialization =>
108112
$@"#If {LateBindConstName} Then
@@ -131,6 +135,7 @@ End Sub
131135
{ModuleCleanupComment}
132136
Set Assert = Nothing
133137
Set Fakes = Nothing
138+
Set Mocks = Nothing
134139
End Sub
135140
136141
'@TestInitialize

0 commit comments

Comments
 (0)