using System.Net;
using NUnit.Framework;
using Website.Controllers;

namespace UnitTests.Controllers
{
    [TestFixture]
    public class ErrorControllerTests : ControllerTester
    {
        [Test]
        public void No_status_code_should_show_general_error_page()
        {
            var controller = new TestErrorController();
            var result = controller.Index(null);

            Assert.AreEqual(0, controller.StatusCode);
            result.AssertViewName("General");
        }

        [Test]
        public void PageNotFound_status_code_should_show_NotFound_page()
        {
            var controller = new TestErrorController();
            var result = controller.Index(HttpStatusCode.NotFound.ToString());

            Assert.AreEqual(404, controller.StatusCode);
            result.AssertViewName("NotFound");
        }

        private class TestErrorController : ErrorController
        {
            public int StatusCode { get; set; }

            protected override void SetResponseStatusCode(int statusCode)
            {
                StatusCode = statusCode;
            }
        }
    }
}