Helper.cs

// -----------------------------------------------------------------------
// <copyright file="Helper.cs" company="SimpleBrowser">
// Copyright © 2010 - 2019, Nathan Ridley and the SimpleBrowser contributors.
// See https://github.com/SimpleBrowserDotNet/SimpleBrowser/blob/master/readme.md
// </copyright>
// -----------------------------------------------------------------------

namespace SimpleBrowser.UnitTests
{
    using System;
    using System.IO;
    using System.Net;
    using System.Reflection;
    using System.Text;
    using System.Threading.Tasks;
    using Moq;
    using SimpleBrowser.Network;

    public class Helper
    {
        internal static string GetFromResources(string resourceName)
        {
            Assembly assem = Assembly.GetExecutingAssembly();
            using (Stream stream = assem.GetManifestResourceStream(resourceName))
            {
                using (StreamReader reader = new StreamReader(stream))
                {
                    return reader.ReadToEnd();
                }
            }
        }

        internal static IWebRequestFactory GetAllways200RequestMocker()
        {
            return new Always200RequestMocker();
        }

        private class Always200RequestMocker : IWebRequestFactory
        {
            public Always200RequestMocker()
            {
                this.ResponseContent = "";
            }

            public string ResponseContent { get; set; }

            #region IWebRequestFactory Members

            public IHttpWebRequest GetWebRequest(Uri url)
            {
                Mock<IHttpWebRequest> mock = new Mock<IHttpWebRequest>();
                mock.SetupAllProperties();
                mock.Setup(m => m.GetResponseAsync())
                    .Returns(() =>
                    {
                        Mock<IHttpWebResponse> mockResponse = new Mock<IHttpWebResponse>();
                        mockResponse.SetupAllProperties();
                        mockResponse.SetupProperty(m => m.Headers, new WebHeaderCollection());

                        byte[] responseContent = Encoding.UTF8.GetBytes(this.ResponseContent);
                        mockResponse.Setup(r => r.GetResponseStream()).Returns(new MemoryStream(responseContent));
                        return Task.FromResult(mockResponse.Object);
                    });
                mock.SetupProperty(m => m.Headers, new WebHeaderCollection());
                mock.Setup(m => m.GetRequestStreamAsync()).Returns(Task.FromResult((System.IO.Stream)new MemoryStream(new byte[60000])));
                
                return mock.Object;
            }

            #endregion IWebRequestFactory Members
        }

        internal static IWebRequestFactory GetMoviesRequestMocker()
        {
            return new MoviesRequestMocker();
        }

        private class MoviesRequestMocker : IWebRequestFactory
        {
            #region IWebRequestFactory Members

            public IHttpWebRequest GetWebRequest(Uri url)
            {
                Mock<IHttpWebRequest> mock = new Mock<IHttpWebRequest>();
                mock.SetupAllProperties();
                mock.Setup(m => m.GetResponseAsync())
                    .Returns(() =>
                    {
                        Mock<IHttpWebResponse> mockResponse = new Mock<IHttpWebResponse>();
                        mockResponse.SetupAllProperties();
                        mockResponse.SetupProperty(m => m.Headers, new WebHeaderCollection());

                        byte[] responseContent = new byte[0];
                        if (mock.Object.Method == "GET")
                        {
                            if (url.AbsolutePath == "/movies/")
                            {
                                responseContent = Encoding.UTF8.GetBytes(GetFromResources("SimpleBrowser.UnitTests.SampleDocs.movies1.htm"));
                            }

                            if (url.AbsolutePath == "/movies/Movies/Create")
                            {
                                responseContent = Encoding.UTF8.GetBytes(GetFromResources("SimpleBrowser.UnitTests.SampleDocs.movies2.htm"));
                            }
                        }
                        else if (mock.Object.Method == "POST")
                        {
                            if (url.AbsolutePath == "/movies/Movies/Create")
                            {
                                mockResponse.Object.StatusCode = HttpStatusCode.Moved;
                                mockResponse.Object.Headers.Add(HttpResponseHeader.Location, "http://localhost/movies/");
                            }
                        }
                        mockResponse.Setup(r => r.GetResponseStream()).Returns(new MemoryStream(responseContent));
                        return Task.FromResult(mockResponse.Object);
                    });
                mock.SetupProperty(m => m.Headers, new WebHeaderCollection());
                mock.Setup(m => m.GetRequestStreamAsync()).Returns(Task.FromResult((System.IO.Stream)new MemoryStream(new byte[60000])));
                
                return mock.Object;
            }

