using System.Reflection;
using System.Collections.Generic;
+[AttributeUsageAttribute(AttributeTargets.All, Inherited = true, AllowMultiple = true)]
public class CategoryAttribute : Attribute
{
public CategoryAttribute (string category) {
static public int RunTests (Type type, string[] args) {
int failed = 0, ran = 0;
- int result, expected, elen;
+ int result, expected;
int i, j, iterations;
string name;
MethodInfo[] methods;
bool do_timings = false;
bool verbose = false;
+ bool quiet = false;
int tms = 0;
DateTime start, end = DateTime.Now;
iterations = 1;
var exclude = new Dictionary<string, string> ();
- List<string> new_args = new List<string> ();
+ List<string> run_only = new List<string> ();
if (args != null && args.Length > 0) {
- for (j = 0; j < args.Length; j++) {
+ for (j = 0; j < args.Length;) {
if (args [j] == "--time") {
- do_timings = true;
+ do_timings = !quiet;
j ++;
} else if (args [j] == "--iter") {
iterations = Int32.Parse (args [j + 1]);
j += 2;
} else if ((args [j] == "-v") || (args [j] == "--verbose")) {
- verbose = true;
+ verbose = !quiet;
+ j += 1;
+ } else if ((args [j] == "-q") || (args [j] == "--quiet")) {
+ quiet = true;
+ verbose = false;
+ do_timings = false;
+ j += 1;
} else if (args [j] == "--exclude") {
exclude [args [j + 1]] = args [j + 1];
j += 2;
+ } else if (args [j] == "--run-only") {
+ run_only.Add (args [j + 1]);
+ j += 2;
} else {
- new_args.Add (args [j]);
+ Console.WriteLine ("Unknown argument: " + args [j]);
+ return 1;
}
}
}
for (int iter = 0; iter < iterations; ++iter) {
for (i = 0; i < methods.Length; ++i) {
name = methods [i].Name;
- if (!name.StartsWith ("test_"))
+ if (!name.StartsWith ("test_", StringComparison.Ordinal))
continue;
- if (new_args.Count > 0) {
+ if (run_only.Count > 0) {
bool found = false;
- for (j = 0; j < new_args.Count; j++) {
- if (name.EndsWith (new_args [j])) {
+ for (j = 0; j < run_only.Count; j++) {
+ if (name.EndsWith (run_only [j])) {
found = true;
break;
}
}
}
- if (do_timings) {
- Console.WriteLine ("Total ms: {0}", tms);
+ if (!quiet) {
+ if (do_timings) {
+ Console.WriteLine ("Total ms: {0}", tms);
+ }
+ if (nskipped > 0)
+ Console.WriteLine ("Regression tests: {0} ran, {1} skipped, {2} failed in {3}", ran, nskipped, failed, type);
+ else
+ Console.WriteLine ("Regression tests: {0} ran, {1} failed in {2}", ran, failed, type);
}
- if (nskipped > 0)
- Console.WriteLine ("Regression tests: {0} ran, {1} skipped, {2} failed in {3}", ran, nskipped, failed, type);
- else
- Console.WriteLine ("Regression tests: {0} ran, {1} failed in {2}", ran, failed, type);
}
//Console.WriteLine ("Regression tests: {0} ran, {1} failed in [{2}]{3}", ran, failed, type.Assembly.GetName().Name, type);
}
}
+/// Provide tests with the ability to find out how much time they have to run before being timed out.
+public class TestTimeout {
+ const string ENV_TIMEOUT = "TEST_DRIVER_TIMEOUT_SEC";
+ private readonly TimeSpan availableTime;
+ private TimeSpan slack;
+ private DateTime startTime;
+
+ /// <summary>
+ /// How much time the test runner provided for us or TimeSpan.Zero if there is no bound.
+ /// </summary>
+ public TimeSpan AvailableTime { get { return availableTime; } }
+
+ public DateTime StartTime { get { return startTime; } }
+
+ /// <summary> Extra time to add when deciding if there
+ /// is still time to run. Bigger slack means less
+ /// time left.
+ /// </summary>
+ public TimeSpan Slack {
+ get { return slack; }
+ set { slack = value; }
+ }
+
+ public TestTimeout () {
+ availableTime = initializeAvailableTime ();
+ slack = defaultSlack ();
+ }
+
+ /// <summary>
+ /// Consider the test started.
+ /// </summary>
+ public void Start ()
+ {
+ startTime = DateTime.UtcNow;
+ }
+
+ public bool HaveTimeLeft ()
+ {
+ if (availableTime == TimeSpan.Zero)
+ return true;
+ var t = DateTime.UtcNow;
+ var finishTime = startTime + availableTime - slack;
+ return (t < finishTime);
+ }
+
+ private TimeSpan defaultSlack ()
+ {
+ return TimeSpan.FromSeconds (5);
+ }
+
+ private TimeSpan initializeAvailableTime ()
+ {
+ var e = System.Environment.GetEnvironmentVariable(ENV_TIMEOUT);
+ double d;
+ if (Double.TryParse(e, out d)) {
+ return TimeSpan.FromSeconds(d);
+ } else {
+ return TimeSpan.Zero;
+ }
+ }
+
+}