DefaultControllerTests.cs

using Coscine.Api.Project.Models;
using Coscine.ApiCommons.Utils;
using Coscine.Database.Model;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.Mvc.Controllers;
using Microsoft.AspNetCore.Routing;
using Moq;
using NUnit.Framework;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Management;

namespace Coscine.Api.Project.Tests
{
    public abstract class DefaultControllerTests<T> where T : Controller
    {
        protected T Controller { get; }

        protected readonly List<Coscine.Database.Model.Project> Projects = new List<Coscine.Database.Model.Project>();
        protected readonly List<User> Users = new List<User>();
        protected readonly List<ProjectRole> ProjectRoles = new List<ProjectRole>();
        protected readonly List<ResourceType> ResourceTypes = new List<ResourceType>();
        protected readonly List<Resource> Resources = new List<Resource>();

        protected int Previous;

        protected DefaultControllerTests(T controller)
        {
            Controller = controller;
        }

        [OneTimeSetUp]
        public void Setup()
        {
            ProjectModel projectModel = new ProjectModel();
            Previous = projectModel.GetAll().ToArray().Length;

            UserModel userModel = new UserModel();

            var user = new User()
            {
                DisplayName = "TestUser",
                EmailAddress = "testUser@test.com",
            };

            userModel.Insert(user);
            Users.Add(user);

            FakeControllerContext(user);

            var project = new Coscine.Database.Model.Project()
            {
                Description = "Description",
                Organization = "Organization",
                DisplayName = "TestProject",
                StartDate = DateTime.Now,
                EndDate = DateTime.Now.AddYears(1),
                Keywords = "Test1;Test2"
            };

            projectModel.Insert(project);
            var projectRole = projectModel.SetOwner(project, user);
            ProjectRoles.Add(projectRole);

            Projects.Add(projectModel.GetById(project.Id));

            var project2 = new Coscine.Database.Model.Project()
            {
                Description = "Description2",
                Organization = "Organization2",
                DisplayName = "TestProject2",
                StartDate = DateTime.Now,
                EndDate = DateTime.Now.AddYears(1),
                Keywords = "Test1;Test2"
            };
            projectModel.Insert(project2);
            Projects.Add(projectModel.GetById(project2.Id));

            ResourceTypeModel resourceTypeModel = new ResourceTypeModel();
            var resourceType = new ResourceType()
            {
                DisplayName = "TestType"
            };
            resourceTypeModel.Insert(resourceType);
            ResourceTypes.Add(resourceType);

            ResourceModel resourceModel = new ResourceModel();
            var resource = new Resource()
            {
                ExternalId = "123",
                Url = "http://rwth-aachen.de/test",
                TypeId = resourceType.Id,
                Type = resourceType
            };
            resourceModel.Insert(resource);
            projectModel.AddResource(project, resource);
            Resources.Add(resource);

            var resource2 = new Resource()
            {
                ExternalId = "12345",
                Url = "http://rwth-aachen.de/test2",
                TypeId = resourceType.Id,
                Type = resourceType
            };
            resourceModel.Insert(resource2);
            projectModel.AddResource(project2, resource2);
            Resources.Add(resource2);
        }


        protected void FakeControllerContext(User user, Stream stream = null)
        {
            var request = new Mock<HttpRequest>();

            JWTHandler jwtHandler = new JWTHandler(Program.Configuration);
            Dictionary<string, string> values = new Dictionary<string, string>
            {
                { "UserId", user.Id.ToString() }
            };

            var additionalProtocol = Program.Configuration.GetStringAndWait("coscine/local/sharepoint/additional/protocol");
            var additionalPort = Program.Configuration.GetStringAndWait("coscine/local/sharepoint/additional/port");
            var additionalSite = Program.Configuration.GetStringAndWait("coscine/local/sharepoint/additional/site");

            var sharePointSite = "https://";

            if (additionalProtocol != null)
            {
                sharePointSite = additionalProtocol + "://";
            }

            //initialize the select query with command text
            ManagementObjectSearcher searcher =
               new ManagementObjectSearcher("root\\CIMV2",
               "SELECT Name FROM Win32_ComputerSystem");
            //execute the query
            foreach (ManagementObject process in searcher.Get())
            {
                //print system info
                process.Get();
                sharePointSite = sharePointSite + process["Name"] + "." + process["Domain"];
            }
            if (additionalPort != null)
            {
                sharePointSite += ":" + additionalPort;
            }
            if (additionalSite != null)
            {
                sharePointSite += "/" + additionalSite;
            }

            request.SetupGet(x => x.Headers).Returns(
                new HeaderDictionary {
                    {"X-Requested-With", "XMLHttpRequest"},
                    {"Authorization", "Bearer " + jwtHandler.GenerateJwtToken(values)},
                    {"Referer", sharePointSite }
                }
            );

            var context = new Mock<HttpContext>();
            context.SetupGet(x => x.Request).Returns(request.Object);

            if (stream != null)
            {
                context.SetupGet(x => x.Request.Method).Returns("POST");
                context.SetupGet(x => x.Request.HttpContext).Returns(context.Object);
                context.SetupGet(x => x.Request.Body).Returns(stream);
                context.SetupGet(x => x.Request.ContentLength).Returns(stream.Length);
                context.SetupGet(x => x.Request.ContentType).Returns("application/json");
            }

            var actionDescriptor = new Mock<ControllerActionDescriptor>();

            Controller.ControllerContext = new ControllerContext(new ActionContext(context.Object, new RouteData(), actionDescriptor.Object));
        }

        [OneTimeTearDown]
        public void End()
        {
            ProjectRoleModel projectRoleModel = new ProjectRoleModel();
            foreach (var projectRole in ProjectRoles)
            {
                projectRoleModel.Delete(projectRole);
            }
            ProjectResourceModel projectResourceModel = new ProjectResourceModel();
            foreach(var projectResource in projectResourceModel.GetAllWhere((projectResource) => projectResource.ResourceId == Resources[0].Id || projectResource.ResourceId == Resources[1].Id))
            {
                projectResourceModel.Delete(projectResource);
            }
            ProjectModel projectModel = new ProjectModel();
            foreach (var project in Projects)
            {
                projectModel.Delete(project);
            }
            UserModel userModel = new UserModel();
            foreach (var user in Users)
            {
                userModel.Delete(user);
            }
            ResourceModel resourceModel = new ResourceModel();
            foreach (var resource in Resources)
            {
                resourceModel.Delete(resource);
            }
            ResourceTypeModel resourceTypeModel = new ResourceTypeModel();
            foreach (var resourceType in ResourceTypes)
            {
                resourceTypeModel.Delete(resourceType);
            }
        }
    }
}