            #endregion IWebRequestFactory Members
        }

        internal static IWebRequestFactory GetIFramesMock()
        {
            return new IFramesRequestMocker();
        }

        private class IFramesRequestMocker : IWebRequestFactory
        {
            public IHttpWebRequest GetWebRequest(Uri url)
            {
                Mock<IHttpWebRequest> mock = new Mock<IHttpWebRequest>();
                mock.SetupAllProperties();
                mock.Setup(m => m.GetResponseAsync())
                    .Returns(() =>
                    {
                        Mock<IHttpWebResponse> mockResponse = new Mock<IHttpWebResponse>();
                        mockResponse.SetupAllProperties();
                        mockResponse.SetupProperty(m => m.Headers, new WebHeaderCollection());

                        byte[] responseContent = new byte[0];
                        if (mock.Object.Method == "GET")
                        {
                            if (url.AbsolutePath == "/")
                            {
                                responseContent = Encoding.UTF8.GetBytes(GetFromResources("SimpleBrowser.UnitTests.SampleDocs.iframecontainer.htm"));
                            }
                            else
                            {
                                responseContent = Encoding.UTF8.GetBytes(GetFromResources("SimpleBrowser.UnitTests.SampleDocs.SimpleForm.htm"));
                            }
                        }
                        mockResponse.Setup(r => r.GetResponseStream()).Returns(new MemoryStream(responseContent));
                        return Task.FromResult(mockResponse.Object);
                    });
                mock.SetupProperty(m => m.Headers, new WebHeaderCollection());
                mock.Setup(m => m.GetRequestStreamAsync()).Returns(Task.FromResult((System.IO.Stream)new MemoryStream(new byte[60000]) ));
                return mock.Object;
            }
        }

        internal static IWebRequestFactory GetFramesetMock()
        {
            return new FramesetRequestMocker();
        }

        private class FramesetRequestMocker : IWebRequestFactory
        {
            public IHttpWebRequest GetWebRequest(Uri url)
            {
                Mock<IHttpWebRequest> mock = new Mock<IHttpWebRequest>();
                mock.SetupAllProperties();
                mock.Setup(m => m.GetResponseAsync())
                    .Returns(() =>
                    {
                        Mock<IHttpWebResponse> mockResponse = new Mock<IHttpWebResponse>();
                        mockResponse.SetupAllProperties();
                        mockResponse.SetupProperty(m => m.Headers, new WebHeaderCollection());

                        byte[] responseContent = new byte[0];
                        if (mock.Object.Method == "GET")
                        {
                            if (url.AbsolutePath == "/")
                            {
                                responseContent = Encoding.UTF8.GetBytes(GetFromResources("SimpleBrowser.UnitTests.SampleDocs.framecontainer.htm"));
                            }
                            else
                            {
                                responseContent = Encoding.UTF8.GetBytes(GetFromResources("SimpleBrowser.UnitTests.SampleDocs.framecontent.htm"));
                            }
                        }
                        mockResponse.Setup(r => r.GetResponseStream()).Returns(new MemoryStream(responseContent));
                        return Task.FromResult( mockResponse.Object );
                    });
                mock.SetupProperty(m => m.Headers, new WebHeaderCollection());
                mock.Setup(m => m.GetRequestStreamAsync()).Returns(Task.FromResult((System.IO.Stream)new MemoryStream(new byte[60000])));
                
                return mock.Object;
            }
        }
    }
}