Merge pull request #2274 from esdrubal/udpclientreceive
[mono.git] / mono / mini / TestDriver.cs
index 05facbab37795bd1845b656a50baceb3c5c57fc6..c3bf999d2729ae30c014fee5a8e32a4f4449ce2f 100644 (file)
 using System;
 using System.Reflection;
+using System.Collections.Generic;
 
+[AttributeUsageAttribute(AttributeTargets.All, Inherited = true, AllowMultiple = true)]
+public class CategoryAttribute : Attribute
+{
+       public CategoryAttribute (string category) {
+               Category = category;
+       }
+
+       public string Category {
+               get; set;
+       }
+}
 
 public class TestDriver {
 
        static public int RunTests (Type type, string[] args) {
                int failed = 0, ran = 0;
-               int result, expected, elen;
-               int i, j;
+               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> run_only = new List<string> ();
                if (args != null && args.Length > 0) {
-                       for (j = 0; j < args.Length; j++) {
-                               bool found = false;
+                       for (j = 0; j < args.Length;) {
                                if (args [j] == "--time") {
-                                       do_timings = true;
-                                       found = true;
-                               }
-                               if ((args [j] == "-v") || (args [j] == "--verbose")) {
-                                       verbose = true;
-                                       found = true;
-                               }
-
-                               if (found) {
-                                       string[] new_args = new string [args.Length - 1];
-                                       for (i = 0; i < j; ++i)
-                                               new_args [i] = args [i];
-                                       j++;
-                                       for (; j < args.Length; ++i, ++j)
-                                               new_args [i] = args [j];
-                                       args = new_args;
-                                       break;
+                                       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 = !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 {
+                                       Console.WriteLine ("Unknown argument: " + args [j]);
+                                       return 1;
                                }
                        }
                }
+               int nskipped = 0;
                methods = type.GetMethods (BindingFlags.Public|BindingFlags.NonPublic|BindingFlags.Static);
-               for (i = 0; i < methods.Length; ++i) {
-                       name = methods [i].Name;
-                       if (!name.StartsWith ("test_"))
-                               continue;
-                       if (args != null && args.Length > 0) {
-                               bool found = false;
-                               for (j = 0; j < args.Length; j++) {
-                                       if (name.EndsWith (args [j])) {
-                                               found = true;
-                                               break;
+               for (int iter = 0; iter < iterations; ++iter) {
+                       for (i = 0; i < methods.Length; ++i) {
+                               name = methods [i].Name;
+                               if (!name.StartsWith ("test_", StringComparison.Ordinal))
+                                       continue;
+                               if (run_only.Count > 0) {
+                                       bool found = false;
+                                       for (j = 0; j < run_only.Count; j++) {
+                                               if (name.EndsWith (run_only [j])) {
+                                                       found = true;
+                                                       break;
+                                               }
                                        }
+                                       if (!found)
+                                               continue;
+                               }
+                               if (exclude.Count > 0) {
+                                       var attrs = methods [i].GetCustomAttributes (typeof (CategoryAttribute), false);
+                                       bool skip = false;
+                                       foreach (CategoryAttribute attr in attrs) {
+                                               if (exclude.ContainsKey (attr.Category))
+                                                       skip = true;
+                                       }
+                                       if (skip) {
+                                               if (verbose)
+                                                       Console.WriteLine ("Skipping '{0}'.", name);
+                                               nskipped ++;
+                                               continue;
+                                       }
+                               }
+                               for (j = 5; j < name.Length; ++j)
+                                       if (!Char.IsDigit (name [j]))
+                                               break;
+                               if (verbose)
+                                       Console.WriteLine ("Running '{0}' ...", name);
+                               expected = Int32.Parse (name.Substring (5, j - 5));
+                               start = DateTime.Now;
+                               result = (int)methods [i].Invoke (null, null);
+                               if (do_timings) {
+                                       end = DateTime.Now;
+                                       long tdiff = end.Ticks - start.Ticks;
+                                       int mdiff = (int)tdiff/10000;
+                                       tms += mdiff;
+                                       Console.WriteLine ("{0} took {1} ms", name, mdiff);
+                               }
+                               ran++;
+                               if (result != expected) {
+                                       failed++;
+                                       Console.WriteLine ("{0} failed: got {1}, expected {2}", name, result, expected);
                                }
-                               if (!found)
-                                       continue;
-                       }
-                       for (j = 5; j < name.Length; ++j)
-                               if (!Char.IsDigit (name [j]))
-                                       break;
-                       if (verbose)
-                               Console.WriteLine ("Running '{0}' ...", name);
-                       expected = Int32.Parse (name.Substring (5, j - 5));
-                       start = DateTime.Now;
-                       result = (int)methods [i].Invoke (null, null);
-                       if (do_timings) {
-                               end = DateTime.Now;
-                               long tdiff = end.Ticks - start.Ticks;
-                               int mdiff = (int)tdiff/10000;
-                               tms += mdiff;
-                               Console.WriteLine ("{0} took {1} ms", name, mdiff);
-                       }
-                       ran++;
-                       if (result != expected) {
-                               failed++;
-                               Console.WriteLine ("{0} failed: got {1}, expected {2}", name, result, expected);
                        }
-               }
                
-               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);
+                       }
                }
-               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);
                return failed;
        }
@@ -89,3 +133,65 @@ public class TestDriver {
        }
 }
 
+/// 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;
+               }
+       }
+
+}