/// <summary>
/// Implements the mock code.
/// </summary>
/// <param name="instance">The instance.</param>
/// <param name="codeSnippet">The code snippet.</param>
public static void ImplementMockCode(
this CodeClass instance,
CodeSnippet codeSnippet)
{
TraceService.WriteLine("CodeClassExtensions::ImplementMockCode file=" + instance.Name);
CodeFunction codeFunction = instance.GetFunction(codeSnippet.TestInitMethod);
if (codeFunction != null)
{
if (string.IsNullOrEmpty(codeSnippet.GetMockInitCode()) == false)
{
codeFunction.InsertCode(codeSnippet.GetMockInitCode(), true);
}
if (string.IsNullOrEmpty(codeSnippet.GetMockConstructorCode()) == false)
{
string code = codeFunction.GetCode();
string testFileName = instance.Name;
//// remove Test at the start - need a better way of doing this!
string fileName = testFileName.Replace("Test", string.Empty);
//// TODO : this wont always work if the function spans multiple lines!
int index = code.IndexOf(fileName + "(", StringComparison.Ordinal);
if (index != 0)
{
string seperator = string.Empty;
//// here we are looking for the closing bracket
//// if we dont have a closing bracket then we already have something
//// on the constructor and therefore need a ',' to make it work!
if (code.Substring(index + fileName.Length + 1, 1) != ")")
{
//// TODO : do we want to create a new line too!
seperator = ", ";
}
StringBuilder sb = new StringBuilder();
sb.Append(code.Substring(0, index + fileName.Length + 1));
sb.Append(codeSnippet.GetMockConstructorCode() + seperator);
sb.Append(code.Substring(index + fileName.Length + 1));
codeFunction.ReplaceCode(sb.ToString());
}
}
}
}
/// <summary>
/// Implements the mock code.
/// </summary>
/// <param name="instance">The instance.</param>
/// <param name="codeSnippet">The code snippet.</param>
public static void ImplementMockCode(
this CodeClass instance,
CodeSnippet codeSnippet)
{
TraceService.WriteLine("CodeClassExtensions::ImplementMockCode file=" + instance.Name);
CodeFunction codeFunction = instance.GetFunction(codeSnippet.TestInitMethod);
if (codeFunction != null)
{
if (string.IsNullOrEmpty(codeSnippet.GetMockInitCode()) == false)
{
codeFunction.InsertCode(codeSnippet.GetMockInitCode(), true);
}
if (string.IsNullOrEmpty(codeSnippet.GetMockConstructorCode()) == false)
{
string code = codeFunction.GetCode();
string testFileName = instance.Name;
//// remove Test at the start - need a better way of doing this!
string fileName = testFileName.Replace("Test", string.Empty);
//// TODO : this wont always work if the function spans multiple lines!
int index = code.IndexOf(fileName + "(", StringComparison.Ordinal);
if (index != 0)
{
string seperator = string.Empty;
//// here we are looking for the closing bracket
//// if we dont have a closing bracket then we already have something
//// on the constructor and therefore need a ',' to make it work!
if (code.Substring(index + fileName.Length + 1, 1) != ")")
{
//// TODO : do we want to create a new line too!
seperator = ", ";
}
StringBuilder sb = new StringBuilder();
sb.Append(code.Substring(0, index + fileName.Length + 1));
sb.Append(codeSnippet.GetMockConstructorCode() + seperator);
sb.Append(code.Substring(index + fileName.Length + 1));
codeFunction.ReplaceCode(sb.ToString());
}
}
}
}
/// <summary>
/// Implements the code snippet.
/// </summary>
/// <param name="instance">The instance.</param>
/// <param name="codeSnippet">The code snippet.</param>
/// <param name="formatFunctionParameters">if set to <c>true</c> [format function parameters].</param>
public static void ImplementCodeSnippet(
this CodeClass instance,
CodeSnippet codeSnippet,
bool formatFunctionParameters)
{
TraceService.WriteLine("CodeClassExtensions::ImplementCodeSnippet file" + instance.Name);
if (codeSnippet.Variables != null)
{
foreach (string[] parts in codeSnippet.Variables
.Select(variable => variable.Split(' ')))
{
//// variable could already exist!
try
{
instance.ImplementVariable(parts[1], parts[0], false);
}
catch (Exception exception)
{
TraceService.WriteError("Error adding variable exception=" + exception.Message + " variable=" + parts[1]);
//// if variable already exists get out - code snippet will already have been applied.
return;
}
}
}
if (codeSnippet.MockVariables != null)
{
foreach (string[] parts in codeSnippet.MockVariables
.Select(variable => variable.Split(' ')))
{
//// variable could already exist!
try
{
instance.ImplementMockVariable(parts[1], parts[0]);
}
catch (Exception exception)
{
TraceService.WriteError("Error adding mock variable exception=" + exception.Message + " variable=" + parts[1]);
//// if variable already exists get out - code snippet will already have been applied.
return;
}
}
}
if (string.IsNullOrEmpty(codeSnippet.Code) == false)
{
instance.ImplementFunction(codeSnippet);
}
if (codeSnippet.Interfaces != null &&
codeSnippet.Interfaces.Count > 0)
{
IEnumerable<CodeFunction> constructors = instance.GetConstructors();
CodeFunction constructor = constructors.FirstOrDefault() ?? instance.AddDefaultConstructor(true);
foreach (string variable in codeSnippet.Interfaces)
{
instance.ImplementInterface(constructor, variable);
}
if (formatFunctionParameters)
{
constructor.FormatParameters();
}
}
if (string.IsNullOrEmpty(codeSnippet.GetMockInitCode()) == false ||
string.IsNullOrEmpty(codeSnippet.GetMockConstructorCode()) == false)
{
instance.ImplementMockCode(codeSnippet);
}
}
/// <summary>
/// Implements the code snippet.
/// </summary>
/// <param name="instance">The instance.</param>
/// <param name="codeSnippet">The code snippet.</param>
/// <param name="formatFunctionParameters">if set to <c>true</c> [format function parameters].</param>
public static void ImplementCodeSnippet(
this CodeClass instance,
CodeSnippet codeSnippet,
bool formatFunctionParameters)
{
TraceService.WriteLine("CodeClassExtensions::ImplementCodeSnippet file" + instance.Name);
if (codeSnippet.Variables != null)
{
foreach (string[] parts in codeSnippet.Variables
.Select(variable => variable.Split(' ')))
{
//// variable could already exist!
try
{
instance.ImplementVariable(parts[1], parts[0], false);
}
catch (Exception exception)
{
TraceService.WriteError("Error adding variable exception=" + exception.Message + " variable=" + parts[1]);
//// if variable already exists get out - code snippet will already have been applied.
return;
}
}
}
if (codeSnippet.MockVariables != null)
{
foreach (string[] parts in codeSnippet.MockVariables
.Select(variable => variable.Split(' ')))
{
//// variable could already exist!
try
{
instance.ImplementMockVariable(parts[1], parts[0], codeSnippet.MockingVariableDeclaration);
}
catch (Exception exception)
{
TraceService.WriteError("Error adding mock variable exception=" + exception.Message + " variable=" + parts[1]);
//// if variable already exists get out - code snippet will already have been applied.
return;
}
}
}
if (string.IsNullOrEmpty(codeSnippet.Code) == false)
{
instance.ImplementFunction(codeSnippet);
}
if (codeSnippet.Interfaces != null &&
codeSnippet.Interfaces.Count > 0)
{
IEnumerable <CodeFunction> constructors = instance.GetConstructors();
CodeFunction constructor = constructors.FirstOrDefault() ?? instance.AddDefaultConstructor(true);
foreach (string variable in codeSnippet.Interfaces)
{
instance.ImplementInterface(constructor, variable);
}
if (formatFunctionParameters)
{
constructor.FormatParameters();
}
}
if (string.IsNullOrEmpty(codeSnippet.GetMockInitCode()) == false ||
string.IsNullOrEmpty(codeSnippet.GetMockConstructorCode()) == false)
{
instance.ImplementMockCode(codeSnippet);
}
}
