123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500 |
- using System;
- using System.Collections.Generic;
- using System.Linq;
- using System.Text;
- using System.Reflection;
- using System.Linq.Expressions;
- using System.IO;
- using System.Diagnostics;
- using System.Text.RegularExpressions;
- namespace PetaTest
- {
- // Use to mark a class as a testfixture
- [AttributeUsage(AttributeTargets.Class, AllowMultiple = true)]
- public class TestFixtureAttribute : TestBaseAttribute
- {
- public TestFixtureAttribute(params object[] args) : base(args) { }
- }
- // Use to mark a method as a test
- [AttributeUsage(AttributeTargets.Method, AllowMultiple = true)]
- public class TestAttribute : TestBaseAttribute
- {
- public TestAttribute(params object[] args) : base(args) { }
- }
- // Base class for Test and TestFixture attributes
- public abstract class TestBaseAttribute : Attribute
- {
- public TestBaseAttribute(params object[] Arguments)
- {
- this.Arguments = Arguments;
- }
- public object[] Arguments { get; private set; }
- public string Source { get; set; }
- public bool Active { get; set; }
- public virtual IEnumerable<object[]> GetArguments(Type owningType, object testFixtureInstance)
- {
- if (Source != null)
- {
- if (testFixtureInstance != null)
- {
- var iter_method = owningType.GetMethod(Source, BindingFlags.Instance | BindingFlags.Public);
- return (IEnumerable<object[]>)iter_method.Invoke(testFixtureInstance, null);
- }
- else
- {
- var iter_method = owningType.GetMethod(Source, BindingFlags.Static | BindingFlags.Public);
- return (IEnumerable<object[]>)iter_method.Invoke(null, null);
- }
- }
- else
- {
- return new object[][] { Arguments };
- }
- }
- }
- [AttributeUsage(AttributeTargets.Method)]
- public class SetUpAttribute : SetUpTearDownAttributeBase
- {
- public SetUpAttribute() : base(true, false) { }
- }
- [AttributeUsage(AttributeTargets.Method)]
- public class TearDownAttribute : SetUpTearDownAttributeBase
- {
- public TearDownAttribute() : base(false, false) { }
- }
- [AttributeUsage(AttributeTargets.Method)]
- public class TestFixtureSetUpAttribute : SetUpTearDownAttributeBase
- {
- public TestFixtureSetUpAttribute() : base(true, true) { }
- }
- [AttributeUsage(AttributeTargets.Method)]
- public class TestFixtureTearDownAttribute : SetUpTearDownAttributeBase
- {
- public TestFixtureTearDownAttribute() : base(false, true) { }
- }
- // Introducing "The Asserts" - all pretty self explanatory
- [SkipInStackTrace]
- public static partial class Assert
- {
- public static void Throw(bool Condition, Func<string> message)
- {
- if (!Condition)
- throw new AssertionException(message());
- }
- public static void IsTrue(bool test)
- {
- Throw(test, () => "Expression is not true");
- }
- public static void IsFalse(bool test)
- {
- Throw(!test, () => "Expression is not false");
- }
- public static void AreSame(object a, object b)
- {
- Throw(object.ReferenceEquals(a, b), () => "Object references are not the same");
- }
- public static void AreNotSame(object a, object b)
- {
- Throw(!object.ReferenceEquals(a, b), () => "Object references are the same");
- }
- private static bool TestEqual(object a, object b)
- {
- if (a == null && b == null)
- return true;
- if (a == null || b == null)
- return false;
- return Object.Equals(a, b);
- }
- private static void AreEqual(object a, object b, Func<bool> Compare)
- {
- Throw(Compare(), () => string.Format("Objects are not equal\n lhs: {0}\n rhs: {1}", Utils.FormatValue(a), Utils.FormatValue(b)));
- }
- private static void AreNotEqual(object a, object b, Func<bool> Compare)
- {
- Throw(!Compare(), () => string.Format("Objects are not equal\n lhs: {0}\n rhs: {1}", Utils.FormatValue(a), Utils.FormatValue(b)));
- }
- public static void AreEqual(object a, object b)
- {
- AreEqual(a, b, () => TestEqual(a, b));
- }
- public static void AreNotEqual(object a, object b)
- {
- AreNotEqual(a, b, () => TestEqual(a, b));
- }
- public static void AreEqual(double a, double b, double within)
- {
- AreEqual(a, b, () => Math.Abs(a - b) < within);
- }
- public static void AreNotEqual(double a, double b, double within)
- {
- AreNotEqual(a, b, () => Math.Abs(a - b) < within);
- }
- public static void AreEqual<T>(T a, T b)
- {
- AreEqual(a, b, () => Object.Equals(a, b));
- }
- public static void AreNotEqual<T>(T a, T b)
- {
- AreNotEqual(a, b, () => Object.Equals(a, b));
- }
- public static void AreEqual(string a, string b, bool ignoreCase = false)
- {
- Throw(string.Compare(a, b, ignoreCase) == 0, () =>
- {
- var offset = Utils.CountCommonPrefix(a, b, ignoreCase);
- var xa = Utils.FormatValue(Utils.GetStringExtract(a, offset));
- var xb = Utils.FormatValue(Utils.GetStringExtract(b, offset));
- return string.Format("Strings are not equal at offset {0}\n lhs: {1}\n rhs: {2}\n{3}^", offset, xa, xb, new string(' ', Utils.CountCommonPrefix(xa, xb, ignoreCase) + 7));
- });
- }
- public static void AreEqual(string a, string b)
- {
- AreEqual(a, b, false);
- }
- public static void AreNotEqual(string a, string b, bool ignoreCase = false)
- {
- Throw(string.Compare(a, b, ignoreCase) != 0, () => string.Format("Strings are not equal\n lhs: {0}\n rhs: {1}", Utils.FormatValue(a), Utils.FormatValue(b)));
- }
- public static void IsEmpty(string val)
- {
- Throw(val != null && val.Length == 0, () => string.Format("String is not empty: {0}", Utils.FormatValue(val)));
- }
- public static void IsNotEmpty(string val)
- {
- Throw(val != null && val.Length != 0, () => "String is empty");
- }
- public static void IsNullOrEmpty(string val)
- {
- Throw(string.IsNullOrEmpty(val), () => string.Format("String is not empty: {0}", Utils.FormatValue(val)));
- }
- public static void IsNotNullOrEmpty(string val)
- {
- Throw(!string.IsNullOrEmpty(val), () => string.Format("String is not empty: {0}", Utils.FormatValue(val)));
- }
- public static void IsEmpty(System.Collections.IEnumerable collection)
- {
- Throw(collection != null && collection.Cast<object>().Count() == 0, () => string.Format("Collection is not empty\n Items: {0}", Utils.FormatValue(collection)));
- }
- public static void IsNotEmpty(System.Collections.IEnumerable collection)
- {
- Throw(collection != null && collection.Cast<object>().Count() != 0, () => "Collection is empty");
- }
- public static void Contains(System.Collections.IEnumerable collection, object item)
- {
- Throw(collection.Cast<object>().Contains(item), () => string.Format("Collection doesn't contain {0}\n Items: {1}", Utils.FormatValue(item), Utils.FormatValue(collection)));
- }
- public static void DoesNotContain(System.Collections.IEnumerable collection, object item)
- {
- Throw(!collection.Cast<object>().Contains(item), () => string.Format("Collection does contain {0}", Utils.FormatValue(item)));
- }
- public static void Contains(string str, string contains, bool ignoreCase)
- {
- Throw(str.IndexOf(contains, ignoreCase ? StringComparison.InvariantCultureIgnoreCase : StringComparison.InvariantCulture) >= 0,
- () => string.Format("String doesn't contain substring\n expected: {0}\n found: {1}", Utils.FormatValue(contains), Utils.FormatValue(str)));
- }
- public static void Contains(string str, string contains)
- {
- Contains(str, contains, false);
- }
- public static void DoesNotContain(string str, string contains, bool ignoreCase = false)
- {
- Throw(str.IndexOf(contains, ignoreCase ? StringComparison.InvariantCultureIgnoreCase : StringComparison.InvariantCulture) < 0,
- () => string.Format("String does contain substring\n didn't expect: {0}\n found: {1}", Utils.FormatValue(contains), Utils.FormatValue(str)));
- }
- public static void Matches(string str, string regex, RegexOptions options = RegexOptions.None)
- {
- Throw(new Regex(regex, options).IsMatch(str), () => string.Format("String doesn't match expression\n regex: \"{0}\"\n found: {1}", regex, Utils.FormatValue(str)));
- }
- public static void DoesNotMatch(string str, string regex, RegexOptions options = RegexOptions.None)
- {
- Throw(!(new Regex(regex, options).IsMatch(str)), () => string.Format("String matches expression\n regex: \"{0}\"\n found: {1}", regex, Utils.FormatValue(str)));
- }
- public static void IsNull(object val)
- {
- Throw(val == null, () => string.Format("Object reference is not null - {0}", Utils.FormatValue(val)));
- }
- public static void IsNotNull(object val)
- {
- Throw(val != null, () => "Object reference is null");
- }
- public static void Compare(object a, object b, Func<int, bool> Check, string comparison)
- {
- Throw(Check((a as IComparable).CompareTo(b)), () => string.Format("Comparison failed: {0} {1} {2}", Utils.FormatValue(a), comparison, Utils.FormatValue(b)));
- }
- public static void Greater<T>(T a, T b)
- {
- Compare(a, b, r => r > 0, ">");
- }
- public static void GreaterOrEqual<T>(T a, T b)
- {
- Compare(a, b, r => r >= 0, ">");
- }
- public static void Less<T>(T a, T b)
- {
- Compare(a, b, r => r < 0, ">");
- }
- public static void LessOrEqual<T>(T a, T b)
- {
- Compare(a, b, r => r <= 0, ">");
- }
- public static void IsInstanceOf(Type t, object o)
- {
- IsNotNull(o); Throw(o.GetType() == t, () => string.Format("Object type mismatch, expected {0} found {1}", t.FullName, o.GetType().FullName));
- }
- public static void IsNotInstanceOf(Type t, object o)
- {
- IsNotNull(o); Throw(o.GetType() != t, () => string.Format("Object type mismatch, should not be {0}", t.FullName));
- }
- public static void IsInstanceOf<T>(object o)
- {
- IsInstanceOf(typeof(T), o);
- }
- public static void IsNotInstanceOf<T>(object o)
- {
- IsNotInstanceOf(typeof(T), o);
- }
- public static void IsAssignableFrom(Type t, object o)
- {
- IsNotNull(o); Throw(o.GetType().IsAssignableFrom(t), () => string.Format("Object type mismatch, expected a type assignable from {0} found {1}", t.FullName, o.GetType().FullName));
- }
- public static void IsNotAssignableFrom(Type t, object o)
- {
- IsNotNull(o); Throw(!o.GetType().IsAssignableFrom(t), () => string.Format("Object type mismatch, didn't expect a type assignable from {0} found {1}", t.FullName, o.GetType().FullName));
- }
- public static void IsAssignableFrom<T>(object o)
- {
- IsAssignableFrom(typeof(T), o);
- }
- public static void IsNotAssignableFrom<T>(object o)
- {
- IsNotAssignableFrom(typeof(T), o);
- }
- public static void IsAssignableTo(Type t, object o)
- {
- IsNotNull(o); Throw(t.IsAssignableFrom(o.GetType()), () => string.Format("Object type mismatch, expected a type assignable to {0} found {1}", t.FullName, o.GetType().FullName));
- }
- public static void IsNotAssignableTo(Type t, object o)
- {
- IsNotNull(o); Throw(!t.IsAssignableFrom(o.GetType()), () => string.Format("Object type mismatch, didn't expect a type assignable to {0} found {1}", t.FullName, o.GetType().FullName));
- }
- public static void IsAssignableTo<T>(object o)
- {
- IsAssignableTo(typeof(T), o);
- }
- public static void IsNotAssignableTo<T>(object o)
- {
- IsNotAssignableTo(typeof(T), o);
- }
- public static Exception Throws(Type t, Action code)
- {
- try
- {
- code();
- }
- catch (Exception x)
- {
- Throw(t.IsAssignableFrom(x.GetType()), () => string.Format("Wrong exception type caught, expected {0} received {1}", t.FullName, Utils.FormatValue(x)));
- return x;
- }
- throw new AssertionException(string.Format("Failed to throw exception of type {0}", t.FullName));
- }
- public static TX Throws<TX>(Action code) where TX : Exception
- {
- return (TX)Throws(typeof(TX), code);
- }
- public static void DoesNotThrow(Action code)
- {
- try
- {
- code();
- }
- catch (Exception x)
- {
- Throw(false, () => string.Format("Unexpected exception {0}", Utils.FormatValue(x)));
- }
- }
- public static void ForEach<TSource>(IEnumerable<TSource> source, Action<TSource> code)
- {
- int index = 0;
- foreach (var i in source)
- {
- try
- {
- code(i);
- }
- catch (Exception x)
- {
- throw new AssertionException(string.Format("Collection assertion failed at item {0}\n Collection: {1}\n Inner Exception: {2}", index, Utils.FormatValue(source), x.Message));
- }
- index++;
- }
- }
- public static void AllItemsAreNotNull<T>(IEnumerable<T> coll)
- {
- int index = 0;
- foreach (var i in coll)
- {
- if (i == null)
- {
- throw new AssertionException(string.Format("Collection has a null item at index {0}", index));
- }
- index++;
- }
- }
- public static void AllItemsAreUnique<T>(IEnumerable<T> coll)
- {
- var list = coll.ToList();
- for (int i = 0; i < list.Count; i++)
- {
- for (int j = i + 1; j < list.Count; j++)
- {
- if (object.Equals(list[i], list[j]))
- throw new AssertionException(string.Format("Collection items are not unique\n [{0}] = {1}\n [{2}] = {3}\n", i, list[i], j, list[j]));
- }
- }
- }
- public static void AllItemsAreEqual<Ta, Tb>(IEnumerable<Ta> a, IEnumerable<Tb> b, Func<Ta, Tb, bool> CompareEqual)
- {
- var e1 = a.GetEnumerator();
- var e2 = b.GetEnumerator();
- int index = 0;
- while (true)
- {
- bool have1 = e1.MoveNext();
- bool have2 = e2.MoveNext();
- if (!have1 && !have2)
- return;
- if (!have1 || !have2 || !CompareEqual(e1.Current, e2.Current))
- throw new AssertionException(string.Format("Collection are not equal at index {0}\n a[{0}] = {1}\n b[{0}] = {2}\n", index, e1.Current, e2.Current));
- index++;
- }
- }
- public static void AllItemsAreEqual<T>(IEnumerable<T> a, IEnumerable<T> b)
- {
- AllItemsAreEqual<T, T>(a, b, (x, y) => object.Equals(x, y));
- }
- public static void AllItemsAreInstancesOf(Type t, System.Collections.IEnumerable coll)
- {
- int index = 0;
- foreach (object o in coll)
- {
- if (o == null || o.GetType() != t)
- {
- throw new AssertionException(string.Format("Collection item at index {0} is of the wrong type, expected {1} but found {2}", index, t.FullName, o == null ? "null" : o.GetType().FullName));
- }
- index++;
- }
- }
- public static void AllItemsAreInstancesOf<T>(System.Collections.IEnumerable coll)
- {
- AllItemsAreInstancesOf(typeof(T), coll);
- }
- private static int IndexOf<Ta, Tb>(Ta Item, List<Tb> list, Func<Ta, Tb, bool> CompareEqual)
- {
- for (int j = 0; j < list.Count; j++)
- {
- if (CompareEqual(Item, list[j]))
- return j;
- }
- return -1;
- }
- public static void IsSubsetOf<Ta, Tb>(IEnumerable<Ta> subset, IEnumerable<Tb> superset, Func<Ta, Tb, bool> CompareEqual)
- {
- var list = superset.ToList();
- int index = 0;
- foreach (var i in subset)
- {
- int pos = IndexOf<Ta, Tb>(i, list, CompareEqual);
- if (pos < 0)
- throw new AssertionException(string.Format("Collection is not a subset (check subset index {0}\n subset = {1}\n superset = {2}", index, Utils.FormatValue(subset), Utils.FormatValue(superset)));
- list.RemoveAt(pos);
- index++;
- }
- }
- public static void IsSubsetOf<T>(IEnumerable<T> subset, IEnumerable<T> superset)
- {
- IsSubsetOf<T, T>(subset, superset, (x, y) => Object.Equals(x, y));
- }
- public static void IsNotSubsetOf<Ta, Tb>(IEnumerable<Ta> subset, IEnumerable<Tb> superset, Func<Ta, Tb, bool> CompareEqual)
- {
- var list = superset.ToList();
- foreach (var i in subset)
- {
- int pos = IndexOf<Ta, Tb>(i, list, CompareEqual);
- if (pos < 0)
- return;
- list.RemoveAt(pos);
- }
- throw new AssertionException(string.Format("Collection is a subset\n subset = {0}\n superset = {1}", Utils.FormatValue(subset), Utils.FormatValue(superset)));
- }
- public static void IsNotSubsetOf<T>(IEnumerable<T> subset, IEnumerable<T> superset)
- {
- IsNotSubsetOf<T, T>(subset, superset, (x, y) => Object.Equals(x, y));
- }
- static bool TestEquivalent<Ta, Tb>(IEnumerable<Ta> a, IEnumerable<Tb> b, Func<Ta, Tb, bool> CompareEqual)
- {
- var list = b.ToList();
- foreach (var i in a)
- {
- int pos = IndexOf(i, list, CompareEqual);
- if (pos < 0)
- return false;
- list.RemoveAt(pos);
- }
- return list.Count == 0;
- }
- public static void AreEquivalent<Ta, Tb>(IEnumerable<Ta> a, IEnumerable<Tb> b, Func<Ta, Tb, bool> CompareEqual)
- {
- Throw(TestEquivalent(a, b, CompareEqual), () => string.Format("Collections are not equivalent\n lhs: {0}\n rhs: {1}", Utils.FormatValue(a), Utils.FormatValue(b)));
- }
- public static void AreNotEquivalent<Ta, Tb>(IEnumerable<Ta> a, IEnumerable<Tb> b, Func<Ta, Tb, bool> CompareEqual)
- {
- Throw(!TestEquivalent(a, b, CompareEqual), () => string.Format("Collections are not equivalent\n lhs: {0}\n rhs: {1}", Utils.FormatValue(a), Utils.FormatValue(b)));
- }
- public static void AreEquivalent<T>(IEnumerable<T> a, IEnumerable<T> b)
- {
- AreEquivalent<T, T>(a, b, (x, y) => Object.Equals(x, y));
- }
- public static void AreNotEquivalent<T>(IEnumerable<T> a, IEnumerable<T> b)
- {
- AreNotEquivalent<T, T>(a, b, (x, y) => Object.Equals(x, y));
- }
- }
- // Runner - runs a set of tests
- public class Runner
- {
- public Runner()
- {
- }
- public ResultsWriter Output
- {
- get;
- set;
- }
- public static bool? BoolFromString(string str)
- {
- if (string.IsNullOrWhiteSpace(str)) return null;
- str = str.ToLowerInvariant();
- return (str == "yes" || str == "y" || str == "true" || str == "t" || str == "1");
- }
- public static int RunMain(string[] args)
- {
- return new Runner().Run(args);
- }
- // Run all test fixtures in the calling assembly - unless one or more
- // marked as active in which case only those will be run
- public int Run(string[] args)
- {
- // Parse command line args
- bool? showreport = null;
- bool? htmlreport = null;
- bool? dirtyexit = null;
- bool? runall = null;
- bool? verbose = null;
- string output_file = null;
- foreach (var a in args)
- {
- if (a.StartsWith("-") || a.StartsWith("/"))
- {
- int colon_pos = a.IndexOf(":");
- string sw = a.Substring(1, colon_pos < 0 ? a.Length - 1 : colon_pos - 1);
- string val = colon_pos < 0 ? null : a.Substring(colon_pos + 1);
- switch (sw)
- {
- case "showreport": showreport = BoolFromString(val) ?? true; break;
- case "htmlreport": htmlreport = BoolFromString(val) ?? true; break;
- case "dirtyexit": dirtyexit = BoolFromString(val) ?? true; break;
- case "runall": runall = BoolFromString(val) ?? true; break;
- case "verbose": verbose = BoolFromString(val) ?? false; break;
- case "out": output_file = val; break;
- default:
- Console.WriteLine("Warning: Unknown switch '{0}' ignored", a);
- break;
- }
- }
- }
- if (htmlreport ?? true)
- Output = new HtmlResultsWriter(showreport, output_file);
- else
- Output = new PlainTextResultsWriter(output_file == null ? null : new StreamWriter(output_file));
- Output.Verbose = verbose ?? false;
- var totalTime = Stopwatch.StartNew();
- _statsStack.Clear();
- _statsStack.Push(new Stats());
- var old = Console.Out;
- Console.SetOut(Output);
- RunInternal(Assembly.GetCallingAssembly(), null, null, runall ?? false);
- Console.SetOut(old);
- totalTime.Stop();
- Output.Complete(_statsStack.Pop(), _otherTimes.ElapsedMilliseconds, totalTime.ElapsedMilliseconds);
- if (dirtyexit ?? true)
- System.Diagnostics.Process.GetCurrentProcess().Kill();
- return 0;
- }
- // Helper to create instances of test fixtures
- private object CreateInstance(Type t, object[] args)
- {
- try
- {
- _otherTimes.Start();
- return Activator.CreateInstance(t, args);
- }
- catch (Exception x)
- {
- Output.WriteException(x);
- Stats.Errors++;
- return null;
- }
- finally
- {
- _otherTimes.Stop();
- }
- }
- // Internally called to recursively run tests in an assembly, testfixture, test method etc...
- private void RunInternal(object scope, object instance, object[] arguments, bool RunAll)
- {
- // Assembly?
- var a = scope as Assembly;
- if (a != null)
- {
- StartTest(a, null);
- RunAll = RunAll || !a.HasActive();
- foreach (var type in a.GetTypes().Where(i => i.IsTestFixture() && (RunAll || i.HasActive())))
- RunInternal(type, null, null, RunAll);
- EndTest();
- }
- // Test Fixture class
- var t = scope as Type;
- if (t != null)
- {
- if (arguments == null)
- {
- bool runAllTestFixturesInstances = RunAll || !t.IsActive();
- bool runAllTestMethods = RunAll || !t.HasActiveMethods();
- foreach (TestFixtureAttribute tfa in t.GetCustomAttributes(typeof(TestFixtureAttribute), false).Where(x => runAllTestFixturesInstances || ((TestFixtureAttribute)x).Active))
- foreach (var args in tfa.GetArguments(t, null))
- RunInternal(t, null, args, runAllTestMethods);
- }
- else
- {
- StartTest(t, arguments);
- var inst = CreateInstance(t, arguments);
- if (inst != null)
- RunInternal(null, inst, null, RunAll);
- EndTest();
- }
- }
- // Test Fixture instance
- if (instance != null && instance.GetType().IsTestFixture())
- {
- var tf = instance;
- if (scope == null)
- {
- if (!RunSetupTeardown(instance, true, true))
- return;
- foreach (var m in tf.GetType().GetMethods().Where(x => RunAll || x.IsActive()))
- RunInternal(m, instance, null, RunAll);
- RunSetupTeardown(instance, false, true);
- }
- var method = scope as MethodInfo;
- if (method != null)
- {
- if (arguments == null)
- {
- foreach (TestAttribute i in method.GetCustomAttributes(typeof(TestAttribute), false).Where(x => RunAll || ((TestAttribute)x).Active))
- foreach (var args in i.GetArguments(method.DeclaringType, instance))
- {
- if (args.Length != method.GetParameters().Length)
- {
- Output.WriteWarning("{0} provided in an incorrect number of arguments (expected {1} but found {2}) - skipped", i.GetType().FullName, method.GetParameters().Length, args.Length);
- Stats.Warnings++;
- continue;
- }
- RunInternal(method, instance, args, RunAll);
- }
- }
- else
- {
- RunTest(method, tf, arguments);
- }
- }
- }
- }
- // Run a single test
- public void RunTest(MethodInfo Target, object instance, object[] Params)
- {
- StartTest(Target, Params);
- var sw = new Stopwatch();
- try
- {
- if (!RunSetupTeardown(instance, true, false))
- {
- EndTest();
- return;
- }
- sw.Start();
- Target.Invoke(instance, Params);
- Stats.Elapsed = sw.ElapsedMilliseconds;
- Stats.Passed++;
- }
- catch (Exception x)
- {
- Stats.Elapsed = sw.ElapsedMilliseconds;
- var invoc = x as TargetInvocationException;
- if (invoc != null)
- x = invoc.InnerException;
- Output.WriteException(x);
- Stats.Errors++;
- }
- RunSetupTeardown(instance, false, false);
- EndTest();
- }
- Stopwatch _otherTimes = new Stopwatch();
- [SkipInStackTrace]
- public bool RunSetupTeardown(object instance, bool setup, bool fixture)
- {
- try
- {
- foreach (var m in instance.GetType().GetMethods().Where(x => x.GetCustomAttributes(typeof(SetUpTearDownAttributeBase), false)
- .Cast<SetUpTearDownAttributeBase>().Any((SetUpTearDownAttributeBase y) => y.ForSetup == setup && y.ForFixture == fixture)))
- {
- _otherTimes.Start();
- try
- {
- m.Invoke(instance, null);
- }
- finally
- {
- _otherTimes.Stop();
- }
- }
- return true;
- }
- catch (Exception x)
- {
- var invoc = x as TargetInvocationException;
- if (invoc != null)
- x = invoc.InnerException;
- Output.WriteException(x);
- Stats.Errors++;
- return false;
- }
- }
- private void StartTest(object Target, object[] Params)
- {
- var stats = new Stats() { Target = Target };
- _statsStack.Push(stats);
- Output.StartTest(Target, Params);
- }
- private void EndTest()
- {
- var old = Stats;
- _statsStack.Pop();
- Stats.Add(old);
- Output.EndTest(old);
- }
- private Stack<Stats> _statsStack = new Stack<Stats>();
- public Stats Stats { get { return _statsStack.Peek(); } }
- }
- public class Stats
- {
- public object Target { get; set; }
- public int Errors { get; set; }
- public int Warnings { get; set; }
- public int Passed { get; set; }
- public long Elapsed { get; set; }
- public void Add(Stats other)
- {
- Errors += other.Errors;
- Warnings += other.Warnings;
- Passed += other.Passed;
- Elapsed += other.Elapsed;
- }
- }
- // The exception thrown when an assertion fails
- public class AssertionException : Exception
- {
- public AssertionException(string message) : base(message) { }
- }
- // Used to mark utility functions that throw assertion exceptions so the stack trace can be unwound to the actual place the assertion originates
- [AttributeUsage(AttributeTargets.Method | AttributeTargets.Class)]
- public class SkipInStackTraceAttribute : Attribute
- {
- }
- // Base class for setup/teardown attributes
- public class SetUpTearDownAttributeBase : Attribute
- {
- public SetUpTearDownAttributeBase(bool forSetup, bool forFixture)
- {
- this.ForSetup = forSetup;
- this.ForFixture = forFixture;
- }
- public bool ForSetup { get; set; }
- public bool ForFixture { get; set; }
- }
- // A bunch of utility functions and extension methods
- public static class Utils
- {
- public static IEnumerable<StackFrame> SimplifyStackTrace(StackTrace st)
- {
- foreach (var f in st.GetFrames())
- {
- if (f.GetMethod().GetCustomAttributes(typeof(SkipInStackTraceAttribute), false).Length != 0 ||
- (f.GetMethod().DeclaringType != null && f.GetMethod().DeclaringType.GetCustomAttributes(typeof(SkipInStackTraceAttribute), false).Length != 0) ||
- f.GetFileName() == null)
- continue;
- if (f.GetMethod().IsSpecialName | f.GetMethod().Name.StartsWith("<"))
- break;
- yield return f;
- }
- }
- public static IEnumerable<Tuple<int, string>> ExtractLinesFromTextFile(string file, int line, int extra = 2)
- {
- try
- {
- if (line <= extra)
- line = extra + 1;
- return System.IO.File.ReadAllLines(file).Skip(line - extra - 1).Take(extra * 2 + 1).Select((l, i) => new Tuple<int, string>(i + line - extra, l));
- }
- catch (Exception)
- {
- return new Tuple<int, string>[] { };
- }
- }
- // Format any value for diagnostic display
- public static string FormatValue(object value)
- {
- if (value == null)
- return "null";
- var str = value as string;
- if (str != null)
- {
- str = str.Replace("\"", "\\\"").Replace("\r", "\\r").Replace("\n", "\\n").Replace("\t", "\\t").Replace("\0", "\\0");
- return string.Format("\"{0}\"", str);
- }
- if (value.GetType() == typeof(int) || value.GetType() == typeof(long) || value.GetType() == typeof(bool))
- return value.ToString();
- var d = value as System.Collections.IDictionary;
- if (d != null)
- return string.Format("{{{0}}}", string.Join(", ", d.AsDictionaryEntries().Select(de => string.Format("{{ {0}, {1} }}", FormatValue(de.Key), FormatValue(de.Value)))));
- var e = value as System.Collections.IEnumerable;
- if (e != null)
- return string.Format("[{0}]", string.Join(", ", e.Cast<object>().Select(v => FormatValue(v))));
- var x = value as Exception;
- if (x != null)
- return string.Format("[{0}] {1}", value.GetType().FullName, x.Message);
- return string.Format("[{0}] {1}", value.GetType().FullName, value.ToString());
- }
- public static IEnumerable<System.Collections.DictionaryEntry> AsDictionaryEntries(this System.Collections.IDictionary dictionary)
- {
- foreach (var de in dictionary)
- yield return (System.Collections.DictionaryEntry)de;
- }
- public static string FormatArguments(object[] args)
- {
- return string.Format("({0})", args == null ? "" : string.Join(", ", args.Select(v => FormatValue(v))));
- }
- // Format the name of a test target
- public static string FormatTarget(object o)
- {
- var mb = o as MethodBase;
- if (mb != null)
- return "test " + mb.Name;
- var t = o as Type;
- if (t != null && t.IsClass)
- return "testfixture " + t.Name;
- var a = o as Assembly;
- if (a != null)
- return "assembly " + a.FullName;
- return null;
- }
- public static int CountCommonPrefix(string a, string b, bool IgnoreCase)
- {
- int i = 0;
- while (i < Math.Min(a.Length, b.Length) && (IgnoreCase ? (char.ToUpperInvariant(a[i]) == char.ToUpperInvariant(b[i])) : (a[i] == b[i])))
- i++;
- return i;
- }
- public static string GetStringExtract(string str, int offset)
- {
- if (offset > 15)
- str = "..." + str.Substring(offset - 10);
- if (str.Length > 30)
- str = str.Substring(0, 20) + "...";
- return str;
- }
- public static bool IsTestFixture(this Type t)
- {
- return t.IsClass && !t.IsAbstract && t.GetCustomAttributes(typeof(TestFixtureAttribute), false).Any();
- }
- public static bool IsTestMethod(this MethodInfo mi)
- {
- return mi.GetCustomAttributes(typeof(TestAttribute), false).Any();
- }
- public static bool IsActive(this ICustomAttributeProvider p)
- {
- return p.GetCustomAttributes(typeof(TestBaseAttribute), false).Any(a => ((TestBaseAttribute)a).Active);
- }
- public static bool HasActiveMethods(this Type t)
- {
- return t.GetMethods().Any(m => m.IsActive());
- }
- public static bool HasActive(this Type t)
- {
- return t.IsActive() || t.HasActiveMethods();
- }
- public static bool HasActive(this Assembly a)
- {
- return a.GetTypes().Any(t => t.HasActive());
- }
- }
- // Base class for result writers
- public abstract class ResultsWriter : TextWriter
- {
- public bool Verbose;
- public abstract void StartTest(object Target, object[] Arguments);
- public abstract void EndTest(Stats stats);
- public virtual void Complete(Stats stats, long OtherTimes, long ActualTime)
- {
- }
- public virtual void WriteWarning(string str, params object[] args)
- {
- WriteLine(str, args);
- }
- public virtual void WriteError(string str, params object[] args)
- {
- WriteLine(str, args);
- }
- public abstract void WriteException(Exception x);
- }
- // Plain text results writer (aka console output)
- public class PlainTextResultsWriter : ResultsWriter
- {
- TextWriter target;
- public PlainTextResultsWriter(TextWriter target = null)
- {
- this.target = target == null ? Console.Out : target;
- }
- public override void StartTest(object Target, object[] Arguments)
- {
- WriteIndented(string.Format("{0}{1}\n", Utils.FormatTarget(Target), Utils.FormatArguments(Arguments)));
- _indentDepth += (Target as MethodBase) != null ? 2 : 1;
- }
- public override void EndTest(Stats stats)
- {
- _indentDepth -= (stats.Target as MethodBase) != null ? 2 : 1;
- }
- public override void Complete(Stats stats, long OtherTimes, long ActualTime)
- {
- bool Success = stats.Errors == 0 && stats.Warnings == 0;
- var delim = new string(Success ? '-' : '*', 40);
- target.WriteLine("\nTest cases: {0,10:#,##0}ms\nSetup/teardown: {1,10:#,##0}ms\nTest framework: {2,10:#,##0}ms",
- stats.Elapsed, OtherTimes, ActualTime - (stats.Elapsed + OtherTimes));
- if (Success)
- target.WriteLine("\n{0}\nAll {1} tests passed\n{0}\n", delim, stats.Passed, stats.Elapsed);
- else
- target.WriteLine("\n{0}\n{1} Errors, {2} Warnings, {3} passed\n{0}\n", delim, stats.Errors, stats.Warnings, stats.Passed);
- target.Flush();
- }
- public override void Write(char value)
- {
- Write(value.ToString());
- }
- public override void Write(char[] buffer, int index, int count)
- {
- Write(new String(buffer, index, count));
- }
- public override void Write(string str)
- {
- if (Verbose)
- WriteIndented(str);
- }
- public void WriteIndented(string str)
- {
- string indent = new string(' ', _indentDepth * 2);
- if (_indentPending)
- target.Write(indent);
- _indentPending = str.EndsWith("\n");
- if (_indentPending)
- str = str.Substring(0, str.Length - 1);
- str = str.Replace("\n", "\n" + indent);
- target.Write(str);
- if (_indentPending)
- target.Write("\n");
- }
- public override Encoding Encoding
- {
- get { return Encoding.UTF8; }
- }
- public override void WriteException(Exception x)
- {
- var assert = x as AssertionException;
- if (assert != null)
- WriteIndented(string.Format("\nAssertion failed - {0}\n\n", assert.Message));
- else
- WriteIndented(string.Format("\nException {0}: {1}\n\n", x.GetType().FullName, x.Message));
- StackFrame first = null;
- foreach (var f in Utils.SimplifyStackTrace(new StackTrace(x, true)))
- {
- if (first == null) first = f;
- WriteIndented(string.Format(" {0} - {1}({2})\n", f.GetMethod().Name, f.GetFileName(), f.GetFileLineNumber()));
- }
- if (first != null)
- {
- WriteIndented("\n");
- foreach (var l in Utils.ExtractLinesFromTextFile(first.GetFileName(), first.GetFileLineNumber()))
- WriteIndented(string.Format(" {0:00000}:{1}{2}\n", l.Item1, l.Item1 == first.GetFileLineNumber() ? "->" : " ", l.Item2));
- }
- WriteIndented("\n");
- }
- bool _indentPending = true;
- int _indentDepth = 0;
- }
- // HTML Results Writer
- public class HtmlResultsWriter : PlainTextResultsWriter
- {
- public HtmlResultsWriter(bool? showreport, string output_file)
- {
- _showreport = showreport;
- _output_file = output_file == null ? "report.html" : output_file;
- _tagHtml.Append(new Tag("head").Append(StylesAndScript));
- var body = new Tag("body");
- _tagHtml.Append(body);
- _current = body;
- _dateOfTest = DateTime.Now;
- }
- bool? _showreport;
- string _output_file;
- Tag _tagHtml = new Tag("html");
- Tag _current;
- Tag _pre;
- DateTime _dateOfTest;
- StringWriter _bufferedOutput;
- public override void StartTest(object Target, object[] Arguments)
- {
- // Work out title
- string title = "", css_class = "";
- var a = Target as Assembly;
- var t = Target as Type;
- var m = Target as MethodInfo;
- if (a != null)
- {
- title = "Test Results for " + a.ManifestModule.Name;
- css_class = "assembly";
- }
- else if (t != null)
- {
- title = string.Format("Test Fixture {0}{1}", t.FullName, Utils.FormatArguments(Arguments));
- css_class = "testfixture";
- }
- else if (m != null)
- {
- title = string.Format("{0}{1}", m.Name, Utils.FormatArguments(Arguments));
- css_class = "test";
- }
- // Create the test div
- var divTest = new Tag("div");
- divTest.SetAttribute("class", css_class);
- divTest.Append(new Tag("div").SetAttribute("class", "title")
- .Append(new Tag("span").AddClass("indicator"))
- .Append(new Tag("a").AddClass("toggle").SetAttribute("href", "#").EncodeAndAppend(title)));
- // Make room for assembly summary which we'll populate later
- if (a != null)
- divTest.Append(new Tag("div").SetAttribute("class", "summary"));
- // Make the collapsable content div
- var divContent = new Tag("div").SetAttribute("class", "content");
- divTest.Append(divContent);
- // Prepare to continue writing in the collapsable content div
- _current.Append(divTest);
- _current = divContent;
- _pre = null;
- _bufferedOutput = Verbose ? null : new StringWriter();
- base.StartTest(Target, Arguments);
- }
- public override void EndTest(Stats stats)
- {
- base.EndTest(stats);
- // If this was a test, and it passed, write something to indicate that
- if (stats.Target as MethodInfo != null && stats.Passed == 1)
- {
- if (_pre != null)
- WriteToTestOutput("\n");
- WriteToTestOutput("Test passed!\n");
- _current.Parent.Find("div", "title", null).Append(string.Format(" <small>{0}ms</small>", stats.Elapsed));
- }
- // Pop the content div
- _current = _current.Parent;
- // Flag the test/testfixture/assembly as pass/fail and collapse passed tests
- if (stats.Errors == 0)
- _current.AddClass("pass" + ((stats.Target as MethodInfo != null) ? " collapsed" : ""));
- else
- _current.AddClass("fail");
- // Write the summary
- var t = _current.Find("div", "summary", null);
- if (t != null)
- {
- t.Append(string.Format("Passed: {0} Failed: {1} Warnings: {2} Time in Test Cases: {3}ms", stats.Passed, stats.Errors, stats.Warnings, stats.Elapsed));
- }
- // Pop stack
- _current = _current.Parent;
- }
- public override void Complete(Stats stats, long OtherTimes, long ActualTime)
- {
- base.Complete(stats, OtherTimes, ActualTime);
- _current.Append(new Tag("div").AddClass("misc_info")
- .Append(string.Format("Time in test cases {0:#,##0}ms, setup/teardown {1:#,##0}ms, test framework {2:#,##0}ms<br/>", stats.Elapsed, OtherTimes, ActualTime - (stats.Elapsed + OtherTimes)))
- .Append(string.Format("Tests run at {0} by {1} on {2} under {3}<br/>", _dateOfTest, System.Environment.UserName, System.Environment.MachineName, System.Environment.OSVersion))
- .Append(string.Format("Command line: {0}", string.Join(" ", System.Environment.GetCommandLineArgs())))
- );
- // Save report
- using (var output = new StreamWriter(_output_file))
- {
- _tagHtml.Render(output);
- }
- if (!_showreport.HasValue)
- {
- System.GC.Collect();
- try
- {
- base.WriteIndented("View Report?");
- var key = Console.ReadKey();
- base.WriteIndented("\n");
- _showreport = key.KeyChar == 'Y' || key.KeyChar == 'y';
- }
- catch
- {
- _showreport = true;
- }
- }
- if (_showreport.Value)
- {
- System.Diagnostics.Process.Start(_output_file);
- }
- }
- public void WriteToTestOutput(string str)
- {
- if (_pre == null)
- {
- _pre = new Tag("pre");
- _current.Append(_pre);
- }
- _pre.EncodeAndAppend(str);
- }
- public override void Write(string str)
- {
- base.Write(str);
- WriteToTestOutput(str);
- if (_bufferedOutput != null)
- _bufferedOutput.Write(str);
- }
- public override void WriteException(Exception x)
- {
- if (_bufferedOutput != null)
- {
- WriteIndented(_bufferedOutput.ToString());
- _bufferedOutput = new StringWriter();
- }
- base.WriteException(x);
- // Split exception message into title and detail
- string title, detail = null;
- var assert = x as AssertionException;
- if (assert != null)
- {
- int cr = assert.Message.IndexOf('\n');
- if (cr > 0)
- {
- title = assert.Message.Substring(0, cr);
- detail = assert.Message.Substring(cr + 1);
- }
- else
- title = assert.Message;
- title = string.Format("Assertion failed - {0}", title);
- }
- else
- {
- title = string.Format("Exception {0}: {1}", x.GetType().FullName, x.Message);
- }
- var div = new Tag("div").SetAttribute("class", "exception");
- div.Append(new Tag("div").SetAttribute("class", "errormessage").EncodeAndAppend(title));
- if (detail != null)
- {
- div.Append(new Tag("p").EncodeAndAppend("Detail"));
- div.Append(new Tag("pre").EncodeAndAppend(detail));
- }
- div.Append(new Tag("p").EncodeAndAppend("Stack Trace"));
- var st = new Tag("pre").SetAttribute("class", "stacktrace");
- div.Append(st);
- StackFrame first = null;
- foreach (var f in Utils.SimplifyStackTrace(new StackTrace(x, true)))
- {
- if (first == null) first = f;
- st.EncodeAndAppend(string.Format("{0} - {1}({2})\n", f.GetMethod().Name, f.GetFileName(), f.GetFileLineNumber()));
- }
- if (first != null)
- {
- div.Append(new Tag("p").EncodeAndAppend("Location"));
- var code = new Tag("pre").SetAttribute("class", "code");
- div.Append(code);
- foreach (var l in Utils.ExtractLinesFromTextFile(first.GetFileName(), first.GetFileLineNumber()))
- {
- if (l.Item1 == first.GetFileLineNumber())
- code.Append("<span class=\"highlighted\">");
- code.EncodeAndAppend(string.Format(" {0:00000}:{1}{2}\n", l.Item1, l.Item1 == first.GetFileLineNumber() ? "->" : " ", l.Item2));
- if (l.Item1 == first.GetFileLineNumber())
- code.Append("</span>");
- }
- }
- _pre = null;
- _current.Append(div);
- }
- // Report works in Chrome, FF4 and Safari. Doesn't work in IE (oh well)
- private const string StylesAndScript = @"<style>
- body { font-family:Arial; margin:20px; font-size:10pt}
- div.assembly>div.title { font-size:20pt; margin-bottom:20px; border-bottom:1px solid silver; padding-bottom:20px; }
- div.testfixture>div.title { font-size:16pt; }
- div.testfixture { margin-bottom:20px; }
- div.test { margin-left:20px; }
- div.test div.title { font-size:12pt; }
- div.errormessage { color:Red; padding-top:10px; }
- div.pass>div.title { color:#808080; }
- span.highlighted { color:Red; }
- pre.code { background-color:#f0f0f0; }
- div.summary { border:1px solid silver; padding:10px; margin-bottom:20px; text-align:center; }
- div.fail div.summary { background-color:Red; color:White; }
- div.pass div.summary { background-color:Lime; }
- div.collapsed div.content { display:none; }
- div.test div.content { border-left:2px solid #d0d0d0; padding-left:10px; margin-left:10px; }
- a { text-decoration:none; color:#606060; }
- a:hover { color:orange; }
- div>div.title>span.indicator { display:inline-block; width:10px; height:10px; background-color:lime; border-radius:7px}
- div.fail>div.title>span.indicator { display:inline-block; width:10px; height:10px; background-color:red; }
- div.assembly>div.title>span.indicator { display:none; }
- div.misc_info { color:#808080; }
- </style>
- <script>
- window.addEventListener(""load"", setup, false);
- function setup() {
- var divs = document.getElementsByClassName(""toggle"");
- for (var i = 0; i < divs.length; i++) {
- divs[i].onclick = function () {
- var top = this.parentNode.parentNode;
- if (top.className.indexOf("" collapsed"") < 0)
- top.className += "" collapsed"";
- else
- top.className = top.className.replace("" collapsed"", """");
- return false;
- }
- }
- }
- </script>";
- }
- public class Tag
- {
- public Tag(string tagName)
- {
- Name = tagName;
- }
- public string Name { get; set; }
- public Tag Parent { get; set; }
- public Dictionary<string, string> Attributes = new Dictionary<string, string>();
- List<object> _Content;
- public void Render(TextWriter w)
- {
- w.Write("<"); w.Write(Name);
- foreach (var a in Attributes)
- {
- w.Write(" "); w.Write(a.Key);
- if (a.Value != null)
- {
- w.Write("=\""); w.Write(Encode(a.Value)); w.Write("\"");
- }
- }
- w.Write(">");
- if (Name != "pre")
- w.Write("\n");
- if (_Content != null)
- {
- bool NeedNewLine = false;
- foreach (var i in _Content)
- {
- var t = i as Tag;
- if (t != null)
- {
- if (NeedNewLine)
- w.Write("\n");
- t.Render(w);
- NeedNewLine = false;
- }
- var s = i as String;
- if (s != null)
- {
- w.Write(s);
- NeedNewLine = true;
- }
- }
- }
- if (Name != "pre")
- w.Write("\n");
- w.Write("</"); w.Write(Name); w.Write(">\n");
- }
- public Tag Append(Tag t)
- {
- if (_Content == null)
- _Content = new List<object>();
- t.Parent = this;
- _Content.Add(t);
- return this;
- }
- public Tag Append(string s)
- {
- if (_Content == null)
- _Content = new List<object>();
- _Content.Add(s);
- return this;
- }
- public Tag EncodeAndAppend(string s)
- {
- Append(Encode(s));
- return this;
- }
- public Tag SetAttribute(string name, string value)
- {
- Attributes[name] = value;
- return this;
- }
- public Tag AddClass(string className)
- {
- if (Attributes.ContainsKey("class"))
- Attributes["class"] = Attributes["class"] + " " + className;
- else
- Attributes["class"] = className;
- return this;
- }
- public Tag Find(string tagName, string className, string id)
- {
- foreach (Tag t in _Content.Where(x => x as Tag != null))
- {
- if (tagName != null && t.Name != tagName)
- continue;
- if (className != null && t.Attributes["class"] != className)
- continue;
- if (id != null && t.Attributes["id"] != id)
- continue;
- return t;
- }
- return null;
- }
- public static string Encode(string str)
- {
- return str.Replace("&", "&").Replace("<", "<").Replace("\"", """).Replace("\'", "'").Replace(">", ">");
- }
- }
- }
|