lyfz_repos/铂金版源码/CassiniDev-master/CassiniDev/CassiniDevBrowserTest.cs

// /* **********************************************************************************
//  *
//  * Copyright (c) Sky Sanders. All rights reserved.
//  * 
//  * This source code is subject to terms and conditions of the Microsoft Public
//  * License (Ms-PL). A copy of the license can be found in the license.htm file
//  * included in this distribution.
//  *
//  * You must not remove this notice, or any other, from this software.
//  *
//  * **********************************************************************************/
using System;
using System.Diagnostics;
using System.Threading;
using CassiniDev.ServerLog;

namespace CassiniDev
{
    /// <summary>
    /// A web test executor base on an idea from Nikhil Kothari's Script#
    /// http://projects.nikhilk.net/ScriptSharp
    /// 
    /// TODO: finer grained control over browser instances.
    /// TODO: create parser/abstraction for RequestEventArgs
    /// </summary>
    public class CassiniDevBrowserTest : CassiniDevServer
    {
        private readonly string _postKey = "testresults.axd";
        public string PostKey
        {
            get
            {
                return _postKey;
            }

        }
        public CassiniDevBrowserTest(string postKey)
        {
            _postKey = postKey;
        }
        public CassiniDevBrowserTest()
        {
        }

        public RequestEventArgs RunTest(string url)
        {
            return RunTest(url, WebBrowser.InternetExplorer, TimeSpan.FromMinutes(1.0));
        }

        public RequestEventArgs RunTest(string url, WebBrowser browser)
        {
            return RunTest(url, browser, TimeSpan.FromMinutes(1.0));
        }

        public RequestEventArgs RunTest(string url, WebBrowser browser, TimeSpan timeout)
        {

            if (browser == null)
            {
                throw new ArgumentNullException("browser");
            }
            if (string.IsNullOrEmpty(browser.ExecutablePath))
            {
                throw new InvalidOperationException("The specified browser could not be located.");
            }
            if (timeout.TotalMilliseconds == 0.0)
            {
                timeout = TimeSpan.FromMinutes(1.0);
            }
            var waitHandle = new AutoResetEvent(false);
            RequestEventArgs result = null;
            Process process;
            EventHandler<RequestEventArgs> logEventHandler = null;
            EventHandler<RequestEventArgs> handler = logEventHandler;
            logEventHandler = delegate(object sender, RequestEventArgs e)
            {
                if (e.RequestLog.Url.ToLower().Contains(_postKey))
                {
                    Server.RequestComplete -= handler;
                    result = e;
                    waitHandle.Set();
                }
            };
            try
            {
                var startInfo = new ProcessStartInfo(browser.ExecutablePath, url)
                {
                    UseShellExecute = true,
                    WindowStyle = ProcessWindowStyle.Minimized
                };
                Server.RequestComplete += logEventHandler;
                process = Process.Start(startInfo);
            }
            catch (Exception ex)
            {
                Server.RequestComplete -= logEventHandler;
                return new RequestEventArgs(Guid.Empty, new LogInfo { StatusCode = -1, Exception = ex.ToString() },
                                            new LogInfo());
            }
            bool flag = waitHandle.WaitOne(timeout);
            try
            {
                if (!process.CloseMainWindow())
                {
                    process.Kill();
                }
            }
            catch
            {
            }
            return flag ? result : new RequestEventArgs(Guid.Empty, new LogInfo { StatusCode = -2 }, new LogInfo());
        }

    }
}