@@ -47,7 +47,7 @@ public void RazorPageScaffolder ()
4747 var scaffolder = new RazorPageScaffolder ( args ) ;
4848 args . Scaffolder = scaffolder ;
4949 scaffolder . GetField ( "Name of the Razor Page:" ) . SelectedValue = "PageName" ;
50- var wizard = new ScaffolderWizard ( project , args . ParentFolder ) ;
50+ var wizard = CreateWizard ( args , project ) ;
5151 var commandLineArgs = wizard . GetArguments ( args ) ;
5252 commandLineArgs = Regex . Replace ( commandLineArgs , @"\s+" , " " ) ;
5353 Assert . AreEqual ( @"aspnet-codegenerator --project ""ProjectName.csproj"" razorpage PageName Empty --referenceScriptLibraries --no-build -outDir ""/MyProject/Pages"" --namespaceName ProjectName" , commandLineArgs ) ;
@@ -64,7 +64,7 @@ public void RazorPageScaffolderWithoutLayoutPage ()
6464 args . Scaffolder = scaffolder ;
6565 scaffolder . GetField ( "Name of the Razor Page:" ) . SelectedValue = "PageName" ;
6666 ( scaffolder . GetField ( "Use a layout page" ) as BoolField ) . Selected = false ;
67- var wizard = new ScaffolderWizard ( project , args . ParentFolder ) ;
67+ var wizard = CreateWizard ( args , project ) ;
6868 var commandLineArgs = wizard . GetArguments ( args ) ;
6969 commandLineArgs = Regex . Replace ( commandLineArgs , @"\s+" , " " ) ;
7070 Assert . AreEqual ( @"aspnet-codegenerator --project ""ProjectName.csproj"" razorpage PageName Empty --referenceScriptLibraries --useDefaultLayout --no-build -outDir ""/MyProject/Pages"" --namespaceName ProjectName" , commandLineArgs ) ;
@@ -80,8 +80,8 @@ public void RazorPageScaffolderWithoutReferenceScriptLibraries ()
8080 var scaffolder = new RazorPageScaffolder ( args ) ;
8181 args . Scaffolder = scaffolder ;
8282 scaffolder . GetField ( "Name of the Razor Page:" ) . SelectedValue = "PageName" ;
83+ var wizard = CreateWizard ( args , project ) ;
8384 ( scaffolder . GetField ( "Reference script libraries" ) as BoolField ) . Selected = false ;
84- var wizard = new ScaffolderWizard ( project , args . ParentFolder ) ;
8585 var commandLineArgs = wizard . GetArguments ( args ) ;
8686 commandLineArgs = Regex . Replace ( commandLineArgs , @"\s+" , " " ) ;
8787 Assert . AreEqual ( @"aspnet-codegenerator --project ""ProjectName.csproj"" razorpage PageName Empty --no-build -outDir ""/MyProject/Pages"" --namespaceName ProjectName" , commandLineArgs ) ;
@@ -99,7 +99,7 @@ public void RazorPageEntityFrameworkScaffolder ()
9999 scaffolder . GetField ( "Name of the Razor Page:" ) . SelectedValue = "PageName" ;
100100 scaffolder . GetField ( "Model class to use:" ) . SelectedValue = "ModelClass" ;
101101 scaffolder . GetField ( "DbContext class to use:" ) . SelectedValue = "DataContext" ;
102- var wizard = new ScaffolderWizard ( project , args . ParentFolder ) ;
102+ var wizard = CreateWizard ( args , project ) ;
103103 var commandLineArgs = wizard . GetArguments ( args ) ;
104104 commandLineArgs = Regex . Replace ( commandLineArgs , @"\s+" , " " ) ;
105105 Assert . AreEqual ( @"aspnet-codegenerator --project ""ProjectName.csproj"" razorpage PageName --model ModelClass --dataContext DataContext --referenceScriptLibraries --no-build -outDir ""/MyProject/Pages"" --namespaceName ProjectName" , commandLineArgs ) ;
@@ -117,7 +117,7 @@ public void RazorPageEntityFrameworkCrudScaffolder ()
117117 // no name field
118118 scaffolder . GetField ( "Model class to use:" ) . SelectedValue = "ModelClass" ;
119119 scaffolder . GetField ( "DbContext class to use:" ) . SelectedValue = "DataContext" ;
120- var wizard = new ScaffolderWizard ( project , args . ParentFolder ) ;
120+ var wizard = CreateWizard ( args , project ) ;
121121 var commandLineArgs = wizard . GetArguments ( args ) ;
122122 commandLineArgs = Regex . Replace ( commandLineArgs , @"\s+" , " " ) ;
123123 Assert . AreEqual ( @"aspnet-codegenerator --project ""ProjectName.csproj"" razorpage --model ModelClass --dataContext DataContext --referenceScriptLibraries --no-build -outDir ""/MyProject/Pages"" --namespaceName ProjectName" , commandLineArgs ) ;
@@ -133,7 +133,7 @@ public void EmptyMvcControllerScaffolder ()
133133 var scaffolder = new EmptyMvcControllerScaffolder ( args ) ;
134134 args . Scaffolder = scaffolder ;
135135 scaffolder . GetField ( "Controller name:" ) . SelectedValue = "ControllerName" ;
136- var wizard = new ScaffolderWizard ( project , args . ParentFolder ) ;
136+ var wizard = CreateWizard ( args , project ) ;
137137 var commandLineArgs = wizard . GetArguments ( args ) ;
138138 commandLineArgs = Regex . Replace ( commandLineArgs , @"\s+" , " " ) ;
139139 Assert . AreEqual ( @"aspnet-codegenerator --project ""ProjectName.csproj"" controller -name ControllerName --no-build -outDir ""/MyProject/Controllers"" --controllerNamespace ProjectName" , commandLineArgs ) ;
@@ -148,7 +148,7 @@ public void EmptyApiControllerScaffolder ()
148148 args . Project = project ;
149149 var scaffolder = new EmptyApiControllerScaffolder ( args ) ;
150150 args . Scaffolder = scaffolder ;
151- var wizard = new ScaffolderWizard ( project , args . ParentFolder ) ;
151+ var wizard = CreateWizard ( args , project ) ;
152152 var commandLineArgs = wizard . GetArguments ( args ) ;
153153 commandLineArgs = Regex . Replace ( commandLineArgs , @"\s+" , " " ) . TrimEnd ( ) ;
154154 Assert . AreEqual ( @"aspnet-codegenerator --project ""ProjectName.csproj"" controller -name --no-build -outDir ""/MyProject/Controllers"" --controllerNamespace ProjectName --restWithNoViews" , commandLineArgs ) ;
@@ -163,7 +163,7 @@ public void MvcControllerWithActionsScaffolder ()
163163 args . Project = project ;
164164 var scaffolder = new MvcControllerWithActionsScaffolder ( args ) ;
165165 args . Scaffolder = scaffolder ;
166- var wizard = new ScaffolderWizard ( project , args . ParentFolder ) ;
166+ var wizard = CreateWizard ( args , project ) ;
167167 var commandLineArgs = wizard . GetArguments ( args ) ;
168168 commandLineArgs = Regex . Replace ( commandLineArgs , @"\s+" , " " ) . TrimEnd ( ) ;
169169 Assert . AreEqual ( @"aspnet-codegenerator --project ""ProjectName.csproj"" controller -name --no-build -outDir ""/MyProject/Controllers"" --controllerNamespace ProjectName --readWriteActions" , commandLineArgs ) ;
@@ -180,7 +180,7 @@ public void ApiControllerEntityFrameworkScaffolder ()
180180 args . Scaffolder = scaffolder ;
181181 scaffolder . GetField ( "Model class to use:" ) . SelectedValue = "ModelClass" ;
182182 scaffolder . GetField ( "DbContext class to use:" ) . SelectedValue = "DataContext" ;
183- var wizard = new ScaffolderWizard ( project , args . ParentFolder ) ;
183+ var wizard = CreateWizard ( args , project ) ;
184184 var commandLineArgs = wizard . GetArguments ( args ) ;
185185 commandLineArgs = Regex . Replace ( commandLineArgs , @"\s+" , " " ) . TrimEnd ( ) ;
186186 Assert . AreEqual ( @"aspnet-codegenerator --project ""ProjectName.csproj"" controller --model ModelClass --dataContext DataContext -name --no-build -outDir ""/MyProject/Controllers"" --controllerNamespace ProjectName" , commandLineArgs ) ;
@@ -195,7 +195,7 @@ public void ApiControllerWithActionsScaffolder ()
195195 args . Project = project ;
196196 var scaffolder = new ApiControllerWithActionsScaffolder ( args ) ;
197197 args . Scaffolder = scaffolder ;
198- var wizard = new ScaffolderWizard ( project , args . ParentFolder ) ;
198+ var wizard = CreateWizard ( args , project ) ;
199199 var commandLineArgs = wizard . GetArguments ( args ) ;
200200 commandLineArgs = Regex . Replace ( commandLineArgs , @"\s+" , " " ) . TrimEnd ( ) ;
201201 Assert . AreEqual ( @"aspnet-codegenerator --project ""ProjectName.csproj"" controller -name --no-build -outDir ""/MyProject/Controllers"" --controllerNamespace ProjectName --restWithNoViews --readWriteActions" , commandLineArgs ) ;
@@ -222,6 +222,12 @@ DotNetProject CreateProject ()
222222
223223 return ( DotNetProject ) Services . ProjectService . CreateProject ( "C#" , info , projectOptions ) ;
224224 }
225+
226+ static ScaffolderWizard CreateWizard ( ScaffolderArgs args , DotNetProject project )
227+ {
228+ var selectPage = new ScaffolderTemplateSelectPage ( args ) ;
229+ return new ScaffolderWizard ( project , args . ParentFolder , selectPage , args ) ;
230+ }
225231 }
226232
227233 static class ScaffolderFieldExtension
@@ -237,4 +243,6 @@ public static ScaffolderField GetField (this ScaffolderBase scaffolder, string d
237243 return optionList . Options . FirstOrDefault ( f => f . DisplayName == displayName ) ;
238244 }
239245 }
246+
247+
240248}
0 commit comments