public void SetUp()
{
MockRepository mockRepository = new MockRepository();
_mockCodeGenerator = mockRepository.Stub <ICodeGenerator>();
CodeNamespace codeNamespace = new CodeNamespace(_sampleNamespace);
_nStubCore =
new NStubCore(new FakeBuildSystem(), codeNamespace, _outputDirectory, _mockCodeGenerator);
}
public void SetUp()
{
MockRepository mockRepository = new MockRepository();
_mockCodeGenerator = mockRepository.CreateMock <ICodeGenerator>();
CodeNamespace codeNamespace = new CodeNamespace(_sampleNamespace);
_nStubCore =
new NStubCore(codeNamespace, _outputDirectory, _mockCodeGenerator);
}
public void ConstructWithParametersBuildSystemCodeNamespaceOutputDirectoryCodeGeneratorTest()
{
var codeNamespace = new System.CodeDom.CodeNamespace();
var outputDirectory = "Value of outputDirectory";
var testObject = new NStubCore(new FakeBuildSystem(), codeNamespace, outputDirectory, _mockCodeGenerator);
Assert.Throws <ArgumentNullException>(() => new NStubCore(null, codeNamespace, outputDirectory, _mockCodeGenerator));
Assert.Throws <ArgumentNullException>(() => new NStubCore(new FakeBuildSystem(), null, outputDirectory, _mockCodeGenerator));
Assert.Throws <ArgumentNullException>(() => new NStubCore(new FakeBuildSystem(), codeNamespace, null, _mockCodeGenerator));
Assert.Throws <ArgumentException>(() => new NStubCore(new FakeBuildSystem(), codeNamespace, string.Empty, _mockCodeGenerator));
Assert.Throws <ArgumentNullException>(() => new NStubCore(new FakeBuildSystem(), codeNamespace, outputDirectory, null));
Assert.Throws <System.IO.DirectoryNotFoundException>(() =>
new NStubCore(new FakeBuildSystem()
{
FakeDirectoryExists = false
}, codeNamespace, outputDirectory, _mockCodeGenerator));
}
/// <summary>
/// Handles the Click event of the btnGo control. Creates a list of the methods
/// for which the user wishes to generate test cases for and instantiates an
/// instance of NStub around these methods. The sources are files are then
/// generated.
/// </summary>
/// <param name="sender">The source of the event.</param>
/// <param name="e">The <see cref="T:System.EventArgs"/> instance containing the
/// event data.</param>
private void btnGo_Click(object sender, EventArgs e)
{
Cursor.Current = Cursors.WaitCursor;
_browseInputAssemblyButton.Enabled = false;
_browseOutputDirectoryButton.Enabled = false;
// Create a new directory for each assembly
for (int h = 0; h < _assemblyGraphTreeView.Nodes.Count; h++)
{
string outputDirectory = _outputDirectoryTextBox.Text +
Path.DirectorySeparatorChar +
Path.GetFileNameWithoutExtension(_assemblyGraphTreeView.Nodes[h].Text) + "Test";
Directory.CreateDirectory(outputDirectory);
// Create our project generator
CSharpProjectGenerator cSharpProjectGenerator =
new CSharpProjectGenerator(
Path.GetFileNameWithoutExtension(_inputAssemblyTextBox.Text) + "Test",
outputDirectory);
// Add our referenced assemblies to the project generator so we
// can build the resulting project
foreach (AssemblyName assemblyName in _referencedAssemblies)
{
cSharpProjectGenerator.ReferencedAssemblies.Add(assemblyName);
}
// Generate the new test namespace
// At the assembly level
for (int i = 0; i < _assemblyGraphTreeView.Nodes[h].Nodes.Count; i++)
{
if (_assemblyGraphTreeView.Nodes[h].Nodes[i].Checked)
{
// Create the namespace and initial inputs
CodeNamespace codeNamespace = new CodeNamespace();
// At the type level
for (int j = 0; j < _assemblyGraphTreeView.Nodes[h].Nodes[i].Nodes.Count; j++)
{
//TODO: This namespace isn't being set correctly.
// Also one namespace per run probably won't work, we may
// need to break this up more.
codeNamespace.Name =
Utility.GetNamespaceFromFullyQualifiedTypeName(
_assemblyGraphTreeView.Nodes[h].Nodes[i].Nodes[j].Text);
if (_assemblyGraphTreeView.Nodes[h].Nodes[i].Nodes[j].Checked)
{
// Create the class
CodeTypeDeclaration testClass =
new CodeTypeDeclaration(_assemblyGraphTreeView.Nodes[h].Nodes[i].Nodes[j].Text);
codeNamespace.Types.Add(testClass);
// At the method level
// Create a test method for each method in this type
for (int k = 0; k < _assemblyGraphTreeView.Nodes[h].Nodes[i].Nodes[j].Nodes.Count; k++)
{
try
{
if (_assemblyGraphTreeView.Nodes[h].Nodes[i].Nodes[j].Nodes[k].Checked)
{
// Retrieve the MethodInfo object from this TreeNode's tag
CodeMemberMethod codeMemberMethod =
CreateMethod(
_assemblyGraphTreeView.Nodes[h].Nodes[i].Nodes[j].Nodes[k].Text,
(MethodInfo)_assemblyGraphTreeView.Nodes[h].Nodes[i].Nodes[j].Nodes[k].Tag);
testClass.Members.Add(codeMemberMethod);
}
}
catch (Exception)
{
return;
}
}
}
}
// Now write the test file
NStubCore nStub =
new NStubCore(codeNamespace, outputDirectory,
new CSharpCodeGenerator(codeNamespace, outputDirectory));
nStub.GenerateCode();
// Add all of our classes to the project
foreach (CodeTypeDeclaration codeType in nStub.CodeNamespace.Types)
{
string fileName = codeType.Name;
fileName = fileName.Remove(0, (fileName.LastIndexOf(".") + 1));
fileName += ".cs";
cSharpProjectGenerator.ClassFiles.Add(fileName);
}
}
}
// Now write the project file and add all of the test files to it
cSharpProjectGenerator.GenerateProjectFile();
}
_browseInputAssemblyButton.Enabled = true;
_browseOutputDirectoryButton.Enabled = true;
Cursor.Current = Cursors.Arrow;
}
