public static ModelBuilderWizardForm CreateWizard(ModelGenerationOption generationOption = (ModelGenerationOption)(-1),
Project project = null, string modelPath = null, IServiceProvider serviceProvider = null)
{
var modelBuilderSettings =
new ModelBuilderSettings
{
Project = project ?? MockDTE.CreateProject(),
GenerationOption = generationOption,
ModelPath = modelPath
};
return CreateWizard(modelBuilderSettings, serviceProvider);
}
public static ModelBuilderWizardForm CreateWizard(ModelGenerationOption generationOption = (ModelGenerationOption)(-1),
Project project = null, string modelPath = null, IServiceProvider serviceProvider = null)
{
var modelBuilderSettings =
new ModelBuilderSettings
{
Project = project ?? MockDTE.CreateProject(),
GenerationOption = generationOption,
ModelPath = modelPath
};
return(CreateWizard(modelBuilderSettings, serviceProvider));
}
private static ModelBuilderWizardForm CreateModelBuilderWizardForm(ModelGenerationOption modelGenerationOption)
{
var mockDte = new MockDTE(".NETFramework, Version=v4.5", references: new Reference[0]);
var modelBuilderSettings = new ModelBuilderSettings
{
NewItemFolder = @"C:\temp",
ModelName = "myModel",
ReplacementDictionary = new Dictionary <string, string>(),
TargetSchemaVersion = EntityFrameworkVersion.Version3,
VsTemplatePath = "fake.vstemplate",
Project = mockDte.Project,
GenerationOption = modelGenerationOption
};
var wizard = ModelBuilderWizardFormHelper.CreateWizard(modelBuilderSettings, mockDte.ServiceProvider);
return(wizard);
}
private static void Run_OnDeactivate_creates_and_verifies_model_path(
ModelGenerationOption generationOption, LangEnum language, bool isWebSite, string expectedExtension)
{
var mockDte =
new MockDTE(
".NETFramework, Version=v4.5",
isWebSite
? MockDTE.CreateWebSite(
properties: new Dictionary <string, object>
{
{ "CurrentWebsiteLanguage", language == LangEnum.CSharp ? "C#" : "VB" }
},
assemblyReferences: new AssemblyReference[0])
: MockDTE.CreateProject(
kind: language == LangEnum.CSharp ? MockDTE.CSharpProjectKind : MockDTE.VBProjectKind,
assemblyReferences: new Reference[0]));
var modelBuilderSettings = new ModelBuilderSettings
{
NewItemFolder = @"C:\temp",
ModelName = "myModel",
Project = mockDte.Project,
GenerationOption = generationOption
};
var mockWizardPageStart =
new Mock <WizardPageStart>(ModelBuilderWizardFormHelper.CreateWizard(modelBuilderSettings, mockDte.ServiceProvider), null)
{
CallBase = true
};
mockWizardPageStart
.Protected()
.Setup <bool>("VerifyModelFilePath", ItExpr.IsAny <string>())
.Returns(false);
mockWizardPageStart.Object.OnDeactivate();
mockWizardPageStart
.Protected()
.Verify("VerifyModelFilePath", Times.Once(), @"C:\temp\myModel" + expectedExtension);
}
private static void Run_OnDeactivate_creates_and_verifies_model_path(
ModelGenerationOption generationOption, LangEnum language, bool isWebSite, string expectedExtension)
{
var mockDte =
new MockDTE(
".NETFramework, Version=v4.5",
isWebSite
? MockDTE.CreateWebSite(
properties: new Dictionary<string, object>
{
{ "CurrentWebsiteLanguage", language == LangEnum.CSharp ? "C#" : "VB" }
},
assemblyReferences: new AssemblyReference[0])
: MockDTE.CreateProject(
kind: language == LangEnum.CSharp ? MockDTE.CSharpProjectKind : MockDTE.VBProjectKind,
assemblyReferences: new Reference[0]));
var modelBuilderSettings = new ModelBuilderSettings
{
NewItemFolder = @"C:\temp",
ModelName = "myModel",
Project = mockDte.Project,
GenerationOption = generationOption
};
var mockWizardPageStart =
new Mock<WizardPageStart>(ModelBuilderWizardFormHelper.CreateWizard(modelBuilderSettings, mockDte.ServiceProvider), null)
{ CallBase = true };
mockWizardPageStart
.Protected()
.Setup<bool>("VerifyModelFilePath", ItExpr.IsAny<string>())
.Returns(false);
mockWizardPageStart.Object.OnDeactivate();
mockWizardPageStart
.Protected()
.Verify("VerifyModelFilePath", Times.Once(), @"C:\temp\myModel" + expectedExtension);
}
protected void bgWorkerPopulateTree_RunWorkerCompleted(object sender, RunWorkerCompletedEventArgs args)
{
try
{
_stopwatch.Stop();
if (args.Cancelled)
{
return;
}
if (args.Error != null)
{
// show error dialog
ModelBuilderWizardForm.ShowDatabaseConnectionErrorDialog(args.Error);
Visited = false;
Wizard.OnPrevious();
}
else
{
using (new VsUtils.HourglassHelper())
{
// No errors, populate nodes in TreeView - create the nodes accordingly
var result = (ICollection <EntityStoreSchemaFilterEntry>[])args.Result;
var tableEntries = result[0];
var viewEntries = result[1];
var sprocEntries = result[2];
CreateTreeForNewModel(tableEntries, viewEntries, sprocEntries);
// if there are Sproc entries then enable the Create Function Imports box and by default set to checked
if (sprocEntries != null && sprocEntries.Count > 0)
{
chkCreateFunctionImports.Enabled = true;
chkCreateFunctionImports.Checked = true;
}
// Hide status message
databaseObjectTreeView.HideStatus();
// Set focus to TreeView
databaseObjectTreeView.TreeViewControl.SelectedNode = databaseObjectTreeView.TreeViewControl.Nodes[0];
databaseObjectTreeView.Focus();
_initializedDataConnection = Wizard.ModelBuilderSettings.DesignTimeConnectionString;
_initializedUsingLegacyProvider = Wizard.ModelBuilderSettings.UseLegacyProvider;
_initializedGenerationOption = Wizard.ModelBuilderSettings.GenerationOption;
modelNamespaceTextBox.Enabled = Wizard.ModelBuilderSettings.GenerationOption != ModelGenerationOption.CodeFirstFromDatabase;
}
}
// Enable wizard navigation
Wizard.OnValidationStateChanged(this);
}
catch (Exception e)
{
databaseObjectTreeView.ShowStatus(
String.Format(
CultureInfo.CurrentCulture,
ModelWizard.Properties.Resources.SelectTablesPage_ErrorRetrievingTablesText, e.Message));
Wizard.EnableButton(ButtonType.Cancel, true);
}
}
protected void bgWorkerPopulateTree_RunWorkerCompleted(object sender, RunWorkerCompletedEventArgs args)
{
try
{
_stopwatch.Stop();
if (args.Cancelled)
{
return;
}
if (args.Error != null)
{
// show error dialog
ModelBuilderWizardForm.ShowDatabaseConnectionErrorDialog(args.Error);
Visited = false;
Wizard.OnPrevious();
}
else
{
using (new VsUtils.HourglassHelper())
{
// No errors, populate nodes in TreeView - create the nodes accordingly
var result = (ICollection<EntityStoreSchemaFilterEntry>[])args.Result;
var tableEntries = result[0];
var viewEntries = result[1];
var sprocEntries = result[2];
CreateTreeForNewModel(tableEntries, viewEntries, sprocEntries);
// if there are Sproc entries then enable the Create Function Imports box and by default set to checked
if (sprocEntries != null && sprocEntries.Count > 0)
{
chkCreateFunctionImports.Enabled = true;
chkCreateFunctionImports.Checked = true;
}
// Hide status message
databaseObjectTreeView.HideStatus();
// Set focus to TreeView
databaseObjectTreeView.TreeViewControl.SelectedNode = databaseObjectTreeView.TreeViewControl.Nodes[0];
databaseObjectTreeView.Focus();
_initializedDataConnection = Wizard.ModelBuilderSettings.DesignTimeConnectionString;
_initializedUsingLegacyProvider = Wizard.ModelBuilderSettings.UseLegacyProvider;
_initializedGenerationOption = Wizard.ModelBuilderSettings.GenerationOption;
modelNamespaceTextBox.Enabled = Wizard.ModelBuilderSettings.GenerationOption != ModelGenerationOption.CodeFirstFromDatabase;
}
}
// Enable wizard navigation
Wizard.OnValidationStateChanged(this);
}
catch (Exception e)
{
databaseObjectTreeView.ShowStatus(
String.Format(
CultureInfo.CurrentCulture,
ModelWizard.Properties.Resources.SelectTablesPage_ErrorRetrievingTablesText, e.Message));
Wizard.EnableButton(ButtonType.Cancel, true);
}
}
private static ModelBuilderWizardForm CreateModelBuilderWizardForm(ModelGenerationOption modelGenerationOption)
{
var mockDte = new MockDTE(".NETFramework, Version=v4.5", references: new Reference[0]);
var modelBuilderSettings = new ModelBuilderSettings
{
NewItemFolder = @"C:\temp",
ModelName = "myModel",
ReplacementDictionary = new Dictionary<string, string>(),
TargetSchemaVersion = EntityFrameworkVersion.Version3,
VsTemplatePath = "fake.vstemplate",
Project = mockDte.Project,
GenerationOption = modelGenerationOption
};
var wizard = ModelBuilderWizardFormHelper.CreateWizard(modelBuilderSettings, mockDte.ServiceProvider);
return wizard;
}
