public void ProcessScriptBlocksVBTest()
{
var pageType = new CodeTypeDeclaration("MyControl");
var snippet = new CodeSnippetTypeMember("public foo as int");
pageType.Members.Add(snippet);
var renderMethod = new CodeMemberMethod();
WebPageSurrogateControlBuilder.ProcessScriptBlocks(pageType, renderMethod, Language.VisualBasic);
var snippets = renderMethod.Statements.OfType <CodeSnippetStatement>().ToList();
Assert.AreEqual(1, snippets.Count);
var snip = snippets[0];
Assert.IsTrue(snip.Value.Contains("Public Module MyControlExtensions"));
Assert.IsTrue(snip.Value.Contains("Public Function MyControl(htmlHelper As System.Web.Mvc.HtmlHelper, optional foo as int = Nothing) As HelperResult"));
Assert.IsTrue(snip.Value.Contains("uc.foo = foo"));
}
public void ProcessScriptBlocksCSTest()
{
var pageType = new CodeTypeDeclaration("MyControl");
var snippet = new CodeSnippetTypeMember("public int foo;");
pageType.Members.Add(snippet);
var renderMethod = new CodeMemberMethod();
WebPageSurrogateControlBuilder.ProcessScriptBlocks(pageType, renderMethod, Language.CSharp);
var snippets = renderMethod.Statements.OfType <CodeSnippetStatement>().ToList();
Assert.AreEqual(1, snippets.Count);
var snip = snippets[0];
Assert.IsTrue(snip.Value.Contains("public static class MyControlExtensions"));
Assert.IsTrue(snip.Value.Contains("public static HelperResult MyControl(this System.Web.Mvc.HtmlHelper htmlHelper, int foo = default(int))"));
Assert.IsTrue(snip.Value.Contains("uc.foo = foo;"));
}