Skip to content

Commit eb7132b

Browse files
committed
fixes #2794
1 parent 69b91ac commit eb7132b

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

RetailCoder.VBE/UI/Command/AddTestModuleCommand.cs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,10 +29,11 @@ public AddTestModuleCommand(IVBE vbe, RubberduckParserState state, IGeneralConfi
2929
_configLoader = configLoader;
3030
}
3131

32+
private const string FolderAnnotation = "'@Folder(\"Tests\")\r\n";
3233
private const string ModuleLateBinding = "Private Assert As Object\r\n";
3334
private const string ModuleEarlyBinding = "Private Assert As New Rubberduck.{0}AssertClass\r\n";
3435

35-
private const string TestModuleEmptyTemplate = "'@TestModule\r\n{0}\r\n'@Folder(\"Tests\")\r\n";
36+
private const string TestModuleEmptyTemplate = "'@TestModule\r\n{0}\r\n{1}\r\n";
3637

3738
private const string ModuleInitLateBinding = "Set Assert = CreateObject(\"Rubberduck.{0}AssertClass\")\r\n";
3839
private readonly string _moduleInit = string.Concat(
@@ -66,7 +67,7 @@ private string GetTestModule(IUnitTestSettings settings)
6667
? string.Format(ModuleEarlyBinding, assertClass)
6768
: ModuleLateBinding;
6869

69-
var formattedModuleTemplate = string.Format(TestModuleEmptyTemplate, moduleBinding);
70+
var formattedModuleTemplate = string.Format(TestModuleEmptyTemplate, FolderAnnotation, moduleBinding);
7071

7172
if (settings.ModuleInit)
7273
{

0 commit comments

Comments
 (0)