if tests are taking too long, create a memory dump and abort

lostmsu · lostmsu · commit 848e6babce35 · 2020-02-21T10:46:52.000-08:00
diff --git a/src/embed_tests/GlobalTestsSetup.cs b/src/embed_tests/GlobalTestsSetup.cs
@@ -1,14 +1,35 @@
+using System;
+using System.ComponentModel;
+using System.Diagnostics;
+using System.IO;
+using System.Runtime.InteropServices;
+using System.Threading;
+
+using Microsoft.Win32.SafeHandles;
 using NUnit.Framework;
 using Python.Runtime;
 
 namespace Python.EmbeddingTest
 {
-
     // As the SetUpFixture, the OneTimeTearDown of this class is executed after
     // all tests have run.
     [SetUpFixture]
     public class GlobalTestsSetup
     {
+        [OneTimeSetUp]
+        public void GlobalSetup()
+        {
+            new Thread(() =>
+            {
+                Thread.Sleep(TimeSpan.FromSeconds(30));
+                UploadDump();
+                Console.Error.WriteLine("Test has been running for too long. Created memory dump");
+                Environment.Exit(1);
+            }) {
+                IsBackground = true,
+            }.Start();
+        }
+
         [OneTimeTearDown]
         public void FinalCleanup()
         {
@@ -17,5 +38,48 @@ public void FinalCleanup()
                 PythonEngine.Shutdown();
             }
         }
+
+        static void UploadDump()
+        {
+            var self = Process.GetCurrentProcess();
+
+            const string dumpPath = "memory.dmp";
+
+            // ensure DbgHelp is loaded
+            MiniDumpWriteDump(IntPtr.Zero, 0, IntPtr.Zero, MiniDumpType.Normal, IntPtr.Zero, IntPtr.Zero, IntPtr.Zero);
+
+            using (var fileHandle = CreateFile(dumpPath, FileAccess.Write, FileShare.Write,
+                securityAttrs: IntPtr.Zero, dwCreationDisposition: FileMode.Create, dwFlagsAndAttributes: 0,
+                hTemplateFile: IntPtr.Zero))
+            {
+                if (fileHandle.IsInvalid)
+                    throw new Win32Exception();
+
+                if (!MiniDumpWriteDump(self.Handle, self.Id, fileHandle.DangerousGetHandle(), MiniDumpType.Normal,
+                    causeException: IntPtr.Zero, userStream: IntPtr.Zero, callback: IntPtr.Zero))
+                    throw new Win32Exception();
+            }
+
+            Process.Start("appveyor", arguments: $"PushArtifact -Path \"{dumpPath}\"").WaitForExit();
+        }
+
+        [DllImport("kernel32", CharSet = CharSet.Auto, SetLastError = true)]
+        private static extern SafeFileHandle CreateFile(
+            string lpFileName,
+            FileAccess dwDesiredAccess,
+            FileShare dwShareMode,
+            IntPtr securityAttrs,
+            FileMode dwCreationDisposition,
+            int dwFlagsAndAttributes,
+            IntPtr hTemplateFile);
+        [DllImport("DbgHelp", SetLastError = true)]
+        static extern bool MiniDumpWriteDump(IntPtr hProcess,
+            int processID, IntPtr hFile, MiniDumpType dumpType,
+            IntPtr causeException, IntPtr userStream, IntPtr callback);
+
+        enum MiniDumpType
+        {
+            Normal,
+        }
     }
 }
